Handle early destruction of CanvasCaptureHandler

content/renderer/media/canvas_capture_handler_unittest.cc

 // Copyright 2015 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 "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/thread_task_runner_handle.h"
 #include "content/child/child_process.h"
 #include "content/renderer/media/canvas_capture_handler.h"
 #include "content/renderer/media/media_stream_video_capturer_source.h"
@@ skipping 124 matching lines @@
  private:
   DISALLOW_COPY_AND_ASSIGN(CanvasCaptureHandlerTest);
 };
 
 // Checks that the initialization-destruction sequence works fine.
 TEST_F(CanvasCaptureHandlerTest, ConstructAndDestruct) {
   EXPECT_TRUE(canvas_capture_handler_->needsNewFrame());
   base::RunLoop().RunUntilIdle();
 }
 
+// Checks that the destruction sequence works fine.
+TEST_F(CanvasCaptureHandlerTest, DestructTrack) {
+  EXPECT_TRUE(canvas_capture_handler_->needsNewFrame());
+  track_.reset();
+  base::RunLoop().RunUntilIdle();
+}
+
+// Checks that the destruction sequence works fine.
+TEST_F(CanvasCaptureHandlerTest, DestructHandler) {
+  EXPECT_TRUE(canvas_capture_handler_->needsNewFrame());
+  canvas_capture_handler_.reset();
+  base::RunLoop().RunUntilIdle();
+}
+
 // Checks that VideoCapturerSource call sequence works fine.
 TEST_P(CanvasCaptureHandlerTest, GetFormatsStartAndStop) {
   InSequence s;
   const blink::WebMediaStreamSource& web_media_stream_source = track_.source();
   EXPECT_FALSE(web_media_stream_source.isNull());
   MediaStreamVideoCapturerSource* const ms_source =
       static_cast<MediaStreamVideoCapturerSource*>(
           web_media_stream_source.getExtraData());
   EXPECT_TRUE(ms_source != nullptr);
   media::VideoCapturerSource* source = GetVideoCapturerSource(ms_source);
@@ skipping 59 matching lines @@
           track_.source().getExtraData()));
   EXPECT_TRUE(source != nullptr);
   EXPECT_TRUE(canvas_capture_handler_->needsNewFrame());
   source->StopCapture();
   EXPECT_FALSE(canvas_capture_handler_->needsNewFrame());
 }
 
 INSTANTIATE_TEST_CASE_P(, CanvasCaptureHandlerTest, ::testing::Bool());
 
 }  // namespace content