Index: content/renderer/media/video_source_handler_unittest.cc |
diff --git a/content/renderer/media/video_source_handler_unittest.cc b/content/renderer/media/video_source_handler_unittest.cc |
index 94dce46a19511a5504654eca553ce5fe10124ef8..1a5cb78326d84f2d701292ab1456b78b1f186641 100644 |
--- a/content/renderer/media/video_source_handler_unittest.cc |
+++ b/content/renderer/media/video_source_handler_unittest.cc |
@@ -17,6 +17,7 @@ |
#include "testing/gtest/include/gtest/gtest.h" |
#include "third_party/WebKit/public/platform/WebMediaStreamTrack.h" |
#include "third_party/WebKit/public/platform/WebString.h" |
+#include "third_party/WebKit/public/web/WebHeap.h" |
namespace content { |
@@ -44,17 +45,22 @@ class VideoSourceHandlerTest : public ::testing::Test { |
public: |
VideoSourceHandlerTest() |
: child_process_(new ChildProcess()), |
- registry_() { |
- handler_.reset(new VideoSourceHandler(®istry_)); |
- registry_.Init(kTestStreamUrl); |
- registry_.AddVideoTrack(kTestVideoTrackId); |
+ registry_(new MockMediaStreamRegistry()) { |
+ handler_.reset(new VideoSourceHandler(registry_.get())); |
+ registry_->Init(kTestStreamUrl); |
+ registry_->AddVideoTrack(kTestVideoTrackId); |
+ } |
+ |
+ virtual void TearDown() { |
+ registry_.reset(); |
+ blink::WebHeap::collectAllGarbageForTesting(); |
} |
protected: |
base::MessageLoop message_loop_; |
scoped_ptr<ChildProcess> child_process_; |
scoped_ptr<VideoSourceHandler> handler_; |
- MockMediaStreamRegistry registry_; |
+ scoped_ptr<MockMediaStreamRegistry> registry_; |
}; |
TEST_F(VideoSourceHandlerTest, OpenClose) { |