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

chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc

Index: chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc
diff --git a/chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc b/chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc
index 2a607b576c78dce58015a75b50f0156d871a580c..b7c822421f81a497c2b5a4e7f0f1fdc32fd6b194 100644
--- a/chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc
+++ b/chrome/browser/media/router/presentation_service_delegate_impl_unittest.cc
@@ -97,6 +97,12 @@ std::unique_ptr<KeyedService> BuildMockOffscreenPresentationManager(
   return base::MakeUnique<MockOffscreenPresentationManager>();
 }
 
+class MockPresentationSessionStartedCallback {
+ public:
+  MOCK_METHOD1(OnPresentationSessionStartedCallback,
+               void(const content::PresentationSessionInfo&));
+};
+
 class PresentationServiceDelegateImplTest
     : public ChromeRenderViewHostTestHarness {
  public:
@@ -506,7 +512,7 @@ TEST_F(PresentationServiceDelegateImplTest, SinksObserverCantRegister) {
       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();
@@ -515,6 +521,19 @@ TEST_F(PresentationServiceDelegateImplTest, ConnectToOffscreenPresentation) {
       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 =
@@ -525,9 +544,12 @@ TEST_F(PresentationServiceDelegateImplTest, ConnectToOffscreenPresentation) {
               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