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..03bfa041276ea2e98e944c90d26afcfc88d38bc2 |
| --- /dev/null |
| +++ b/chrome/browser/media/router/offscreen_presentation_manager_unittest.cc |
| @@ -0,0 +1,290 @@ |
| +// 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(SetOffscreenPresentationClient, |
| + void(OffscreenPresentationClient*)); |
| + MOCK_METHOD1(RemoveOffscreenPresentationClient, |
| + void(OffscreenPresentationClient*)); |
| +}; |
| + |
| +class OffscreenPresentationManagerTest : public ::testing::Test { |
| + public: |
| + OffscreenPresentationManager* manager() { return &manager_; } |
| + |
| + void verifyPresentationsSize(size_t expected); |
|
imcheng
2016/09/28 07:28:36
nit: VerifyNumPresentations
zhaobin
2016/09/29 17:20:44
Done.
|
| + void verifyControllerSize(size_t expected, |
|
imcheng
2016/09/28 07:28:36
nit: VerifyNumControllers
zhaobin
2016/09/29 17:20:44
Done.
|
| + const OffscreenPresentation* presentation); |
| + |
| + private: |
| + OffscreenPresentationManager manager_; |
| +}; |
| + |
| +class MockReceiverConnectionAvailableCallback { |
| + public: |
| + MOCK_METHOD2(OnReceiverConnectionAvailable, |
| + void(const content::PresentationSessionInfo&, |
| + content::OffscreenPresentationClient*)); |
| +}; |
| + |
| +void OffscreenPresentationManagerTest::verifyPresentationsSize( |
|
imcheng
2016/09/28 07:28:36
These helper methods can be inlined.
zhaobin
2016/09/29 17:20:43
Done.
|
| + 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) { |
|
imcheng
2016/09/28 07:28:36
nit: I would just name the test RegisterController
zhaobin
2016/09/29 17:20:43
Done.
|
| + MockOffscreenPresentationClient controller; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + verifyPresentationsSize(1); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, RegisterOneReceiverDoNothing) { |
|
imcheng
2016/09/28 07:28:36
RegisterReceiver
zhaobin
2016/09/29 17:20:44
Done.
|
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, UnregisterControllerNoCrash) { |
|
imcheng
2016/09/28 07:28:36
UnregisterNonexistentController
zhaobin
2016/09/29 17:20:44
Done.
|
| + MockOffscreenPresentationClient controller; |
|
imcheng
2016/09/28 07:28:36
Unused?
zhaobin
2016/09/29 17:20:44
Done.
|
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 1); |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverNoCrash) { |
|
imcheng
2016/09/28 07:28:36
UnregisterNonexistentReceiver
zhaobin
2016/09/29 17:20:43
Done.
|
| + manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, |
| + RegisterControllerCreatePresentationWithTwoControllers) { |
|
imcheng
2016/09/28 07:28:36
RegisterMultipleControllersSamePresentation ?
zhaobin
2016/09/29 17:20:43
Done.
|
| + MockOffscreenPresentationClient controller1; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller1); |
| + MockOffscreenPresentationClient controller2; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 2, |
| + &controller2); |
| + verifyPresentationsSize(1); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, |
| + RegisterControllerCreateTwoPresentations) { |
|
imcheng
2016/09/28 07:28:36
RegisterMultipleControllersDifferentPresentations
zhaobin
2016/09/29 17:20:44
Done.
|
| + MockOffscreenPresentationClient controller1; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller1); |
| + MockOffscreenPresentationClient controller2; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId2, 1, |
| + &controller2); |
| + verifyPresentationsSize(2); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, |
| + RegisterControllerAndReceiverFireOnConnectionAvailable) { |
|
imcheng
2016/09/28 07:28:36
RegisterControllerThenReceiverInvokesCallback ?
zhaobin
2016/09/29 17:20:44
Done.
|
| + MockOffscreenPresentationClient controller; |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + |
| + verifyPresentationsSize(0); |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
|
imcheng
2016/09/28 07:28:36
Move this EXPECT to just before RegisterOffscreenP
zhaobin
2016/09/29 17:20:44
Done.
|
| + |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, RegisterUnregisterController) { |
|
imcheng
2016/09/28 07:28:36
Combine this case with RegisterController
zhaobin
2016/09/29 17:20:43
Done.
|
| + MockOffscreenPresentationClient controller; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 1); |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverOnly) { |
|
imcheng
2016/09/28 07:28:36
UnregisterReceiverFromConnectedPresentation
zhaobin
2016/09/29 17:20:44
Done.
|
| + MockOffscreenPresentationClient controller; |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
| + |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, UnregisterControllerOnly) { |
|
imcheng
2016/09/28 07:28:36
UnregisterControllerFromConnectedPresentation
zhaobin
2016/09/29 17:20:44
Done.
|
| + MockOffscreenPresentationClient controller; |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + |
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 1); |
| + verifyPresentationsSize(1); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverController) { |
| + MockOffscreenPresentationClient controller; |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + |
| + manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 1); |
| + |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, UnregisterControllerReceiver) { |
| + MockOffscreenPresentationClient controller; |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + |
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 1); |
| + manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); |
| + |
| + verifyPresentationsSize(0); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, |
| + RegisterTwoControllersThenReceiverFireOnConnectionAvailableTwice) { |
| + MockOffscreenPresentationClient controller; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + MockOffscreenPresentationClient controller2; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 2, |
| + &controller2); |
| + |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + EXPECT_CALL(receiver_callback, |
| + OnReceiverConnectionAvailable(_, &controller2)); |
| + |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, |
| + RegisterControllerReceiverConontrollerFireOnConnectionAvailableTwice) { |
| + MockOffscreenPresentationClient controller; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + |
| + MockOffscreenPresentationClient controller2; |
| + EXPECT_CALL(receiver_callback, |
| + OnReceiverConnectionAvailable(_, &controller2)); |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 2, |
| + &controller2); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, |
| + OnePresentationUnregisterFirstController) { |
| + MockOffscreenPresentationClient controller; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + MockOffscreenPresentationClient controller2; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 2, |
| + &controller2); |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, _)).Times(2); |
| + |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 1); |
| + manager()->UnregisterOffscreenPresentationController(kPresentationId, 2); |
| + |
| + verifyPresentationsSize(1); |
| +} |
| + |
| +TEST_F(OffscreenPresentationManagerTest, TwoPresentations) { |
| + MockOffscreenPresentationClient controller; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId, 1, |
| + &controller); |
| + MockReceiverConnectionAvailableCallback receiver_callback; |
| + EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller)); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback))); |
| + MockOffscreenPresentationClient controller2; |
| + manager()->RegisterOffscreenPresentationController(kPresentationId2, 2, |
| + &controller2); |
| + MockReceiverConnectionAvailableCallback receiver_callback2; |
| + EXPECT_CALL(receiver_callback2, |
| + OnReceiverConnectionAvailable(_, &controller2)); |
| + manager()->RegisterOffscreenPresentationReceiver( |
| + kPresentationId2, base::Bind(&MockReceiverConnectionAvailableCallback:: |
| + OnReceiverConnectionAvailable, |
| + base::Unretained(&receiver_callback2))); |
| + |
| + verifyPresentationsSize(2); |
| + |
| + manager()->UnregisterOffscreenPresentationReceiver(kPresentationId2); |
| + verifyPresentationsSize(1); |
| +} |
| + |
| +} // namespace media_router |