Home | History | Annotate | Download | only in dbus 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "base/bind.h"
      6 #include "base/memory/ref_counted.h"
      7 #include "base/run_loop.h"
      8 #include "dbus/bus.h"
      9 #include "dbus/object_proxy.h"
     10 #include "dbus/test_service.h"
     11 #include "testing/gtest/include/gtest/gtest.h"
     12 
     13 namespace dbus {
     14 namespace {
     15 
     16 class ObjectProxyTest : public testing::Test {
     17  protected:
     18   virtual void SetUp() OVERRIDE {
     19     Bus::Options bus_options;
     20     bus_options.bus_type = Bus::SESSION;
     21     bus_options.connection_type = Bus::PRIVATE;
     22     bus_ = new Bus(bus_options);
     23 
     24     object_proxy_ = bus_->GetObjectProxy(
     25         "org.chromium.TestService", ObjectPath("/org/chromium/TestObject"));
     26   }
     27 
     28   virtual void TearDown() OVERRIDE {
     29     bus_->ShutdownAndBlock();
     30   }
     31 
     32   base::MessageLoopForIO message_loop_;
     33   scoped_refptr<Bus> bus_;
     34   ObjectProxy* object_proxy_;
     35 };
     36 
     37 // Used as a WaitForServiceToBeAvailableCallback.
     38 void OnServiceIsAvailable(scoped_ptr<base::RunLoop>* run_loop,
     39                           bool service_is_available) {
     40   EXPECT_TRUE(service_is_available);
     41   ASSERT_TRUE(*run_loop);
     42   (*run_loop)->Quit();
     43 }
     44 
     45 TEST_F(ObjectProxyTest, WaitForServiceToBeAvailable) {
     46   scoped_ptr<base::RunLoop> run_loop;
     47 
     48   // Callback is not yet called because the service is not available.
     49   object_proxy_->WaitForServiceToBeAvailable(
     50       base::Bind(&OnServiceIsAvailable, &run_loop));
     51   base::RunLoop().RunUntilIdle();
     52 
     53   // Start the service.
     54   TestService::Options options;
     55   TestService test_service(options);
     56   ASSERT_TRUE(test_service.StartService());
     57   ASSERT_TRUE(test_service.WaitUntilServiceIsStarted());
     58   ASSERT_TRUE(test_service.has_ownership());
     59 
     60   // Callback is called beacuse the service became available.
     61   run_loop.reset(new base::RunLoop);
     62   run_loop->Run();
     63 
     64   // Callback is called because the service is already available.
     65   run_loop.reset(new base::RunLoop);
     66   object_proxy_->WaitForServiceToBeAvailable(
     67       base::Bind(&OnServiceIsAvailable, &run_loop));
     68   run_loop->Run();
     69 
     70   // Shut down the service.
     71   test_service.ShutdownAndBlock();
     72   test_service.Stop();
     73 }
     74 
     75 }  // namespace
     76 }  // namespace dbus
     77