Notify demux stream clients when a stream is disabled.

media/filters/chunk_demuxer_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/filters/chunk_demuxer.h"
 
 #include <stddef.h>
 #include <stdint.h>
 #include <algorithm>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
+#include "base/synchronization/waitable_event.h"
 #include "media/base/audio_decoder_config.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decrypt_config.h"
 #include "media/base/media.h"
 #include "media/base/media_tracks.h"
 #include "media/base/mock_demuxer_host.h"
 #include "media/base/mock_media_log.h"
 #include "media/base/test_data_util.h"
 #include "media/base/test_helpers.h"
 #include "media/base/timestamp_constants.h"
@@ skipping 4670 matching lines @@
   ASSERT_TRUE(AppendData(kSourceId, cluster->data() + video_start,
                          cluster->size() - video_start));
 
   CheckExpectedRanges(DemuxerStream::AUDIO, "{ [30,90) }");
   CheckExpectedRanges(DemuxerStream::VIDEO, "{ [0,91) }");
   CheckExpectedRanges("{ [30,90) }");
   CheckExpectedBuffers(audio_stream, "30K 40K 50K 60K 70K 80K");
   CheckExpectedBuffers(video_stream, "71K 81");
 }
 
+void OnStreamStatusChanged(base::WaitableEvent* event,
+                           DemuxerStream* expected_stream,
+                           DemuxerStream* actual_stream,
+                           bool enabled,
+                           base::TimeDelta) {
+  EXPECT_EQ(expected_stream, actual_stream);
+  EXPECT_EQ(enabled, actual_stream->enabled());
+  event->Signal();
+}
+
+void CheckStreamStatusNotifications(DemuxerStream* stream) {
+  base::WaitableEvent event(base::WaitableEvent::ResetPolicy::AUTOMATIC,
+                            base::WaitableEvent::InitialState::NOT_SIGNALED);
+
+  ASSERT_TRUE(stream->enabled());
+  stream->SetStreamStatusChangeCB(base::Bind(&OnStreamStatusChanged,
+                                             base::Unretained(&event),
+                                             base::Unretained(stream)));
+
+  stream->set_enabled(false, base::TimeDelta());
+  base::RunLoop().RunUntilIdle();
+  ASSERT_TRUE(event.IsSignaled());
+
+  event.Reset();
+  stream->set_enabled(true, base::TimeDelta());
+  base::RunLoop().RunUntilIdle();
+  ASSERT_TRUE(event.IsSignaled());
+}
+
+TEST_F(ChunkDemuxerTest, StreamStatusNotifications) {
+  ASSERT_TRUE(InitDemuxer(HAS_AUDIO | HAS_VIDEO));
+  DemuxerStream* audio_stream = demuxer_->GetStream(DemuxerStream::AUDIO);
+  EXPECT_NE(nullptr, audio_stream);
+  CheckStreamStatusNotifications(audio_stream);
+  DemuxerStream* video_stream = demuxer_->GetStream(DemuxerStream::VIDEO);
+  EXPECT_NE(nullptr, video_stream);
+  CheckStreamStatusNotifications(video_stream);
+}
+
 }  // namespace media