Index: chrome/browser/media/router/offscreen_presentation_manager_unittest.cc |
diff --git a/chrome/browser/media/router/offscreen_presentation_manager_unittest.cc b/chrome/browser/media/router/offscreen_presentation_manager_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..0da1f86f2e51cc2d399aef9a7ae3e7e789bfc5f6 |
--- /dev/null |
+++ b/chrome/browser/media/router/offscreen_presentation_manager_unittest.cc |
@@ -0,0 +1,308 @@ |
+// Copyright 2016 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 <vector> |
+ |
+#include "base/bind.h" |
+#include "chrome/browser/media/router/offscreen_presentation_manager.h" |
+#include "chrome/browser/media/router/test_helper.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+using testing::_; |
+ |
+namespace media_router { |
+ |
+namespace { |
+const char kPresentationId[] = "presentationId"; |
+const char kPresentationId2[] = "presentationId2"; |
+} // namespace |
+ |
+class MockOffscreenPresentationClient |
+ : public content::OffscreenPresentationClient { |
+ public: |
+ MOCK_METHOD1(OnReceiverConnectionAvailable, |
+ void(const content::PresentationSessionInfo&)); |
+ MOCK_METHOD1(AddOffscreenPresentationObserver, |
+ void(OffscreenPresentationClient*)); |
+ MOCK_METHOD1(RemoveOffscreenPresentationObserver, |
+ void(OffscreenPresentationClient*)); |
+}; |
+ |
+class OffscreenPresentationManagerTest : public ::testing::Test { |
+ public: |
+ OffscreenPresentationManager* manager() { return &manager_; } |
+ |
+ void verifyPresentationsSize(size_t expected); |
+ void verifyControllerSize(size_t expected, |
+ const OffscreenPresentation* presentation); |
+ |
+ private: |
+ OffscreenPresentationManager manager_; |
+}; |
+ |
+void OffscreenPresentationManagerTest::verifyPresentationsSize( |
imcheng
2016/09/19 23:01:17
These don't need to be their own functions since t
zhaobin
2016/09/23 17:18:18
Create these functions to access private variable
|
+ size_t expected) { |
+ EXPECT_EQ(expected, manager_.offscreen_presentations_.size()); |
+} |
+ |
+void OffscreenPresentationManagerTest::verifyControllerSize( |
+ size_t expected, |
+ const OffscreenPresentation* presentation) { |
+ EXPECT_EQ(expected, presentation->controllers_.size()); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, |
+ RegisterControllerCreateNewPresentation) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ verifyPresentationsSize(1); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, RegisterOneReceiverDoNothing) { |
+ MockOffscreenPresentationClient receiver; |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, UnregisterControllerNoCrash) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverNoCrash) { |
+ manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, |
+ RegisterControllerCreatePresentationWithTwoControllers) { |
+ MockOffscreenPresentationClient controller1; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller1); |
+ MockOffscreenPresentationClient controller2; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller2); |
+ verifyPresentationsSize(1); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, |
+ RegisterControllerCreateTwoPresentations) { |
+ MockOffscreenPresentationClient controller1; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller1); |
+ MockOffscreenPresentationClient controller2; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId2, |
+ &controller2); |
+ verifyPresentationsSize(2); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, |
+ RegisterControllerAndReceiverFireOnConnectionAvailable) { |
+ MockOffscreenPresentationClient controller; |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); |
+ |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, RegisterUnregisterController) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverOnly) { |
+ MockOffscreenPresentationClient controller; |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); |
+ |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ |
+ EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)); |
+ manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
+ |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, UnregisterControllerOnly) { |
+ MockOffscreenPresentationClient controller; |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); |
+ |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ |
+ EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)); |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, UnregisterControllerReceiver) { |
+ MockOffscreenPresentationClient controller; |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); |
+ |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ |
+ EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)) |
+ .Times(1); |
+ EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)) |
+ .Times(1); |
+ manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverController) { |
+ MockOffscreenPresentationClient controller; |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); |
+ |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ |
+ EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)) |
+ .Times(1); |
+ EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)) |
+ .Times(1); |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
+ |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, |
+ RegisterTwoControllersThenReceiverFireOnConnectionAvailableTwice) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ MockOffscreenPresentationClient controller2; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller2); |
+ |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller2)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(2); |
+ |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+} |
+ |
+TEST_F( |
+ OffscreenPresentationManagerTest, |
+ RegisterControllerThenReceiverThenConontrollerFireOnConnectionAvailableTwice) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(2); |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ MockOffscreenPresentationClient controller2; |
+ EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller2)); |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller2); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, |
+ OnePresentationUnregisterFirstController) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ MockOffscreenPresentationClient controller2; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller2); |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller2)); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(2); |
+ |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ |
+ EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(controller2, RemoveOffscreenPresentationObserver(&receiver)); |
+ EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(_)).Times(2); |
+ |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId, |
+ &controller2); |
+ |
+ verifyPresentationsSize(0); |
+} |
+ |
+TEST_F(OffscreenPresentationManagerTest, TwoPresentations) { |
+ MockOffscreenPresentationClient controller; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId, |
+ &controller); |
+ MockOffscreenPresentationClient receiver; |
+ EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)).Times(1); |
+ EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)).Times(1); |
+ EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(1); |
+ |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); |
+ |
+ MockOffscreenPresentationClient controller2; |
+ manager()->RegisterOffscreenPresentationController(kPresentationId2, |
+ &controller2); |
+ MockOffscreenPresentationClient receiver2; |
+ EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver2)) |
+ .Times(1); |
+ EXPECT_CALL(receiver2, AddOffscreenPresentationObserver(&controller2)) |
+ .Times(1); |
+ EXPECT_CALL(receiver2, OnReceiverConnectionAvailable(_)).Times(1); |
+ |
+ manager()->RegisterOffscreenPresentationReceiver(kPresentationId2, |
+ &receiver2); |
+ |
+ verifyPresentationsSize(2); |
+ |
+ EXPECT_CALL(receiver2, RemoveOffscreenPresentationObserver(&controller2)) |
+ .Times(1); |
+ EXPECT_CALL(controller2, RemoveOffscreenPresentationObserver(&receiver2)) |
+ .Times(1); |
+ manager()->UnregisterOffscreenPresentationController(kPresentationId2, |
+ &controller2); |
+ |
+ verifyPresentationsSize(1); |
+} |
+ |
+} // namespace media_router |