// 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.

#ifndef CHROMEOS_DBUS_MOCK_DBUS_THREAD_MANAGER_H_
#define CHROMEOS_DBUS_MOCK_DBUS_THREAD_MANAGER_H_

#include <string>

#include "chromeos/dbus/dbus_thread_manager.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace dbus {

class Bus;

}  // namespace dbus

namespace chromeos {

class  MockBluetoothAdapterClient;
class  MockBluetoothDeviceClient;
class  MockBluetoothInputClient;
class  MockBluetoothManagerClient;
class  MockBluetoothNodeClient;
class  MockCashewClient;
class  MockCrosDisksClient;
class  MockCryptohomeClient;
class  MockDebugDaemonClient;
class  MockFlimflamDeviceClient;
class  MockFlimflamIPConfigClient;
class  MockFlimflamManagerClient;
class  MockFlimflamNetworkClient;
class  MockFlimflamProfileClient;
class  MockFlimflamServiceClient;
class  MockGsmSMSClient;
class  MockIBusClient;
class  MockIBusInputContextClient;
class  MockImageBurnerClient;
class  MockIntrospectableClient;
class  MockPowerManagerClient;
class  MockSessionManagerClient;
class  MockSpeechSynthesizerClient;
class  MockUpdateEngineClient;

// This class provides a mock DBusThreadManager with mock clients
// installed. You can customize the behaviors of mock clients with
// mock_foo_client() functions.
class MockDBusThreadManager : public DBusThreadManager {
 public:
  MockDBusThreadManager();
  virtual ~MockDBusThreadManager();

  MOCK_METHOD1(InitIBusBus, void(const std::string& ibus_address));
  MOCK_METHOD0(GetSystemBus, dbus::Bus*(void));
  MOCK_METHOD0(GetIBusBus, dbus::Bus*(void));
  MOCK_METHOD0(GetBluetoothAdapterClient, BluetoothAdapterClient*(void));
  MOCK_METHOD0(GetBluetoothDeviceClient, BluetoothDeviceClient*(void));
  MOCK_METHOD0(GetBluetoothInputClient, BluetoothInputClient*(void));
  MOCK_METHOD0(GetBluetoothManagerClient, BluetoothManagerClient*(void));
  MOCK_METHOD0(GetBluetoothNodeClient, BluetoothNodeClient*(void));
  MOCK_METHOD0(GetCashewClient, CashewClient*(void));
  MOCK_METHOD0(GetCrosDisksClient, CrosDisksClient*(void));
  MOCK_METHOD0(GetCryptohomeClient, CryptohomeClient*(void));
  MOCK_METHOD0(GetDebugDaemonClient, DebugDaemonClient*(void));
  MOCK_METHOD0(GetFlimflamDeviceClient, FlimflamDeviceClient*(void));
  MOCK_METHOD0(GetFlimflamIPConfigClient, FlimflamIPConfigClient*(void));
  MOCK_METHOD0(GetFlimflamManagerClient, FlimflamManagerClient*(void));
  MOCK_METHOD0(GetFlimflamNetworkClient, FlimflamNetworkClient*(void));
  MOCK_METHOD0(GetFlimflamProfileClient, FlimflamProfileClient*(void));
  MOCK_METHOD0(GetFlimflamServiceClient, FlimflamServiceClient*(void));
  MOCK_METHOD0(GetGsmSMSClient, GsmSMSClient*(void));
  MOCK_METHOD0(GetImageBurnerClient, ImageBurnerClient*(void));
  MOCK_METHOD0(GetIntrospectableClient, IntrospectableClient*(void));
  MOCK_METHOD0(GetPowerManagerClient, PowerManagerClient*(void));
  MOCK_METHOD0(GetSessionManagerClient, SessionManagerClient*(void));
  MOCK_METHOD0(GetSpeechSynthesizerClient, SpeechSynthesizerClient*(void));
  MOCK_METHOD0(GetUpdateEngineClient, UpdateEngineClient*(void));
  MOCK_METHOD0(GetIBusClient, IBusClient*(void));
  MOCK_METHOD0(GetIBusInputContextClient, IBusInputContextClient*(void));

