// 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 "base/bind.h"
#include "base/memory/ref_counted.h"
#include "base/run_loop.h"
#include "dbus/bus.h"
#include "dbus/object_proxy.h"
#include "dbus/test_service.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace dbus {
namespace {

class ObjectProxyTest : public testing::Test {
 protected:
  void SetUp() override {
    Bus::Options bus_options;
    bus_options.bus_type = Bus::SESSION;
    bus_options.connection_type = Bus::PRIVATE;
    bus_ = new Bus(bus_options);
  }

  void TearDown() override { bus_->ShutdownAndBlock(); }

  base::MessageLoopForIO message_loop_;
  scoped_refptr<Bus> bus_;
};

// Used as a WaitForServiceToBeAvailableCallback.
void OnServiceIsAvailable(std::unique_ptr<base::RunLoop>* run_loop,
                          bool service_is_available) {
  EXPECT_TRUE(service_is_available);
  ASSERT_TRUE(*run_loop);
  (*run_loop)->Quit();
}

TEST_F(ObjectProxyTest, WaitForServiceToBeAvailable) {
  std::unique_ptr<base::RunLoop> run_loop;

  TestService::Options options;
  TestService test_service(options);

  // Callback is not yet called because the service is not available.
  ObjectProxy* object_proxy = bus_->GetObjectProxy(
      test_service.service_name(), ObjectPath("/org/chromium/TestObject"));
  object_proxy->WaitForServiceToBeAvailable(
      base::Bind(&OnServiceIsAvailable, &run_loop));
  base::RunLoop().RunUntilIdle();

  // Start the service.
  ASSERT_TRUE(test_service.StartService());
  ASSERT_TRUE(test_service.WaitUntilServiceIsStarted());
  ASSERT_TRUE(test_service.has_ownership());

  // Callback is called beacuse the service became available.
  run_loop.reset(new base::RunLoop);
  run_loop->Run();

  // Callback is called because the service is already available.
  run_loop.reset(new base::RunLoop);
  object_proxy->WaitForServiceToBeAvailable(
      base::Bind(&OnServiceIsAvailable, &run_loop));
  run_loop->Run();

  // Shut down the service.
  test_service.ShutdownAndBlock();
  test_service.Stop();
}

}  // namespace
}  // namespace dbus