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
|
/*
* Copyright (c) 2010 Remko Tronçon
* Licensed under the GNU General Public License v3.
* See Documentation/Licenses/GPLv3.txt for more information.
*/
#pragma once
#include <boost/optional.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <vector>
#include <Swiften/Base/API.h>
#include <Swiften/Network/Connection.h>
#include <Swiften/Network/HostAddressPort.h>
#include <Swiften/EventLoop/EventOwner.h>
#include <Swiften/EventLoop/EventLoop.h>
namespace Swift {
class SWIFTEN_API FakeConnection :
public Connection,
public EventOwner,
public boost::enable_shared_from_this<FakeConnection> {
public:
enum State {
Initial,
Connecting,
Connected,
Disconnected,
DisconnectedWithError
};
FakeConnection(EventLoop* eventLoop);
~FakeConnection();
virtual void listen();
virtual HostAddressPort getLocalAddress() const {
return HostAddressPort();
}
void setError(const Error& e);
virtual void connect(const HostAddressPort& address);
virtual void disconnect();
virtual void write(const SafeByteArray& data) {
dataWritten.push_back(data);
}
void setDelayConnect() {
delayConnect = true;
}
EventLoop* eventLoop;
boost::optional<HostAddressPort> connectedTo;
std::vector<SafeByteArray> dataWritten;
boost::optional<Error> error;
State state;
bool delayConnect;
};
}
|