OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "media/test/mock_media_source.h" | 5 #include "media/test/mock_media_source.h" |
6 | 6 |
7 #include "base/threading/thread_task_runner_handle.h" | 7 #include "base/threading/thread_task_runner_handle.h" |
8 #include "media/base/test_data_util.h" | 8 #include "media/base/test_data_util.h" |
9 #include "media/base/timestamp_constants.h" | 9 #include "media/base/timestamp_constants.h" |
10 | 10 |
11 namespace media { | 11 namespace media { |
12 | 12 |
13 constexpr char kSourceId[] = "SourceId"; | 13 constexpr char kSourceId[] = "SourceId"; |
14 const size_t kAppendWholeFile = std::numeric_limits<size_t>::max(); | |
15 | 14 |
16 MockMediaSource::MockMediaSource(const std::string& filename, | 15 MockMediaSource::MockMediaSource(const std::string& filename, |
17 const std::string& mimetype, | 16 const std::string& mimetype, |
18 size_t initial_append_size) | 17 size_t initial_append_size) |
19 : current_position_(0), | 18 : current_position_(0), |
20 initial_append_size_(initial_append_size), | 19 initial_append_size_(initial_append_size), |
21 mimetype_(mimetype), | 20 mimetype_(mimetype), |
22 chunk_demuxer_(new ChunkDemuxer( | 21 chunk_demuxer_(new ChunkDemuxer( |
23 base::Bind(&MockMediaSource::DemuxerOpened, base::Unretained(this)), | 22 base::Bind(&MockMediaSource::DemuxerOpened, base::Unretained(this)), |
24 base::Bind(&MockMediaSource::OnEncryptedMediaInitData, | 23 base::Bind(&MockMediaSource::OnEncryptedMediaInitData, |
25 base::Unretained(this)), | 24 base::Unretained(this)), |
26 &media_log_)), | 25 &media_log_)), |
27 owned_chunk_demuxer_(chunk_demuxer_) { | 26 owned_chunk_demuxer_(chunk_demuxer_) { |
28 file_data_ = ReadTestDataFile(filename); | 27 file_data_ = ReadTestDataFile(filename); |
29 | 28 |
30 if (initial_append_size_ == kAppendWholeFile) | 29 if (initial_append_size_ == kAppendWholeFile) |
31 initial_append_size_ = file_data_->data_size(); | 30 initial_append_size_ = file_data_->data_size(); |
32 | 31 |
33 DCHECK_GT(initial_append_size_, 0u); | 32 CHECK_GT(initial_append_size_, 0u); |
34 DCHECK_LE(initial_append_size_, file_data_->data_size()); | 33 CHECK_LE(initial_append_size_, file_data_->data_size()); |
35 } | 34 } |
36 | 35 |
37 MockMediaSource::~MockMediaSource() {} | 36 MockMediaSource::~MockMediaSource() {} |
38 | 37 |
39 std::unique_ptr<Demuxer> MockMediaSource::GetDemuxer() { | 38 std::unique_ptr<Demuxer> MockMediaSource::GetDemuxer() { |
40 return std::move(owned_chunk_demuxer_); | 39 return std::move(owned_chunk_demuxer_); |
41 } | 40 } |
42 | 41 |
43 void MockMediaSource::Seek(base::TimeDelta seek_time, | 42 void MockMediaSource::Seek(base::TimeDelta seek_time, |
44 size_t new_position, | 43 size_t new_position, |
45 size_t seek_append_size) { | 44 size_t seek_append_size) { |
46 chunk_demuxer_->StartWaitingForSeek(seek_time); | 45 chunk_demuxer_->StartWaitingForSeek(seek_time); |
47 | 46 |
48 chunk_demuxer_->ResetParserState(kSourceId, base::TimeDelta(), | 47 chunk_demuxer_->ResetParserState(kSourceId, base::TimeDelta(), |
49 kInfiniteDuration, &last_timestamp_offset_); | 48 kInfiniteDuration, &last_timestamp_offset_); |
50 | 49 |
51 DCHECK_LT(new_position, file_data_->data_size()); | 50 CHECK_LT(new_position, file_data_->data_size()); |
52 current_position_ = new_position; | 51 current_position_ = new_position; |
53 | 52 |
54 AppendData(seek_append_size); | 53 AppendData(seek_append_size); |
55 } | 54 } |
56 | 55 |
57 void MockMediaSource::Seek(base::TimeDelta seek_time) { | 56 void MockMediaSource::Seek(base::TimeDelta seek_time) { |
58 chunk_demuxer_->StartWaitingForSeek(seek_time); | 57 chunk_demuxer_->StartWaitingForSeek(seek_time); |
59 } | 58 } |
60 | 59 |
61 void MockMediaSource::AppendData(size_t size) { | 60 void MockMediaSource::AppendData(size_t size) { |
62 DCHECK(chunk_demuxer_); | 61 CHECK(chunk_demuxer_); |
63 DCHECK_LT(current_position_, file_data_->data_size()); | 62 CHECK_LT(current_position_, file_data_->data_size()); |
64 DCHECK_LE(current_position_ + size, file_data_->data_size()); | 63 CHECK_LE(current_position_ + size, file_data_->data_size()); |
65 | 64 |
66 ASSERT_TRUE(chunk_demuxer_->AppendData( | 65 ASSERT_TRUE(chunk_demuxer_->AppendData( |
67 kSourceId, file_data_->data() + current_position_, size, | 66 kSourceId, file_data_->data() + current_position_, size, |
68 base::TimeDelta(), kInfiniteDuration, &last_timestamp_offset_)); | 67 base::TimeDelta(), kInfiniteDuration, &last_timestamp_offset_)); |
69 current_position_ += size; | 68 current_position_ += size; |
70 } | 69 } |
71 | 70 |
72 bool MockMediaSource::AppendAtTime(base::TimeDelta timestamp_offset, | 71 bool MockMediaSource::AppendAtTime(base::TimeDelta timestamp_offset, |
73 const uint8_t* pData, | 72 const uint8_t* pData, |
74 int size) { | 73 int size) { |
(...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
163 codecs_param = mimetype_.substr(codecs_param_start, | 162 codecs_param = mimetype_.substr(codecs_param_start, |
164 codecs_param_end - codecs_param_start); | 163 codecs_param_end - codecs_param_start); |
165 } | 164 } |
166 | 165 |
167 return chunk_demuxer_->AddId(kSourceId, type, codecs_param); | 166 return chunk_demuxer_->AddId(kSourceId, type, codecs_param); |
168 } | 167 } |
169 | 168 |
170 void MockMediaSource::OnEncryptedMediaInitData( | 169 void MockMediaSource::OnEncryptedMediaInitData( |
171 EmeInitDataType init_data_type, | 170 EmeInitDataType init_data_type, |
172 const std::vector<uint8_t>& init_data) { | 171 const std::vector<uint8_t>& init_data) { |
173 DCHECK(!init_data.empty()); | 172 CHECK(!init_data.empty()); |
174 CHECK(!encrypted_media_init_data_cb_.is_null()); | 173 CHECK(!encrypted_media_init_data_cb_.is_null()); |
175 encrypted_media_init_data_cb_.Run(init_data_type, init_data); | 174 encrypted_media_init_data_cb_.Run(init_data_type, init_data); |
176 } | 175 } |
177 | 176 |
178 void MockMediaSource::InitSegmentReceived(std::unique_ptr<MediaTracks> tracks) { | 177 void MockMediaSource::InitSegmentReceived(std::unique_ptr<MediaTracks> tracks) { |
179 CHECK(tracks.get()); | 178 CHECK(tracks.get()); |
180 EXPECT_GT(tracks->tracks().size(), 0u); | 179 EXPECT_GT(tracks->tracks().size(), 0u); |
181 CHECK(chunk_demuxer_); | 180 CHECK(chunk_demuxer_); |
182 // Verify that track ids are unique. | 181 // Verify that track ids are unique. |
183 std::set<MediaTrack::Id> track_ids; | 182 std::set<MediaTrack::Id> track_ids; |
184 for (const auto& track : tracks->tracks()) { | 183 for (const auto& track : tracks->tracks()) { |
185 EXPECT_EQ(track_ids.end(), track_ids.find(track->id())); | 184 EXPECT_EQ(track_ids.end(), track_ids.find(track->id())); |
186 track_ids.insert(track->id()); | 185 track_ids.insert(track->id()); |
187 } | 186 } |
188 InitSegmentReceivedMock(tracks); | 187 InitSegmentReceivedMock(tracks); |
189 } | 188 } |
190 | 189 |
191 } // namespace media | 190 } // namespace media |
OLD | NEW |