[Presentation API] 1-UA presentation support + presenter APIs.

content/browser/presentation/presentation_service_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 "content/browser/presentation/presentation_service_impl.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <memory>
@@ skipping 121 matching lines @@
                    const SendMessageCallback& send_message_cb) override {
     SendMessageRawPtr(render_process_id, render_frame_id, session,
                       message_request.release(), send_message_cb);
   }
   MOCK_METHOD4(ListenForConnectionStateChange,
                void(int render_process_id,
                     int render_frame_id,
                     const content::PresentationSessionInfo& connection,
                     const content::PresentationConnectionStateChangedCallback&
                         state_changed_cb));
+  MOCK_METHOD3(
+      GetReceiverConnections,
+      std::vector<content::PresentationSessionInfo>(
+          int render_process_id,
+          int render_frame_id,
+          const content::PresentationSessionStartedCallback& callback));
 
   void set_screen_availability_listening_supported(bool value) {
     screen_availability_listening_supported_ = value;
   }
 
  private:
   bool screen_availability_listening_supported_ = true;
 };
 
 class MockPresentationServiceClient
@@ skipping 31 matching lines @@
   }
   MOCK_METHOD0(MessagesReceived, void());
 
   void OnDefaultSessionStarted(
       blink::mojom::PresentationSessionInfoPtr session_info) override {
     OnDefaultSessionStarted(*session_info);
   }
   MOCK_METHOD1(OnDefaultSessionStarted,
                void(const blink::mojom::PresentationSessionInfo& session_info));
 
+  void OnReceiverConnectionAvailable(
+      blink::mojom::PresentationSessionInfoPtr connection) {
+    OnReceiverConnectionAvailable(*connection);
+  }
+  MOCK_METHOD1(OnReceiverConnectionAvailable,
+               void(const blink::mojom::PresentationSessionInfo& connection));
+
   mojo::Array<blink::mojom::SessionMessagePtr> messages_received_;
 };
 
 class PresentationServiceImplTest : public RenderViewHostImplTestHarness {
  public:
   PresentationServiceImplTest() {}
 
   void SetUp() override {
     RenderViewHostImplTestHarness::SetUp();
 
@@ skipping 48 matching lines @@
   void SaveQuitClosureAndRunLoop() {
     base::RunLoop run_loop;
     run_loop_quit_closure_ = run_loop.QuitClosure();
     run_loop.Run();
     run_loop_quit_closure_.Reset();
   }
 
   void SimulateScreenAvailabilityChangeAndWait(
       const std::string& url, bool available) {
     auto listener_it = service_impl_->screen_availability_listeners_.find(url);
+    ASSERT_TRUE(listener_it !=
+                service_impl_->screen_availability_listeners_.end());
     ASSERT_TRUE(listener_it->second);
 
     base::RunLoop run_loop;
     EXPECT_CALL(mock_client_,
                 OnScreenAvailabilityUpdated(mojo::String(url), available))
       .WillOnce(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit));
     listener_it->second->OnScreenAvailabilityChanged(available);
     run_loop.Run();
   }
 
@@ skipping 37 matching lines @@
     }
   }
 
   void ExpectSendMessageMojoCallback(bool success) {
     EXPECT_TRUE(success);
     EXPECT_FALSE(service_impl_->send_message_callback_);
     if (!run_loop_quit_closure_.is_null())
       run_loop_quit_closure_.Run();
   }
 
+  void ExpectReceiverConnections(
+      size_t expected_num_connections,
+      mojo::Array<blink::mojom::PresentationSessionInfoPtr> connections) {
+    EXPECT_EQ(expected_num_connections, connections.size());
+    if (!run_loop_quit_closure_.is_null())
+      run_loop_quit_closure_.Run();
+  }
+
   void RunListenForSessionMessages(const std::string& text_msg,
                                    const std::vector<uint8_t>& binary_data,
                                    bool pass_ownership) {
     mojo::Array<blink::mojom::SessionMessagePtr> expected_msgs(2);
     expected_msgs[0] = blink::mojom::SessionMessage::New();
     expected_msgs[0]->type = blink::mojom::PresentationMessageType::TEXT;
     expected_msgs[0]->message = text_msg;
     expected_msgs[1] = blink::mojom::SessionMessage::New();
     expected_msgs[1]->type =
         blink::mojom::PresentationMessageType::ARRAY_BUFFER;
@@ skipping 521 matching lines @@
 TEST_F(PresentationServiceImplTest, ScreenAvailabilityNotSupported) {
   mock_delegate_.set_screen_availability_listening_supported(false);
   base::RunLoop run_loop;
   EXPECT_CALL(mock_client_,
               OnScreenAvailabilityNotSupported(Eq(kPresentationUrl)))
       .WillOnce(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit));
   ListenForScreenAvailabilityAndWait(kPresentationUrl, false);
   run_loop.Run();
 }
 
+TEST_F(PresentationServiceImplTest, GetReceiverConnections) {
+  std::vector<content::PresentationSessionInfo> connections;
+  content::PresentationSessionInfo connection("", kPresentationId);
+  connections.push_back(connection);
+  EXPECT_CALL(mock_delegate_, GetReceiverConnections(_, _, _))
+      .WillOnce(Return(connections));
+  service_ptr_->GetReceiverConnections(
+      base::Bind(&PresentationServiceImplTest::ExpectReceiverConnections,
+                 base::Unretained(this), 1));
+  SaveQuitClosureAndRunLoop();
+}
+
 }  // namespace content