Presentation API: reject screen availability observing on Android low ram devices.

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 <vector>
 
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/media/router/media_source.h"
 #include "chrome/browser/media/router/media_source_helper.h"
 #include "chrome/browser/media/router/mock_media_router.h"
@@ skipping 42 matching lines @@
     PresentationServiceDelegateImpl::CreateForWebContents(wc);
     delegate_impl_ = PresentationServiceDelegateImpl::FromWebContents(wc);
     delegate_impl_->SetMediaRouterForTest(&router_);
   }
 
   PresentationServiceDelegateImpl* delegate_impl_;
   MockMediaRouter router_;
 };
 
 TEST_F(PresentationServiceDelegateImplTest, AddScreenAvailabilityListener) {
+  ON_CALL(router_, RegisterMediaSinksObserver(_)).WillByDefault(Return(true));
+
   std::string presentation_url1("http://url1");
   std::string presentation_url2("http://url2");
   MediaSource source1 = MediaSourceForPresentationUrl(presentation_url1);
   MediaSource source2 = MediaSourceForPresentationUrl(presentation_url2);
   MockScreenAvailabilityListener listener1(presentation_url1);
   MockScreenAvailabilityListener listener2(presentation_url2);
   int render_process_id = 10;
   int render_frame_id1 = 1;
   int render_frame_id2 = 2;
 
@@ skipping 14 matching lines @@
       render_process_id, render_frame_id1, &listener1);
   delegate_impl_->RemoveScreenAvailabilityListener(
       render_process_id, render_frame_id2, &listener2);
   EXPECT_FALSE(delegate_impl_->HasScreenAvailabilityListenerForTest(
       render_process_id, render_frame_id1, source1.id()));
   EXPECT_FALSE(delegate_impl_->HasScreenAvailabilityListenerForTest(
       render_process_id, render_frame_id2, source2.id()));
 }
 
 TEST_F(PresentationServiceDelegateImplTest, AddSameListenerTwice) {
+  ON_CALL(router_, RegisterMediaSinksObserver(_)).WillByDefault(Return(true));
+
   std::string presentation_url1("http://url1");
   MediaSource source1(MediaSourceForPresentationUrl(presentation_url1));
   MockScreenAvailabilityListener listener1(presentation_url1);
   int render_process_id = 1;
   int render_frame_id = 0;
 
   EXPECT_CALL(router_, RegisterMediaSinksObserver(_)).Times(1);
   EXPECT_TRUE(delegate_impl_->AddScreenAvailabilityListener(
       render_process_id, render_frame_id, &listener1));
   EXPECT_FALSE(delegate_impl_->AddScreenAvailabilityListener(
@@ skipping 68 matching lines @@
 
   EXPECT_TRUE(Mock::VerifyAndClearExpectations(&observer1));
   // Remove default presentation URL.
   EXPECT_CALL(observer1, OnDefaultMediaSourceChanged(
                              Equals(MediaSource()),
                              GURL("http://www.google.com"))).Times(1);
   delegate_impl_->SetDefaultPresentationUrl(render_process_id, routing_id, "");
 }
 
 TEST_F(PresentationServiceDelegateImplTest, Reset) {
+  ON_CALL(router_, RegisterMediaSinksObserver(_)).WillByDefault(Return(true));
+
   std::string presentation_url1("http://url1");
   MediaSource source = MediaSourceForPresentationUrl(presentation_url1);
   MockScreenAvailabilityListener listener1(presentation_url1);
   int render_process_id = 1;
   int render_frame_id = 0;
 
   EXPECT_TRUE(delegate_impl_->AddScreenAvailabilityListener(
       render_process_id, render_frame_id, &listener1));
   EXPECT_TRUE(delegate_impl_->HasScreenAvailabilityListenerForTest(
       render_process_id, render_frame_id, source.id()));
@@ skipping 14 matching lines @@
   manager->AddObserver(123, 234, &delegate_observer1);
   manager->AddObserver(345, 456, &delegate_observer2);
 
   // Removes |delegate_observer2|.
   manager->RemoveObserver(345, 456);
 
   EXPECT_CALL(delegate_observer1, OnDelegateDestroyed()).Times(1);
   manager.reset();
 }
 
+TEST_F(PresentationServiceDelegateImplTest, SinksObserverCantRegister) {
+  ON_CALL(router_, RegisterMediaSinksObserver(_)).WillByDefault(Return(false));
+
+  const std::string presentation_url("http://url1");
+  MockScreenAvailabilityListener listener(presentation_url);
+  const int render_process_id = 10;
+  const int render_frame_id = 1;
+
+  EXPECT_CALL(router_, RegisterMediaSinksObserver(_)).Times(1);
+  EXPECT_FALSE(delegate_impl_->AddScreenAvailabilityListener(
+      render_process_id, render_frame_id, &listener));
+}
+
 }  // namespace media_router