blob: cae66db9d908b40c8d54185e465eba234ac055f3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
// Copyright 2015 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 COMPONENTS_CRYPTAUTH_FAKE_CONNECTION_H_
#define COMPONENTS_CRYPTAUTH_FAKE_CONNECTION_H_
#include "base/macros.h"
#include "components/cryptauth/connection.h"
namespace cryptauth {
class ConnectionObserver;
// A fake implementation of Connection to use in tests.
class FakeConnection : public Connection {
public:
FakeConnection(const RemoteDevice& remote_device);
FakeConnection(const RemoteDevice& remote_device, bool should_auto_connect);
~FakeConnection() override;
// Connection:
void Connect() override;
void Disconnect() override;
void AddObserver(ConnectionObserver* observer) override;
void RemoveObserver(ConnectionObserver* observer) override;
// Completes a connection attempt which was originally started via a call to
// |Connect()|. If |success| is true, the connection's status shifts to
// |CONNECTED|; otherwise, the status shifts to |DISCONNECTED|. Note that this
// function should only be called when |should_auto_connect| is false.
void CompleteInProgressConnection(bool success);
// Completes the current send operation with success |success|.
void FinishSendingMessageWithSuccess(bool success);
// Simulates receiving a wire message with the given |payload|, bypassing the
// container WireMessage format.
void ReceiveMessage(const std::string& feature, const std::string& payload);
// Returns the current message in progress of being sent.
WireMessage* current_message() { return current_message_.get(); }
std::vector<ConnectionObserver*>& observers() {
return observers_;
}
using Connection::SetStatus;
private:
// Connection:
void SendMessageImpl(std::unique_ptr<WireMessage> message) override;
std::unique_ptr<WireMessage> DeserializeWireMessage(
bool* is_incomplete_message) override;
// The message currently being sent. Only set between a call to
// SendMessageImpl() and FinishSendingMessageWithSuccess().
std::unique_ptr<WireMessage> current_message_;
// The feature and payload that should be returned when
// DeserializeWireMessage() is called.
std::string pending_feature_;
std::string pending_payload_;
std::vector<ConnectionObserver*> observers_;
const bool should_auto_connect_;
DISALLOW_COPY_AND_ASSIGN(FakeConnection);
};
} // namespace cryptauth
#endif // COMPONENTS_CRYPTAUTH_FAKE_CONNECTION_H_
|
