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..47446b7c06570df7013e9a8225bf13ca85808674 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: |
@@ -515,6 +521,18 @@ 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); |
+ 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 = |
@@ -528,6 +546,9 @@ TEST_F(PresentationServiceDelegateImplTest, ConnectToOffscreenPresentation) { |
delegate_impl_->ConnectToOffscreenPresentation(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 |