blob: a98a6164308cd468a07a335b897b96f5d3ea31fa [file] [log] [blame]
// Copyright 2017 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 DEVICE_BLUETOOTH_TEST_FAKE_PERIPHERAL_H_
#define DEVICE_BLUETOOTH_TEST_FAKE_PERIPHERAL_H_
#include <string>
#include "base/compiler_specific.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "base/optional.h"
#include "device/bluetooth/bluetooth_device.h"
#include "device/bluetooth/test/fake_central.h"
#include "device/bluetooth/test/fake_remote_gatt_service.h"
namespace bluetooth {
// Implements device::BluetoothDevice. Meant to be used by FakeCentral
// to keep track of the peripheral's state and attributes.
//
// Not intended for direct use by clients. See README.md.
class FakePeripheral : public device::BluetoothDevice {
public:
FakePeripheral(FakeCentral* fake_central, const std::string& address);
~FakePeripheral() override;
// Changes the name of the device.
void SetName(base::Optional<std::string> name);
// Set it to indicate if the system has connected to the Peripheral outside of
// the Bluetooth interface e.g. the user connected to the device through
// system settings.
void SetSystemConnected(bool gatt_connected);
// Updates the peripheral's UUIDs that are returned by
// BluetoothDevice::GetUUIDs().
void SetServiceUUIDs(UUIDSet service_uuids);
// If |code| is kHCISuccess calls a pending success callback for
// CreateGattConnection. Otherwise calls a pending error callback
// with the ConnectErrorCode corresponding to |code|.
void SetNextGATTConnectionResponse(uint16_t code);
// If |code| is kHCISuccess, calls
// BluetoothAdapter::Observer::GattServicesDiscovered otherwise
// sets IsGattDiscoveryComplete to false. Both of these happen
// after IsGattDiscoveryComplete is called.
void SetNextGATTDiscoveryResponse(uint16_t code);
// Returns true if there are no pending responses for this peripheral or any
// of its GATT services.
bool AllResponsesConsumed();
// Simulates a GATT disconnection from the peripheral.
void SimulateGATTDisconnection();
// Adds a fake primary service with |service_uuid| to this peripheral.
// Returns the service's Id.
std::string AddFakeService(const device::BluetoothUUID& service_uuid);
// BluetoothDevice overrides:
uint32_t GetBluetoothClass() const override;
#if defined(OS_CHROMEOS) || defined(OS_LINUX)
device::BluetoothTransport GetType() const override;
#endif
std::string GetIdentifier() const override;
std::string GetAddress() const override;
VendorIDSource GetVendorIDSource() const override;
uint16_t GetVendorID() const override;
uint16_t GetProductID() const override;
uint16_t GetDeviceID() const override;
uint16_t GetAppearance() const override;
base::Optional<std::string> GetName() const override;
base::string16 GetNameForDisplay() const override;
bool IsPaired() const override;
bool IsConnected() const override;
bool IsGattConnected() const override;
bool IsConnectable() const override;
bool IsConnecting() const override;
UUIDSet GetUUIDs() const override;
bool ExpectingPinCode() const override;
bool ExpectingPasskey() const override;
bool ExpectingConfirmation() const override;
void GetConnectionInfo(const ConnectionInfoCallback& callback) override;
void SetConnectionLatency(ConnectionLatency connection_latency,
const base::Closure& callback,
const ErrorCallback& error_callback) override;
void Connect(PairingDelegate* pairing_delegate,
const base::Closure& callback,
const ConnectErrorCallback& error_callback) override;
void SetPinCode(const std::string& pincode) override;
void SetPasskey(uint32_t passkey) override;
void ConfirmPairing() override;
void RejectPairing() override;
void CancelPairing() override;
void Disconnect(const base::Closure& callback,
const ErrorCallback& error_callback) override;
void Forget(const base::Closure& callback,
const ErrorCallback& error_callback) override;
void ConnectToService(
const device::BluetoothUUID& uuid,
const ConnectToServiceCallback& callback,
const ConnectToServiceErrorCallback& error_callback) override;
void ConnectToServiceInsecurely(
const device::BluetoothUUID& uuid,
const ConnectToServiceCallback& callback,
const ConnectToServiceErrorCallback& error_callback) override;
void CreateGattConnection(
const GattConnectionCallback& callback,
const ConnectErrorCallback& error_callback) override;
bool IsGattServicesDiscoveryComplete() const override;
protected:
void CreateGattConnectionImpl() override;
void DisconnectGatt() override;
private:
void DispatchConnectionResponse();
void DispatchDiscoveryResponse();
const std::string address_;
base::Optional<std::string> name_;
UUIDSet service_uuids_;
// True when the system has connected to the device outside of the Bluetooth
// interface e.g. the user connected to the device through system settings.
bool system_connected_;
// True when this Bluetooth interface is connected to the device.
bool gatt_connected_;
// Keeps track of the last Id used to create a fake service. Incremented
// every time a new fake service is added.
size_t last_service_id_;
// Used to simulate a GATT Discovery procedure.
// Mutable because IsGattServicesDiscoveryComplete needs to set this but
// is const.
mutable bool pending_gatt_discovery_;
// Used to decide which callback should be called when
// CreateGattConnection is called.
base::Optional<uint16_t> next_connection_response_;
// Used to decide if the GattServicesDiscovered method is called.
base::Optional<uint16_t> next_discovery_response_;
// Mutable because IsGattServicesDiscoveryComplete needs to post a task but
// is const.
mutable base::WeakPtrFactory<FakePeripheral> weak_ptr_factory_;
DISALLOW_COPY_AND_ASSIGN(FakePeripheral);
};
} // namespace bluetooth
#endif // DEVICE_BLUETOOTH_TEST_FAKE_PERIPHERAL_H_