Index: chrome/browser/media/router/media_router_mojo_impl_unittest.cc |
diff --git a/chrome/browser/media/router/media_router_mojo_impl_unittest.cc b/chrome/browser/media/router/media_router_mojo_impl_unittest.cc |
index 09e2572a905779199d7bcc6be16031709c99426e..87ad51158da4710b433d59723189fc4eec830117 100644 |
--- a/chrome/browser/media/router/media_router_mojo_impl_unittest.cc |
+++ b/chrome/browser/media/router/media_router_mojo_impl_unittest.cc |
@@ -60,6 +60,11 @@ class RouteResponseCallbackHandler { |
void(MediaRoute* route, const std::string& error_text)); |
}; |
+class SendMessageCallbackHandler { |
+ public: |
+ MOCK_METHOD1(Invoke, void(bool)); |
+}; |
+ |
template <typename T> |
void StoreAndRun(T* result, const base::Closure& closure, const T& result_val) { |
*result = result_val; |
@@ -273,10 +278,21 @@ TEST_F(MediaRouterMojoImplTest, RegisterAndUnregisterMediaRoutesObserver) { |
ProcessEventLoop(); |
} |
-TEST_F(MediaRouterMojoImplTest, PostMessage) { |
- EXPECT_CALL(mock_mojo_media_router_service_, |
- PostMessage(mojo::String(kRouteId), mojo::String(kMessage))); |
- router()->PostMessage(kRouteId, kMessage); |
+TEST_F(MediaRouterMojoImplTest, SendRouteMessage) { |
+ EXPECT_CALL( |
+ mock_mojo_media_router_service_, |
+ SendRouteMessage(mojo::String(kRouteId), mojo::String(kMessage), _)) |
+ .WillOnce(Invoke( |
+ [](const MediaRoute::Id& route_id, const std::string& message, |
+ const interfaces::MediaRouter::SendRouteMessageCallback& cb) { |
+ cb.Run(true); |
+ })); |
+ |
+ SendMessageCallbackHandler handler; |
+ EXPECT_CALL(handler, Invoke(true)); |
+ router()->SendRouteMessage(kRouteId, kMessage, |
+ base::Bind(&SendMessageCallbackHandler::Invoke, |
+ base::Unretained(&handler))); |
ProcessEventLoop(); |
} |