  MockBluetoothAdapterClient* mock_bluetooth_adapter_client() {
    return mock_bluetooth_adapter_client_.get();
  }
  MockBluetoothDeviceClient* mock_bluetooth_device_client() {
    return mock_bluetooth_device_client_.get();
  }
  MockBluetoothInputClient* mock_bluetooth_input_client() {
    return mock_bluetooth_input_client_.get();
  }
  MockBluetoothManagerClient* mock_bluetooth_manager_client() {
    return mock_bluetooth_manager_client_.get();
  }
  MockBluetoothNodeClient* mock_bluetooth_node_client() {
    return mock_bluetooth_node_client_.get();
  }
  MockCashewClient* mock_cashew_client() {
    return mock_cashew_client_.get();
  }
  MockCrosDisksClient* mock_cros_disks_client() {
    return mock_cros_disks_client_.get();
  }
  MockCryptohomeClient* mock_cryptohome_client() {
    return mock_cryptohome_client_.get();
  }
  MockDebugDaemonClient* mock_debugdaemon_client() {
    return mock_debugdaemon_client_.get();
  }
  MockFlimflamDeviceClient* mock_flimflam_device_client() {
    return mock_flimflam_device_client_.get();
  }
  MockFlimflamIPConfigClient* mock_flimflam_ipconfig_client() {
    return mock_flimflam_ipconfig_client_.get();
  }
  MockFlimflamManagerClient* mock_flimflam_manager_client() {
    return mock_flimflam_manager_client_.get();
  }
  MockFlimflamNetworkClient* mock_flimflam_network_client() {
    return mock_flimflam_network_client_.get();
  }
  MockFlimflamProfileClient* mock_flimflam_profile_client() {
    return mock_flimflam_profile_client_.get();
  }
  MockFlimflamServiceClient* mock_flimflam_service_client() {
    return mock_flimflam_service_client_.get();
  }
  MockGsmSMSClient* mock_gsm_sms_client() {
    return mock_gsm_sms_client_.get();
  }
  MockImageBurnerClient* mock_image_burner_client() {
    return mock_image_burner_client_.get();
  }
  MockIntrospectableClient* mock_introspectable_client() {
    return mock_introspectable_client_.get();
  }
  MockPowerManagerClient* mock_power_manager_client() {
    return mock_power_manager_client_.get();
  }
  MockSessionManagerClient* mock_session_manager_client() {
    return mock_session_manager_client_.get();
  }
  MockSpeechSynthesizerClient* mock_speech_synthesizer_client() {
    return mock_speech_synthesizer_client_.get();
  }
  MockUpdateEngineClient* mock_update_engine_client() {
    return mock_update_engine_client_.get();
  }
  MockIBusClient* mock_ibus_client() {
    return mock_ibus_client_.get();
  }
  MockIBusInputContextClient* mock_ibus_input_context_client() {
    return mock_ibus_input_context_client_.get();
  }

 private:
  scoped_ptr<MockBluetoothAdapterClient> mock_bluetooth_adapter_client_;
  scoped_ptr<MockBluetoothDeviceClient> mock_bluetooth_device_client_;
  scoped_ptr<MockBluetoothInputClient> mock_bluetooth_input_client_;
  scoped_ptr<MockBluetoothManagerClient> mock_bluetooth_manager_client_;
  scoped_ptr<MockBluetoothNodeClient> mock_bluetooth_node_client_;
  scoped_ptr<MockCashewClient> mock_cashew_client_;
  scoped_ptr<MockCrosDisksClient> mock_cros_disks_client_;
  scoped_ptr<MockCryptohomeClient> mock_cryptohome_client_;
  scoped_ptr<MockDebugDaemonClient> mock_debugdaemon_client_;
  scoped_ptr<MockFlimflamDeviceClient> mock_flimflam_device_client_;
  scoped_ptr<MockFlimflamIPConfigClient> mock_flimflam_ipconfig_client_;
  scoped_ptr<MockFlimflamManagerClient> mock_flimflam_manager_client_;
  scoped_ptr<MockFlimflamNetworkClient> mock_flimflam_network_client_;
  scoped_ptr<MockFlimflamProfileClient> mock_flimflam_profile_client_;
  scoped_ptr<MockFlimflamServiceClient> mock_flimflam_service_client_;
  scoped_ptr<MockGsmSMSClient> mock_gsm_sms_client_;
  scoped_ptr<MockImageBurnerClient> mock_image_burner_client_;
  scoped_ptr<MockIntrospectableClient> mock_introspectable_client_;
  scoped_ptr<MockPowerManagerClient> mock_power_manager_client_;
  scoped_ptr<MockSessionManagerClient> mock_session_manager_client_;
  scoped_ptr<MockSpeechSynthesizerClient> mock_speech_synthesizer_client_;
  scoped_ptr<MockUpdateEngineClient> mock_update_engine_client_;
  scoped_ptr<MockIBusClient> mock_ibus_client_;
  scoped_ptr<MockIBusInputContextClient> mock_ibus_input_context_client_;

  DISALLOW_COPY_AND_ASSIGN(MockDBusThreadManager);
};

}  // namespace chromeos

#endif  // CHROMEOS_DBUS_MOCK_DBUS_THREAD_MANAGER_H_