Supporting callback for Disconnected signal.

dbus/end_to_end_async_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 <algorithm>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 14 matching lines @@
 // See comments in ObjectProxy::RunResponseCallback() for why the number was
 // chosen.
 const int kHugePayloadSize = 64 << 20;  // 64 MB
 
 }  // namespace
 
 // The end-to-end test exercises the asynchronous APIs in ObjectProxy and
 // ExportedObject.
 class EndToEndAsyncTest : public testing::Test {
  public:
-  EndToEndAsyncTest() {
-  }
+  EndToEndAsyncTest() : on_disconnected_call_count_(0) {}
 
   virtual void SetUp() {
     // Make the main thread not to allow IO.
     base::ThreadRestrictions::SetIOAllowed(false);
 
     // Start the D-Bus thread.
     dbus_thread_.reset(new base::Thread("D-Bus Thread"));
     base::Thread::Options thread_options;
     thread_options.message_loop_type = MessageLoop::TYPE_IO;
     ASSERT_TRUE(dbus_thread_->StartWithOptions(thread_options));
@@ skipping 71 matching lines @@
     test_service_->ShutdownAndBlock();
 
     // Reset to the default.
     base::ThreadRestrictions::SetIOAllowed(true);
 
     // Stopping a thread is considered an IO operation, so do this after
     // allowing IO.
     test_service_->Stop();
   }
 
+  // Callback function which is called when the connection with dbus-daemon is
+  // disconnected.
+  void OnDisconnected() {
+    message_loop_.Quit();
+    ++on_disconnected_call_count_;
+  }
+
  protected:
   // Replaces the bus with a broken one.
   void SetUpBrokenBus() {
     // Shut down the existing bus.
     bus_->ShutdownOnDBusThreadAndBlock();
 
     // Create new bus with invalid address.
     const char kInvalidAddress[] = "";
     dbus::Bus::Options bus_options;
     bus_options.bus_type = dbus::Bus::CUSTOM_ADDRESS;
@@ skipping 115 matching lines @@
   std::vector<std::string> error_names_;
   scoped_ptr<base::Thread> dbus_thread_;
   scoped_refptr<dbus::Bus> bus_;
   dbus::ObjectProxy* object_proxy_;
   dbus::ObjectProxy* root_object_proxy_;
   scoped_ptr<dbus::TestService> test_service_;
   // Text message from "Test" signal.
   std::string test_signal_string_;
   // Text message from "Test" signal delivered to root.
   std::string root_test_signal_string_;
+  int on_disconnected_call_count_;
 };
 
 TEST_F(EndToEndAsyncTest, Echo) {
   const char* kHello = "hello";
 
   // Create the method call.
   dbus::MethodCall method_call("org.chromium.TestInterface", "Echo");
   dbus::MessageWriter writer(&method_call);
   writer.AppendString(kHello);
 
@@ skipping 292 matching lines @@
 TEST_F(EndToEndAsyncTest, TestHugeSignal) {
   const std::string kHugeMessage(kHugePayloadSize, 'o');
 
   // Send the huge signal from the exported object.
   test_service_->SendTestSignal(kHugeMessage);
   // This caused a DCHECK failure before. Ensure that the issue is fixed.
   WaitForTestSignal();
   ASSERT_EQ(kHugeMessage, test_signal_string_);
 }
 
+TEST_F(EndToEndAsyncTest, DisconnectedSignal) {
+  bus_->SetDisconnectedCallback(
+      base::Bind(&EndToEndAsyncTest::OnDisconnected,
+                 base::Unretained(this)));
+  bus_->PostTaskToDBusThread(FROM_HERE,
+                             base::Bind(&dbus::Bus::CloseConnection,
+                                        base::Unretained(bus_.get())));
+  // OnDisconnected callback quits message loop.
+  message_loop_.Run();
+  EXPECT_EQ(1, on_disconnected_call_count_);
+}
+
 class SignalReplacementTest : public EndToEndAsyncTest {
  public:
   SignalReplacementTest() {
   }
 
   virtual void SetUp() {
     // Set up base class.
     EndToEndAsyncTest::SetUp();
 
     // Reconnect the root object proxy's signal handler to a new handler
@@ skipping 35 matching lines @@
   const char kMessage[] = "hello, world";
   // Send the test signal from the exported object.
   test_service_->SendTestSignal(kMessage);
   // Receive the signal with the object proxy.
   WaitForTestSignal();
   // Verify the string WAS NOT received by the original handler.
   ASSERT_TRUE(test_signal_string_.empty());
   // Verify the signal WAS received by the replacement handler.
   ASSERT_EQ(kMessage, replacement_test_signal_string_);
 }