[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 "base/strings/stringprintf.h"
 #include "chrome/browser/media/router/media_source.h"
 #include "chrome/browser/media/router/media_source_helper.h"
@@ skipping 79 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 {
+ public:
+  MOCK_METHOD1(OnPresentationSessionStartedCallback,
+               void(const content::PresentationSessionInfo&));
+};
+
 class PresentationServiceDelegateImplTest
     : public ChromeRenderViewHostTestHarness {
  public:
   PresentationServiceDelegateImplTest()
       : delegate_impl_(nullptr),
         presentation_url1_(kPresentationUrl1),
         presentation_url2_(kPresentationUrl2) {}
 
   void SetUp() override {
     ChromeRenderViewHostTestHarness::SetUp();
@@ skipping 389 matching lines @@
   ASSERT_TRUE(main_frame);
   int render_process_id = main_frame->GetProcess()->GetID();
   int render_frame_id = main_frame->GetRoutingID();
 
   EXPECT_CALL(router_, RegisterMediaSinksObserver(_)).WillOnce(Return(false));
   EXPECT_CALL(listener, OnScreenAvailabilityNotSupported());
   EXPECT_FALSE(delegate_impl_->AddScreenAvailabilityListener(
       render_process_id, render_frame_id, &listener));
 }
 
-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();
   content::PresentationSessionInfo session_info(
       GURL("http://www.example.com/presentation.html"), "presentation_id",
       true);
 
+  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::GetInstance()->SetTestingFactory(
       profile(), &BuildMockOffscreenPresentationManager);
   MockOffscreenPresentationManager* mock_offscreen_manager =
       static_cast<MockOffscreenPresentationManager*>(
           OffscreenPresentationManagerFactory::GetOrCreateForBrowserContext(
               profile()));
   EXPECT_CALL(*mock_offscreen_manager,
               RegisterOffscreenPresentationController(_, _, _));
 
   content::PresentationConnectionPtr connection_ptr;
-  delegate_impl_->ConnectToOffscreenPresentation(render_process_id,
-                                                 render_frame_id, session_info,
-                                                 std::move(connection_ptr));
+  delegate_impl_->ConnectToPresentation(render_process_id, render_frame_id,
+                                        session_info,
+                                        std::move(connection_ptr));
+
+  EXPECT_CALL(router_, DetachRoute("route_id"));
+  delegate_impl_->Reset(render_process_id, render_frame_id);
 }
 
 }  // namespace media_router