#ifndef MEDIA_BASE_MOCK_DATA_SOURCE_HOST_H_
#define MEDIA_BASE_MOCK_DATA_SOURCE_HOST_H_
#include <string>
#include "media/base/data_source.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace media {
class MockDataSourceHost : public DataSourceHost {
public:
MockDataSourceHost();
virtual ~MockDataSourceHost();
MOCK_METHOD1(SetTotalBytes, void(int64 total_bytes));
MOCK_METHOD2(AddBufferedByteRange, void(int64 start, int64 end));
private:
DISALLOW_COPY_AND_ASSIGN(MockDataSourceHost);
};
}
#endif