Add FlushForTesting to InterfacePtr and Binding.

mojo/public/cpp/bindings/tests/interface_ptr_unittest.cc

 // Copyright 2013 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 <stdint.h>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 60 matching lines @@
   }
 
   void Multiply(double value, const base::Closure& closure) {
     calculator_->Multiply(
         value,
         base::Bind(&MathCalculatorUI::Output, base::Unretained(this), closure));
   }
 
   double GetOutput() const { return output_; }
 
+  math::CalculatorPtr& GetInterfacePtr() { return calculator_; }
+
  private:
   void Output(const base::Closure& closure, double output) {
     output_ = output;
     if (!closure.is_null())
       closure.Run();
   }
 
   math::CalculatorPtr calculator_;
   double output_;
   base::Closure closure_;
@@ skipping 637 matching lines @@
   EXPECT_TRUE(FuseInterface(std::move(request), proxy.PassInterface()));
 
   // Ping!
   bool called = false;
   base::RunLoop loop;
   ptr->Ping(base::Bind(&SetFlagAndRunClosure, &called, loop.QuitClosure()));
   loop.Run();
   EXPECT_TRUE(called);
 }
 
+void Fail() {
+  FAIL() << "Unexpected connection error";
+}
+
+TEST_F(InterfacePtrTest, FlushForTesting) {
+  math::CalculatorPtr calc;
+  MathCalculatorImpl calc_impl(GetProxy(&calc));
+  calc.set_connection_error_handler(base::Bind(&Fail));
+
+  MathCalculatorUI calculator_ui(std::move(calc));
+
+  calculator_ui.Add(2.0, base::Bind(&base::DoNothing));
+  calculator_ui.GetInterfacePtr().FlushForTesting();
+  EXPECT_EQ(2.0, calculator_ui.GetOutput());
+
+  calculator_ui.Multiply(5.0, base::Bind(&base::DoNothing));
+  calculator_ui.GetInterfacePtr().FlushForTesting();
+
+  EXPECT_EQ(10.0, calculator_ui.GetOutput());
+}
+
+void SetBool(bool* value) {
+  *value = true;
+}
+
+TEST_F(InterfacePtrTest, FlushForTestingWithClosedPeer) {
+  math::CalculatorPtr calc;
+  GetProxy(&calc);
+  bool called = false;
+  calc.set_connection_error_handler(base::Bind(&SetBool, &called));
+  calc.FlushForTesting();
+  EXPECT_TRUE(called);
+  calc.FlushForTesting();
+}
+
 }  // namespace
 }  // namespace test
 }  // namespace mojo