Add CombinedDesktopMediaList class

chrome/browser/media/native_desktop_media_list_unittest.cc

 // Copyright 2013 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 "chrome/browser/media/native_desktop_media_list.h"
 
 #include <stddef.h>
 #include <stdint.h>
 #include <string.h>
 
@@ skipping 11 matching lines @@
 #include "third_party/webrtc/modules/desktop_capture/screen_capturer.h"
 #include "third_party/webrtc/modules/desktop_capture/window_capturer.h"
 
 using testing::_;
 using testing::DoAll;
 
 namespace {
 
 class MockObserver : public DesktopMediaListObserver {
  public:
-  MOCK_METHOD1(OnSourceAdded, void(int index));
-  MOCK_METHOD1(OnSourceRemoved, void(int index));
-  MOCK_METHOD2(OnSourceMoved, void(int old_index, int new_index));
-  MOCK_METHOD1(OnSourceNameChanged, void(int index));
-  MOCK_METHOD1(OnSourceThumbnailChanged, void(int index));
+  MOCK_METHOD2(OnSourceAdded, void(DesktopMediaList* list, int index));
+  MOCK_METHOD2(OnSourceRemoved, void(DesktopMediaList* list, int index));
+  MOCK_METHOD3(OnSourceMoved,
+               void(DesktopMediaList* list, int old_index, int new_index));
+  MOCK_METHOD2(OnSourceNameChanged, void(DesktopMediaList* list, int index));
+  MOCK_METHOD2(OnSourceThumbnailChanged,
+               void(DesktopMediaList* list, int index));
 };
 
 class FakeScreenCapturer : public webrtc::ScreenCapturer {
  public:
   FakeScreenCapturer() {}
   ~FakeScreenCapturer() override {}
 
   // webrtc::ScreenCapturer implementation.
   void Start(Callback* callback) override { callback_ = callback; }
 
@@ skipping 121 matching lines @@
       window.id = i + 1;
       window.title = "Test window";
       list.push_back(window);
     }
     window_capturer_->SetWindowList(list);
 
     {
       testing::InSequence dummy;
       size_t source_count = window_only ? count : count + 1;
       for (size_t i = 0; i < source_count; ++i) {
-        EXPECT_CALL(observer_, OnSourceAdded(i))
-          .WillOnce(CheckListSize(model_.get(), static_cast<int>(i + 1)));
+        EXPECT_CALL(observer_, OnSourceAdded(model_.get(), i))
+            .WillOnce(CheckListSize(model_.get(), static_cast<int>(i + 1)));
       }
       for (size_t i = 0; i < source_count - 1; ++i) {
-        EXPECT_CALL(observer_, OnSourceThumbnailChanged(i));
+        EXPECT_CALL(observer_, OnSourceThumbnailChanged(model_.get(), i));
       }
-      EXPECT_CALL(observer_, OnSourceThumbnailChanged(source_count - 1))
-        .WillOnce(QuitMessageLoop(&message_loop_));
+      EXPECT_CALL(observer_,
+                  OnSourceThumbnailChanged(model_.get(), source_count - 1))
+          .WillOnce(QuitMessageLoop(&message_loop_));
     }
     model_->StartUpdating(&observer_);
     message_loop_.Run();
 
     for (size_t i = 0; i < count; ++i) {
       size_t source_index = window_only ? i : i + 1;
       EXPECT_EQ(model_->GetSource(source_index).id.type,
                                   content::DesktopMediaID::TYPE_WINDOW);
       EXPECT_EQ(model_->GetSource(source_index).id.id, static_cast<int>(i + 1));
       EXPECT_EQ(model_->GetSource(source_index).name,
@@ skipping 44 matching lines @@
   AddWindowsAndVerify(1, true);
 }
 
 TEST_F(DesktopMediaListTest, ScreenOnly) {
   model_.reset(new NativeDesktopMediaList(
       scoped_ptr<webrtc::ScreenCapturer>(new FakeScreenCapturer),
       scoped_ptr<webrtc::WindowCapturer>()));
 
   {
     testing::InSequence dummy;
-    EXPECT_CALL(observer_, OnSourceAdded(0))
-      .WillOnce(CheckListSize(model_.get(), 1));
-    EXPECT_CALL(observer_, OnSourceThumbnailChanged(0))
-      .WillOnce(QuitMessageLoop(&message_loop_));
+    EXPECT_CALL(observer_, OnSourceAdded(model_.get(), 0))
+        .WillOnce(CheckListSize(model_.get(), 1));
+    EXPECT_CALL(observer_, OnSourceThumbnailChanged(model_.get(), 0))
+        .WillOnce(QuitMessageLoop(&message_loop_));
   }
   model_->StartUpdating(&observer_);
 
   message_loop_.Run();
 
   EXPECT_EQ(model_->GetSource(0).id.type, content::DesktopMediaID::TYPE_SCREEN);
 }
 
 TEST_F(DesktopMediaListTest, AddWindow) {
   CreateWithDefaultCapturers();
   webrtc::WindowCapturer::WindowList list = AddWindowsAndVerify(1, false);
 
-  EXPECT_CALL(observer_, OnSourceAdded(2))
-    .WillOnce(DoAll(CheckListSize(model_.get(), 3),
-                    QuitMessageLoop(&message_loop_)));
+  EXPECT_CALL(observer_, OnSourceAdded(model_.get(), 2))
+      .WillOnce(DoAll(CheckListSize(model_.get(), 3),
+                      QuitMessageLoop(&message_loop_)));
 
   webrtc::WindowCapturer::Window window;
   window.id = 10;  // id=0 is invalid.
   window.title = "Test window 10";
   list.push_back(window);
   window_capturer_->SetWindowList(list);
 
   message_loop_.Run();
 
   EXPECT_EQ(model_->GetSource(2).id.type, content::DesktopMediaID::TYPE_WINDOW);
   EXPECT_EQ(model_->GetSource(2).id.id, 10);
 }
 
 TEST_F(DesktopMediaListTest, RemoveWindow) {
   CreateWithDefaultCapturers();
   webrtc::WindowCapturer::WindowList list = AddWindowsAndVerify(2, false);
 
-  EXPECT_CALL(observer_, OnSourceRemoved(2))
-    .WillOnce(DoAll(CheckListSize(model_.get(), 2),
-                    QuitMessageLoop(&message_loop_)));
+  EXPECT_CALL(observer_, OnSourceRemoved(model_.get(), 2))
+      .WillOnce(DoAll(CheckListSize(model_.get(), 2),
+                      QuitMessageLoop(&message_loop_)));
 
   list.erase(list.begin() + 1);
   window_capturer_->SetWindowList(list);
 
   message_loop_.Run();
 }
 
 TEST_F(DesktopMediaListTest, RemoveAllWindows) {
   CreateWithDefaultCapturers();
   webrtc::WindowCapturer::WindowList list = AddWindowsAndVerify(2, false);
 
   testing::InSequence seq;
-  EXPECT_CALL(observer_, OnSourceRemoved(1))
-    .WillOnce(CheckListSize(model_.get(), 2));
-  EXPECT_CALL(observer_, OnSourceRemoved(1))
-    .WillOnce(DoAll(CheckListSize(model_.get(), 1),
-                    QuitMessageLoop(&message_loop_)));
+  EXPECT_CALL(observer_, OnSourceRemoved(model_.get(), 1))
+      .WillOnce(CheckListSize(model_.get(), 2));
+  EXPECT_CALL(observer_, OnSourceRemoved(model_.get(), 1))
+      .WillOnce(DoAll(CheckListSize(model_.get(), 1),
+                      QuitMessageLoop(&message_loop_)));
 
   list.erase(list.begin(), list.end());
   window_capturer_->SetWindowList(list);
 
   message_loop_.Run();
 }
 
 TEST_F(DesktopMediaListTest, UpdateTitle) {
   CreateWithDefaultCapturers();
   webrtc::WindowCapturer::WindowList list = AddWindowsAndVerify(1, false);
 
-  EXPECT_CALL(observer_, OnSourceNameChanged(1))
-    .WillOnce(QuitMessageLoop(&message_loop_));
+  EXPECT_CALL(observer_, OnSourceNameChanged(model_.get(), 1))
+      .WillOnce(QuitMessageLoop(&message_loop_));
 
   const std::string kTestTitle = "New Title";
 
   list[0].title = kTestTitle;
   window_capturer_->SetWindowList(list);
 
   message_loop_.Run();
 
   EXPECT_EQ(model_->GetSource(1).name, base::UTF8ToUTF16(kTestTitle));
 }
 
 TEST_F(DesktopMediaListTest, UpdateThumbnail) {
   CreateWithDefaultCapturers();
   AddWindowsAndVerify(2, false);
 
-  EXPECT_CALL(observer_, OnSourceThumbnailChanged(1))
-    .WillOnce(QuitMessageLoop(&message_loop_));
+  EXPECT_CALL(observer_, OnSourceThumbnailChanged(model_.get(), 1))
+      .WillOnce(QuitMessageLoop(&message_loop_));
   // Update frame for the window and verify that we get notification about it.
   window_capturer_->SetNextFrameValue(1, 1);
 
   message_loop_.Run();
 }
 
 TEST_F(DesktopMediaListTest, MoveWindow) {
   CreateWithDefaultCapturers();
   webrtc::WindowCapturer::WindowList list = AddWindowsAndVerify(2, false);
 
-  EXPECT_CALL(observer_, OnSourceMoved(2, 1))
-    .WillOnce(DoAll(CheckListSize(model_.get(), 3),
-                    QuitMessageLoop(&message_loop_)));
+  EXPECT_CALL(observer_, OnSourceMoved(model_.get(), 2, 1))
+      .WillOnce(DoAll(CheckListSize(model_.get(), 3),
+                      QuitMessageLoop(&message_loop_)));
 
   // Swap the two windows.
   webrtc::WindowCapturer::Window temp = list[0];
   list[0] = list[1];
   list[1] = temp;
   window_capturer_->SetWindowList(list);
 
   message_loop_.Run();
 }
 
 }  // namespace