Chromium Code Reviews| 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 |