[Presentation API] (5th) (1-UA) integrate controller and receiver side for 1-UA messaging

chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc

 // Copyright 2015 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/router/presentation_service_delegate_impl.h"
 
 #include "base/memory/ptr_util.h"
 #include "chrome/browser/media/router/media_source.h"
 #include "chrome/browser/media/router/media_source_helper.h"
 #include "chrome/browser/media/router/mock_media_router.h"
@@ skipping 84 matching lines @@
                         receiver_callback));
   MOCK_METHOD1(OnOffscreenPresentationReceiverTerminated,
                void(const std::string& presentation_id));
 };
 
 std::unique_ptr<KeyedService> BuildMockOffscreenPresentationManager(
     content::BrowserContext* context) {
   return base::MakeUnique<MockOffscreenPresentationManager>();
 }
 
+class MockPresentationSessionStartedCallback {

[imcheng, 2017/01/31 01:53:25]

Use MockCallback

[zhaobin, 2017/01/31 18:44:16]

Done.

+ public:
+  MOCK_METHOD1(OnPresentationSessionStartedCallback,
+               void(const content::PresentationSessionInfo&));
+};
+
 class PresentationServiceDelegateImplTest
     : public ChromeRenderViewHostTestHarness {
  public:
   PresentationServiceDelegateImplTest()
       : delegate_impl_(nullptr),
         presentation_url1_(kPresentationUrl1),
         presentation_url2_(kPresentationUrl2),
         source1_(MediaSourceForPresentationUrl(presentation_url1_)),
         source2_(MediaSourceForPresentationUrl(presentation_url2_)),
         listener1_(presentation_url1_),
@@ skipping 363 matching lines @@
   manager.reset();
 }
 
 TEST_F(PresentationServiceDelegateImplTest, SinksObserverCantRegister) {
   EXPECT_CALL(router_, RegisterMediaSinksObserver(_)).WillOnce(Return(false));
   EXPECT_CALL(listener1_, OnScreenAvailabilityNotSupported());
   EXPECT_FALSE(delegate_impl_->AddScreenAvailabilityListener(
       main_frame_process_id_, main_frame_routing_id_, &listener1_));
 }
 
-TEST_F(PresentationServiceDelegateImplTest, ConnectToOffscreenPresentation) {
+TEST_F(PresentationServiceDelegateImplTest, ConnectToPresentation) {
   content::RenderFrameHost* main_frame = GetWebContents()->GetMainFrame();
   ASSERT_TRUE(main_frame);
   int render_process_id = main_frame->GetProcess()->GetID();
   int render_frame_id = main_frame->GetRoutingID();
   std::string presentation_id = "presentation_id";
   GURL presentation_url = GURL("http://www.example.com/presentation.html");
   content::PresentationSessionInfo session_info(presentation_url,
                                                 presentation_id);
 
+  MockPresentationSessionStartedCallback mock_callback;
+  EXPECT_CALL(mock_callback, OnPresentationSessionStartedCallback(_));
+  MediaRoute media_route(
+      "route_id", MediaSourceForPresentationUrl(session_info.presentation_url),
+      "mediaSinkId", "", true, "", true);
+  media_route.set_offscreen_presentation(true);
+  delegate_impl_->OnStartSessionSucceeded(
+      render_process_id, render_frame_id,
+      base::Bind(&MockPresentationSessionStartedCallback::
+                     OnPresentationSessionStartedCallback,
+                 base::Unretained(&mock_callback)),
+      session_info, media_route);
+
   OffscreenPresentationManagerFactory::GetInstanceForTest()->SetTestingFactory(
       profile(), &BuildMockOffscreenPresentationManager);
   MockOffscreenPresentationManager* mock_offscreen_manager =
       static_cast<MockOffscreenPresentationManager*>(
           OffscreenPresentationManagerFactory::GetOrCreateForBrowserContext(
               profile()));
   EXPECT_CALL(*mock_offscreen_manager,
               RegisterOffscreenPresentationController(
                   presentation_id, presentation_url,
                   RenderFrameHostId(render_process_id, render_frame_id)));
 
   content::PresentationConnectionPtr connection_ptr;
   content::PresentationConnectionRequest connection_request;
-  delegate_impl_->ConnectToOffscreenPresentation(
-      render_process_id, render_frame_id, session_info,
-      std::move(connection_ptr), std::move(connection_request));
+  delegate_impl_->ConnectToPresentation(render_process_id, render_frame_id,
+                                        session_info, std::move(connection_ptr),
+                                        std::move(connection_request));
+
+  EXPECT_CALL(*mock_offscreen_manager,
+              UnregisterOffscreenPresentationController(
+                  presentation_id,
+                  RenderFrameHostId(render_process_id, render_frame_id)));
+  EXPECT_CALL(router_, DetachRoute("route_id"));
+  delegate_impl_->Reset(render_process_id, render_frame_id);
 }
 
 #if !defined(OS_ANDROID)
 TEST_F(PresentationServiceDelegateImplTest, AutoJoinRequest) {
   GURL frame_url(kFrameUrl);
   std::string origin(url::Origin(frame_url).Serialize());
   content::WebContentsTester::For(GetWebContents())
       ->NavigateAndCommit(frame_url);
 
   MockCreatePresentationConnnectionCallbacks mock_create_connection_callbacks;
@@ skipping 92 matching lines @@
       base::Bind(&MockCreatePresentationConnnectionCallbacks::
                      OnCreateConnectionSuccess,
                  base::Unretained(&mock_create_connection_callbacks)),
       base::Bind(
           &MockCreatePresentationConnnectionCallbacks::OnCreateConnectionError,
           base::Unretained(&mock_create_connection_callbacks)));
 }
 #endif  // !defined(OS_ANDROID)
 
 }  // namespace media_router