1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 // 5 #ifndef MEDIA_BASE_MOCK_DATA_SOURCE_HOST_H_ 6 #define MEDIA_BASE_MOCK_DATA_SOURCE_HOST_H_ 7 8 #include <string> 9 10 #include "media/base/data_source.h" 11 #include "testing/gmock/include/gmock/gmock.h" 12 13 namespace media { 14 15 class MockDataSourceHost : public DataSourceHost { 16 public: 17 MockDataSourceHost(); 18 virtual ~MockDataSourceHost(); 19 20 // DataSourceHost implementation. 21 MOCK_METHOD1(SetTotalBytes, void(int64 total_bytes)); 22 MOCK_METHOD2(AddBufferedByteRange, void(int64 start, int64 end)); 23 MOCK_METHOD2(AddBufferedTimeRange, void(base::TimeDelta start, 24 base::TimeDelta end)); 25 26 private: 27 DISALLOW_COPY_AND_ASSIGN(MockDataSourceHost); 28 }; 29 30 } // namespace media 31 32 #endif // MEDIA_BASE_MOCK_DATA_SOURCE_HOST_H_ 33