Add host-side rate-limiting to desktop resize events.

remoting/host/resizing_host_observer_unittest.cc

 // Copyright (c) 2012 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 <list>
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
+#include "base/memory/ref_counted.h"
+#include "base/message_loop.h"
+#include "base/run_loop.h"
 #include "remoting/host/desktop_resizer.h"
 #include "remoting/host/resizing_host_observer.h"
 #include "remoting/host/screen_resolution.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/skia/include/core/SkSize.h"
 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"
 
 std::ostream& operator<<(std::ostream& os, const SkISize& size) {
   return os << size.width() << "x" << size.height();
 }
@@ skipping 45 matching lines @@
   SkISize initial_size_;
   SkISize current_size_;
   bool exact_size_supported_;
   std::list<SkISize> supported_sizes_;
 
   int set_size_call_count_;
 };
 
 class ResizingHostObserverTest : public testing::Test {
  public:
-  ResizingHostObserverTest() : desktop_resizer_(NULL) {
+  ResizingHostObserverTest()
+      : desktop_resizer_(NULL),
+        now_(base::Time::Now()) {
   }
 
+  // This needs to be public because the derived test-case class needs to
+  // pass it to Bind, which fails if it's protected.
+  base::Time GetTime() {
+    return now_;
+  }
+
+ protected:
   void SetDesktopResizer(scoped_ptr<FakeDesktopResizer> desktop_resizer) {
     CHECK(!desktop_resizer_) << "Call SetDeskopResizer once per test";
     desktop_resizer_ = desktop_resizer.get();
 
     resizing_host_observer_.reset(
         new ResizingHostObserver(desktop_resizer.PassAs<DesktopResizer>()));
+    resizing_host_observer_->SetNowFunctionForTesting(
+        base::Bind(&ResizingHostObserverTest::GetTimeAndIncrement,
+                   base::Unretained(this)));
   }
 
   SkISize GetBestSize(const SkISize& client_size) {
     resizing_host_observer_->SetScreenResolution(ScreenResolution(
         webrtc::DesktopSize(client_size.width(), client_size.height()),
         webrtc::DesktopVector(kDefaultDPI, kDefaultDPI)));
     return desktop_resizer_->GetCurrentSize();
   }
 
   void VerifySizes(const SkISize* client_sizes, const SkISize* expected_sizes,
                    int number_of_sizes) {
     for (int i = 0; i < number_of_sizes; ++i) {
       SkISize best_size = GetBestSize(client_sizes[i]);
       EXPECT_EQ(expected_sizes[i], best_size)
           << "Input size = " << client_sizes[i];
     }
   }
 
- private:
+  base::Time GetTimeAndIncrement() {
+    base::Time result = now_;
+    now_ += base::TimeDelta::FromSeconds(1);
+    return result;
+  }
+
   scoped_ptr<ResizingHostObserver> resizing_host_observer_;
   FakeDesktopResizer* desktop_resizer_;
+  base::Time now_;
 };
 
 // Check that the host is not resized if GetSupportedSizes returns an empty
 // list (even if GetCurrentSize is supported).
 TEST_F(ResizingHostObserverTest, EmptyGetSupportedSizes) {
   SkISize initial = { 640, 480 };
   scoped_ptr<FakeDesktopResizer> desktop_resizer(
       new FakeDesktopResizer(initial, false, NULL, 0));
   SetDesktopResizer(desktop_resizer.Pass());
 
@@ skipping 70 matching lines @@
       new FakeDesktopResizer(SkISize::Make(640, 480), false,
                              supported_sizes, arraysize(supported_sizes));
   SetDesktopResizer(scoped_ptr<FakeDesktopResizer>(desktop_resizer));
 
   SkISize client_sizes[] = { { 640, 640 }, { 1024, 768 }, { 640, 480 } };
   SkISize expected_sizes[] = { { 640, 480 }, { 640, 480 }, { 640, 480 } };
   VerifySizes(client_sizes, expected_sizes, arraysize(client_sizes));
   EXPECT_EQ(desktop_resizer->set_size_call_count(), 0);
 }
 
+// Check that desktop resizes are rate-limited, and that if multiple resize
+// requests are received in the time-out period, the most recent is respected.
+TEST_F(ResizingHostObserverTest, RateLimited) {
+  FakeDesktopResizer* desktop_resizer =
+      new FakeDesktopResizer(SkISize::Make(640, 480), true, NULL, 0);
+  SetDesktopResizer(scoped_ptr<FakeDesktopResizer>(desktop_resizer));
+  resizing_host_observer_->SetNowFunctionForTesting(
+      base::Bind(&ResizingHostObserverTest::GetTime, base::Unretained(this)));
+
+  base::MessageLoop message_loop;
+  base::RunLoop run_loop;
+
+  EXPECT_EQ(GetBestSize(SkISize::Make(100, 100)), SkISize::Make(100, 100));
+  now_ += base::TimeDelta::FromMilliseconds(900);
+  EXPECT_EQ(GetBestSize(SkISize::Make(200, 200)), SkISize::Make(100, 100));
+  now_ += base::TimeDelta::FromMilliseconds(99);
+  EXPECT_EQ(GetBestSize(SkISize::Make(300, 300)), SkISize::Make(100, 100));
+  now_ += base::TimeDelta::FromMilliseconds(1);
+
+  // Due to the kMinimumResizeIntervalMs constant in resizing_host_observer.cc,
+  // We need to wait a total of 1000ms for the final resize to be processed.
+  // Since it was queued 900 + 99 ms after the first, we need to wait an
+  // additional 1ms. However, since RunLoop is not guaranteed to process tasks
+  // with the same due time in FIFO order, wait an additional 1ms for safety.
+  message_loop.PostDelayedTask(
+      FROM_HERE,
+      run_loop.QuitClosure(),
+      base::TimeDelta::FromMilliseconds(2));
+  run_loop.Run();
+
+  // If the QuitClosure fired before the final resize, it's a test failure.
+  EXPECT_EQ(desktop_resizer_->GetCurrentSize(), SkISize::Make(300, 300));
+}
+
 }  // namespace remoting