Index: components/proximity_auth/fake_remote_device_life_cycle.h |
diff --git a/components/proximity_auth/fake_remote_device_life_cycle.h b/components/proximity_auth/fake_remote_device_life_cycle.h |
index 548099b1f7195d91ea330d886f9cc4dd7fbdf306..4307b3c36831ef7cc5ec4961b3517ee086c8033f 100644 |
--- a/components/proximity_auth/fake_remote_device_life_cycle.h |
+++ b/components/proximity_auth/fake_remote_device_life_cycle.h |
@@ -7,6 +7,7 @@ |
#include "base/macros.h" |
#include "base/observer_list.h" |
+#include "components/cryptauth/fake_connection.h" |
#include "components/cryptauth/remote_device.h" |
#include "components/proximity_auth/remote_device_life_cycle.h" |
@@ -20,6 +21,7 @@ class FakeRemoteDeviceLifeCycle : public RemoteDeviceLifeCycle { |
// RemoteDeviceLifeCycle: |
void Start() override; |
cryptauth::RemoteDevice GetRemoteDevice() const override; |
+ cryptauth::Connection* GetConnection() const override; |
State GetState() const override; |
Messenger* GetMessenger() override; |
void AddObserver(Observer* observer) override; |
@@ -30,6 +32,10 @@ class FakeRemoteDeviceLifeCycle : public RemoteDeviceLifeCycle { |
void set_messenger(Messenger* messenger) { messenger_ = messenger; } |
+ void set_connection(cryptauth::Connection* connection) { |
+ connection_ = connection; |
+ } |
+ |
bool started() { return started_; } |
base::ObserverList<Observer>& observers() { return observers_; } |
@@ -43,6 +49,8 @@ class FakeRemoteDeviceLifeCycle : public RemoteDeviceLifeCycle { |
State state_; |
+ cryptauth::Connection* connection_; |
+ |
Messenger* messenger_; |
DISALLOW_COPY_AND_ASSIGN(FakeRemoteDeviceLifeCycle); |