Swiften/Network/FakeConnection.h


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

/*
 * Copyright (c) 2010-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <memory>
#include <vector>

#include <boost/optional.hpp>

#include <Swiften/Base/API.h>
#include <Swiften/EventLoop/EventLoop.h>
#include <Swiften/EventLoop/EventOwner.h>
#include <Swiften/Network/Connection.h>
#include <Swiften/Network/HostAddressPort.h>

namespace Swift {
    class SWIFTEN_API FakeConnection :
            public Connection,
            public EventOwner,
            public std::enable_shared_from_this<FakeConnection> {
        public:
            enum State {
                Initial,
                Connecting,
                Connected,
                Disconnected,
                DisconnectedWithError
            };

            FakeConnection(EventLoop* eventLoop);
            virtual ~FakeConnection();

            virtual void listen();
            virtual HostAddressPort getLocalAddress() const {
                return HostAddressPort();
            }
            virtual HostAddressPort getRemoteAddress() 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;
    };
}