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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_
#define REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_
#include <string>
#include "base/memory/weak_ptr.h"
#include "base/observer_list.h"
#include "base/sequence_checker.h"
#include "base/time/time.h"
#include "remoting/signaling/iq_sender.h"
#include "remoting/signaling/signal_strategy.h"
#include "remoting/signaling/signaling_address.h"
namespace base {
class SingleThreadTaskRunner;
} // namespace base
namespace remoting {
class FakeSignalStrategy : public SignalStrategy {
public:
using PeerCallback = base::RepeatingCallback<void(
std::unique_ptr<jingle_xmpp::XmlElement> message)>;
// Calls ConnectTo() to connect |peer1| and |peer2|. Both |peer1| and |peer2|
// must belong to the current thread.
static void Connect(FakeSignalStrategy* peer1, FakeSignalStrategy* peer2);
FakeSignalStrategy(const SignalingAddress& address);
FakeSignalStrategy(const FakeSignalStrategy&) = delete;
FakeSignalStrategy& operator=(const FakeSignalStrategy&) = delete;
~FakeSignalStrategy() override;
const std::vector<std::unique_ptr<jingle_xmpp::XmlElement>>&
received_messages() {
return received_messages_;
}
void set_send_delay(base::TimeDelta delay) { send_delay_ = delay; }
void SetError(Error error);
void SetIsSignInError(bool is_sign_in_error);
void SetState(State state);
void SetPeerCallback(const PeerCallback& peer_callback);
// Connects current FakeSignalStrategy to receive messages from |peer|.
void ConnectTo(FakeSignalStrategy* peer);
void SetLocalAddress(const SignalingAddress& address);
// Simulate IQ messages re-ordering by swapping the delivery order of
// next pair of messages.
void SimulateMessageReordering();
// If this is enabled, calling Connect() will transition the signal strategy
// state to CONNECTING instead of CONNECTED, and caller needs to call
// ProceedConnect() to transition to CONNECTED, or Disconnect() to transition
// to DISCONNECTED.
void SimulateTwoStageConnect();
// Called by the |peer_|. Takes ownership of |stanza|.
void OnIncomingMessage(std::unique_ptr<jingle_xmpp::XmlElement> stanza);
void ProceedConnect();
// SignalStrategy interface.
void Connect() override;
void Disconnect() override;
State GetState() const override;
Error GetError() const override;
const SignalingAddress& GetLocalAddress() const override;
void AddListener(Listener* listener) override;
void RemoveListener(Listener* listener) override;
bool SendStanza(std::unique_ptr<jingle_xmpp::XmlElement> stanza) override;
bool SendMessage(const SignalingAddress& destination_address,
const ftl::ChromotingMessage& message) override;
std::string GetNextId() override;
bool IsSignInError() const override;
private:
static void DeliverMessageOnThread(
scoped_refptr<base::SingleThreadTaskRunner> thread,
base::WeakPtr<FakeSignalStrategy> target,
std::unique_ptr<jingle_xmpp::XmlElement> stanza);
void NotifyListeners(std::unique_ptr<jingle_xmpp::XmlElement> stanza);
scoped_refptr<base::SingleThreadTaskRunner> main_thread_;
Error error_ = OK;
bool is_sign_in_error_ = false;
State state_ = CONNECTED;
SignalingAddress address_;
PeerCallback peer_callback_;
base::ObserverList<Listener, true> listeners_;
int last_id_;
base::TimeDelta send_delay_;
bool simulate_reorder_ = false;
bool simulate_two_stage_connect_ = false;
std::unique_ptr<jingle_xmpp::XmlElement> pending_stanza_;
// All received messages, includes thouse still in |pending_messages_|.
std::vector<std::unique_ptr<jingle_xmpp::XmlElement>> received_messages_;
SEQUENCE_CHECKER(sequence_checker_);
base::WeakPtrFactory<FakeSignalStrategy> weak_factory_{this};
};
} // namespace remoting
#endif // REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_
|
