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.
6 #include "base/files/file_path.h"
7 #include "base/synchronization/waitable_event.h"
8 #include "media/base/test_data_util.h"
9 #include "media/ffmpeg/ffmpeg_common.h"
10 #include "media/filters/blocking_url_protocol.h"
11 #include "media/filters/file_data_source.h"
12 #include "testing/gmock/include/gmock/gmock.h"
13 #include "testing/gtest/include/gtest/gtest.h"
17 class BlockingUrlProtocolTest : public testing::Test {
19 BlockingUrlProtocolTest()
20 : url_protocol_(&data_source_,
21 base::Bind(&BlockingUrlProtocolTest::OnDataSourceError,
22 base::Unretained(this))) {
23 CHECK(data_source_.Initialize(GetTestDataFilePath("bear-320x240.webm")));
26 virtual ~BlockingUrlProtocolTest() {
27 base::WaitableEvent stop_event(false, false);
28 data_source_.Stop(base::Bind(
29 &base::WaitableEvent::Signal, base::Unretained(&stop_event)));
33 MOCK_METHOD0(OnDataSourceError, void());
35 FileDataSource data_source_;
36 BlockingUrlProtocol url_protocol_;
39 DISALLOW_COPY_AND_ASSIGN(BlockingUrlProtocolTest);
43 TEST_F(BlockingUrlProtocolTest, Read) {
44 // Set read head to zero as Initialize() will have parsed a bit of the file.
46 EXPECT_TRUE(url_protocol_.SetPosition(0));
47 EXPECT_TRUE(url_protocol_.GetPosition(&position));
48 EXPECT_EQ(0, position);
50 // Read 32 bytes from offset zero and verify position.
52 EXPECT_EQ(32, url_protocol_.Read(32, buffer));
53 EXPECT_TRUE(url_protocol_.GetPosition(&position));
54 EXPECT_EQ(32, position);
56 // Read an additional 32 bytes and verify position.
57 EXPECT_EQ(32, url_protocol_.Read(32, buffer));
58 EXPECT_TRUE(url_protocol_.GetPosition(&position));
59 EXPECT_EQ(64, position);
61 // Seek to end and read until EOF.
63 EXPECT_TRUE(url_protocol_.GetSize(&size));
64 EXPECT_TRUE(url_protocol_.SetPosition(size - 48));
65 EXPECT_EQ(32, url_protocol_.Read(32, buffer));
66 EXPECT_TRUE(url_protocol_.GetPosition(&position));
67 EXPECT_EQ(size - 16, position);
69 EXPECT_EQ(16, url_protocol_.Read(32, buffer));
70 EXPECT_TRUE(url_protocol_.GetPosition(&position));
71 EXPECT_EQ(size, position);
73 EXPECT_EQ(0, url_protocol_.Read(32, buffer));
74 EXPECT_TRUE(url_protocol_.GetPosition(&position));
75 EXPECT_EQ(size, position);
78 TEST_F(BlockingUrlProtocolTest, ReadError) {
79 data_source_.force_read_errors_for_testing();
82 EXPECT_CALL(*this, OnDataSourceError());
83 EXPECT_EQ(AVERROR(EIO), url_protocol_.Read(32, buffer));
86 TEST_F(BlockingUrlProtocolTest, GetSetPosition) {
89 EXPECT_TRUE(url_protocol_.GetSize(&size));
90 EXPECT_TRUE(url_protocol_.GetPosition(&position));
92 EXPECT_TRUE(url_protocol_.SetPosition(512));
93 EXPECT_FALSE(url_protocol_.SetPosition(size + 1));
94 EXPECT_FALSE(url_protocol_.SetPosition(-1));
95 EXPECT_TRUE(url_protocol_.GetPosition(&position));
96 EXPECT_EQ(512, position);
98 EXPECT_TRUE(url_protocol_.SetPosition(size));
99 EXPECT_TRUE(url_protocol_.GetPosition(&position));
100 EXPECT_EQ(size, position);
103 TEST_F(BlockingUrlProtocolTest, GetSize) {
104 int64 data_source_size = 0;
105 int64 url_protocol_size = 0;
106 EXPECT_TRUE(data_source_.GetSize(&data_source_size));
107 EXPECT_TRUE(url_protocol_.GetSize(&url_protocol_size));
108 EXPECT_NE(0, data_source_size);
109 EXPECT_EQ(data_source_size, url_protocol_size);
112 TEST_F(BlockingUrlProtocolTest, IsStreaming) {
113 EXPECT_FALSE(data_source_.IsStreaming());
114 EXPECT_FALSE(url_protocol_.IsStreaming());
116 data_source_.force_streaming_for_testing();
117 EXPECT_TRUE(data_source_.IsStreaming());
118 EXPECT_TRUE(url_protocol_.IsStreaming());