WebRtcMediaStreamTrackMap added

content/renderer/media/webrtc/webrtc_media_stream_track_adapter_map_unittest.cc

Index: content/renderer/media/webrtc/webrtc_media_stream_track_adapter_map_unittest.cc
diff --git a/content/renderer/media/webrtc/webrtc_media_stream_track_adapter_map_unittest.cc b/content/renderer/media/webrtc/webrtc_media_stream_track_adapter_map_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..faaf7c6e358549e4fef11e2f9ee46b7f1730b8c8
--- /dev/null
+++ b/content/renderer/media/webrtc/webrtc_media_stream_track_adapter_map_unittest.cc
@@ -0,0 +1,210 @@
+// Copyright (c) 2017 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 "content/renderer/media/webrtc/webrtc_media_stream_track_adapter_map.h"
+
+#include <memory>
+
+#include "base/memory/ref_counted.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "base/single_thread_task_runner.h"
+#include "content/child/child_process.h"
+#include "content/renderer/media/media_stream_audio_source.h"
+#include "content/renderer/media/webrtc/mock_peer_connection_dependency_factory.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/WebMediaStreamSource.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 {
+
+class WebRtcMediaStreamTrackAdapterMapTest : public ::testing::Test {
+ public:
+  void SetUp() override {
+    dependency_factory_.reset(new MockPeerConnectionDependencyFactory());
+    main_thread_ = base::ThreadTaskRunnerHandle::Get();
+    map_ = new WebRtcMediaStreamTrackAdapterMap(dependency_factory_.get(),
+                                                main_thread_);
+  }
+
+  void TearDown() override { blink::WebHeap::CollectAllGarbageForTesting(); }
+
+  blink::WebMediaStreamTrack CreateLocalTrack(const std::string& id) {
+    blink::WebMediaStreamSource web_source;
+    web_source.Initialize(
+        blink::WebString::FromUTF8(id), blink::WebMediaStreamSource::kTypeAudio,
+        blink::WebString::FromUTF8("local_audio_track"), false);
+    MediaStreamAudioSource* audio_source = new MediaStreamAudioSource(true);
+    // Takes ownership of |audio_source|.
+    web_source.SetExtraData(audio_source);
+
+    blink::WebMediaStreamTrack web_track;
+    web_track.Initialize(web_source.Id(), web_source);
+    audio_source->ConnectToTrack(web_track);
+    return web_track;
+  }
+
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef>
+  GetOrCreateRemoteTrackAdapter(
+      webrtc::MediaStreamTrackInterface* webrtc_track) {
+    DCHECK(main_thread_->BelongsToCurrentThread());
+    std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> adapter;
+    dependency_factory_->GetWebRtcSignalingThread()->PostTask(
+        FROM_HERE,
+        base::Bind(&WebRtcMediaStreamTrackAdapterMapTest::
+                       GetOrCreateRemoteTrackAdapterOnSignalingThread,
+                   base::Unretained(this), webrtc_track, &adapter));
+    RunMessageLoopsUntilIdle();
+    return adapter;
+  }
+
+  void GetOrCreateRemoteTrackAdapterOnSignalingThread(
+      webrtc::MediaStreamTrackInterface* webrtc_track,
+      std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef>* adapter) {
+    DCHECK(dependency_factory_->GetWebRtcSignalingThread()
+               ->BelongsToCurrentThread());
+    *adapter = map_->GetOrCreateRemoteTrackAdapter(webrtc_track);
+  }
+
+  // Runs message loops on the webrtc signaling thread and the main thread until
+  // idle.
+  void RunMessageLoopsUntilIdle() {
+    DCHECK(main_thread_->BelongsToCurrentThread());
+    base::WaitableEvent waitable_event(
+        base::WaitableEvent::ResetPolicy::MANUAL,
+        base::WaitableEvent::InitialState::NOT_SIGNALED);
+    dependency_factory_->GetWebRtcSignalingThread()->PostTask(
+        FROM_HERE, base::Bind(&WebRtcMediaStreamTrackAdapterMapTest::
+                                  RunMessageLoopUntilIdleOnSignalingThread,
+                              base::Unretained(this), &waitable_event));
+    waitable_event.Wait();
+    base::RunLoop().RunUntilIdle();
+  }
+
+  void RunMessageLoopUntilIdleOnSignalingThread(
+      base::WaitableEvent* waitable_event) {
+    DCHECK(dependency_factory_->GetWebRtcSignalingThread()
+               ->BelongsToCurrentThread());
+    base::RunLoop().RunUntilIdle();
+    waitable_event->Signal();
+  }
+
+ protected:
+  // Message loop and child processes is needed for task queues and threading to
+  // work, as is necessary to create tracks and adapters.
+  base::MessageLoop message_loop_;
+  ChildProcess child_process_;
+
+  std::unique_ptr<MockPeerConnectionDependencyFactory> dependency_factory_;
+  scoped_refptr<base::SingleThreadTaskRunner> main_thread_;
+  scoped_refptr<WebRtcMediaStreamTrackAdapterMap> map_;
+};
+
+TEST_F(WebRtcMediaStreamTrackAdapterMapTest, AddAndRemoveLocalTrackAdapter) {
+  blink::WebMediaStreamTrack web_track = CreateLocalTrack("local_track");
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> adapter_ref =
+      map_->GetOrCreateLocalTrackAdapter(web_track);
+  EXPECT_TRUE(adapter_ref->is_initialized());
+  EXPECT_EQ(adapter_ref->GetAdapterForTesting(),
+            map_->GetLocalTrackAdapter(web_track.Id().Utf8())
+                ->GetAdapterForTesting());
+  EXPECT_EQ(1u, map_->GetLocalTrackCount());
+
+  // "GetOrCreate" for already existing track.
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> adapter_ref2 =
+      map_->GetOrCreateLocalTrackAdapter(web_track);
+  EXPECT_EQ(adapter_ref->GetAdapterForTesting(),
+            adapter_ref2->GetAdapterForTesting());
+  EXPECT_EQ(1u, map_->GetLocalTrackCount());
+
+  adapter_ref2.reset();  // Not the last reference.
+  EXPECT_TRUE(adapter_ref->GetAdapterForTesting()->is_initialized());
+  EXPECT_EQ(1u, map_->GetLocalTrackCount());
+
+  // Destroying all references to the adapter should remove it from the map and
+  // dispose it.
+  adapter_ref.reset();
+  EXPECT_EQ(0u, map_->GetLocalTrackCount());
+  EXPECT_EQ(nullptr, map_->GetLocalTrackAdapter(web_track.Id().Utf8()));
+  // Allow the disposing of track to occur.
+  RunMessageLoopsUntilIdle();
+}
+
+TEST_F(WebRtcMediaStreamTrackAdapterMapTest, AddAndRemoveRemoteTrackAdapter) {
+  scoped_refptr<MockWebRtcAudioTrack> webrtc_track =
+      MockWebRtcAudioTrack::Create("remote_track");
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> adapter_ref =
+      GetOrCreateRemoteTrackAdapter(webrtc_track.get());
+  EXPECT_TRUE(adapter_ref->is_initialized());
+  EXPECT_EQ(
+      adapter_ref->GetAdapterForTesting(),
+      map_->GetRemoteTrackAdapter(webrtc_track->id())->GetAdapterForTesting());
+  EXPECT_EQ(1u, map_->GetRemoteTrackCount());
+
+  // "GetOrCreate" for already existing track.
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> adapter_ref2 =
+      GetOrCreateRemoteTrackAdapter(webrtc_track.get());
+  EXPECT_EQ(adapter_ref->GetAdapterForTesting(),
+            adapter_ref2->GetAdapterForTesting());
+  EXPECT_EQ(1u, map_->GetRemoteTrackCount());
+
+  adapter_ref2.reset();  // Not the last reference.
+  EXPECT_TRUE(adapter_ref->GetAdapterForTesting()->is_initialized());
+  EXPECT_EQ(1u, map_->GetRemoteTrackCount());
+
+  // Destroying all references to the adapter should remove it from the map and
+  // dispose it.
+  adapter_ref.reset();
+  EXPECT_EQ(0u, map_->GetRemoteTrackCount());
+  EXPECT_EQ(nullptr, map_->GetRemoteTrackAdapter(webrtc_track->id()));
+  // Allow the disposing of track to occur.
+  RunMessageLoopsUntilIdle();
+}
+
+TEST_F(WebRtcMediaStreamTrackAdapterMapTest,
+       LocalAndRemoteTrackAdaptersWithSameID) {
+  // Local and remote tracks should be able to use the same id without conflict.
+  const char* id = "id";
+
+  blink::WebMediaStreamTrack local_web_track = CreateLocalTrack(id);
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> local_adapter =
+      map_->GetOrCreateLocalTrackAdapter(local_web_track);
+  EXPECT_TRUE(local_adapter->is_initialized());
+  EXPECT_EQ(local_adapter->GetAdapterForTesting(),
+            map_->GetLocalTrackAdapter(id)->GetAdapterForTesting());
+  EXPECT_EQ(1u, map_->GetLocalTrackCount());
+
+  scoped_refptr<MockWebRtcAudioTrack> remote_webrtc_track =
+      MockWebRtcAudioTrack::Create(id);
+  std::unique_ptr<WebRtcMediaStreamTrackAdapterMap::AdapterRef> remote_adapter =
+      GetOrCreateRemoteTrackAdapter(remote_webrtc_track.get());
+  EXPECT_TRUE(remote_adapter->is_initialized());
+  EXPECT_EQ(remote_adapter->GetAdapterForTesting(),
+            map_->GetRemoteTrackAdapter(id)->GetAdapterForTesting());
+  EXPECT_NE(local_adapter->GetAdapterForTesting(),
+            remote_adapter->GetAdapterForTesting());
+  EXPECT_EQ(1u, map_->GetRemoteTrackCount());
+
+  // Destroying all references to the adapters should remove them from the map.
+  local_adapter.reset();
+  remote_adapter.reset();
+  EXPECT_EQ(0u, map_->GetLocalTrackCount());
+  EXPECT_EQ(0u, map_->GetRemoteTrackCount());
+  EXPECT_EQ(nullptr, map_->GetLocalTrackAdapter(id));
+  EXPECT_EQ(nullptr, map_->GetRemoteTrackAdapter(id));
+  // Allow the disposing of tracks to occur.
+  RunMessageLoopsUntilIdle();
+}
+
+TEST_F(WebRtcMediaStreamTrackAdapterMapTest, GetLocalTrackAdapterInvalidID) {
+  EXPECT_EQ(nullptr, map_->GetLocalTrackAdapter("invalid"));
+}
+
+TEST_F(WebRtcMediaStreamTrackAdapterMapTest, GetRemoteTrackAdapterInvalidID) {
+  EXPECT_EQ(nullptr, map_->GetRemoteTrackAdapter("invalid"));
+}
+
+}  // namespace content