Add hooks for flushing input from BeginFrame dispatch on Aura

content/browser/renderer_host/begin_frame_observer_proxy_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 <algorithm>
 #include <list>
 
 #include "base/basictypes.h"
 #include "base/compiler_specific.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 76 matching lines @@
   // SendBeginFrame is not called when compositor is reset.
   begin_frame_observer.ResetCompositor();
   cc::BeginFrameArgs args3 =
     cc::CreateBeginFrameArgsForTesting(BEGINFRAME_FROM_HERE,
                                        base::TimeTicks::FromInternalValue(99));
   EXPECT_CALL(client, SendBeginFrame(_)).Times(0);
   compositor()->SendBeginFramesToChildren(args3);
   Mock::VerifyAndClearExpectations(&client);
 }
 
+TEST_F(BeginFrameObserverProxyTest, SingleBeginFrameScheduling) {
+  MockBeginFrameObserverProxyClient client;
+  BeginFrameObserverProxy begin_frame_observer(&client);
+  begin_frame_observer.SetCompositor(compositor());
+  begin_frame_observer.SetNeedsOneBeginFrame();
+
+  // SendBeginFrame is called when new |args| is delivered.
+  cc::BeginFrameArgs args = cc::CreateBeginFrameArgsForTesting(
+      BEGINFRAME_FROM_HERE, base::TimeTicks::FromInternalValue(33));
+  EXPECT_CALL(client, SendBeginFrame(args));
+  compositor()->SendBeginFramesToChildren(args);
+  Mock::VerifyAndClearExpectations(&client);
+
+  // SendBeginFrame is not called when new |args2| is delivered, as the
+  // single frame request was already satisified.
+  cc::BeginFrameArgs args2 = cc::CreateBeginFrameArgsForTesting(
+      BEGINFRAME_FROM_HERE, base::TimeTicks::FromInternalValue(66));
+  EXPECT_CALL(client, SendBeginFrame(_)).Times(0);
+  compositor()->SendBeginFramesToChildren(args2);
+  Mock::VerifyAndClearExpectations(&client);
+
+  begin_frame_observer.ResetCompositor();
+}
+
 }  // namespace content