aboutsummaryrefslogtreecommitdiffstats
path: root/whisperTopic.cpp
blob: 02fccbc9f035c93e47e6ef26c91749a8a3a7f3ff (plain) (blame)
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
/*
    This file is part of cpp-ethereum.

    cpp-ethereum is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    cpp-ethereum is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with cpp-ethereum.  If not, see <http://www.gnu.org/licenses/>.
*/
/** @file whisperTopic.cpp
 * @author Gav Wood <i@gavwood.com>
 * @date 2014
 */
#include <functional>
#include <boost/test/unit_test.hpp>
#include <libp2p/Host.h>
#include <libwhisper/WhisperPeer.h>
#include <libwhisper/WhisperHost.h>
using namespace std;
using namespace dev;
using namespace dev::p2p;
using namespace dev::shh;

BOOST_AUTO_TEST_SUITE(whisper)

BOOST_AUTO_TEST_CASE(topic)
{
    cnote << "Testing Whisper...";
    auto oldLogVerbosity = g_logVerbosity;
    g_logVerbosity = 5;

    Host phOther("Test", NetworkPreferences(30303, "127.0.0.1", true, true));
    auto whOther = phOther.registerCapability(new WhisperHost());
    phOther.start();
    
    bool started = false;
    unsigned result = 0;
    std::thread listener([&]()
    {
        setThreadName("other");
        started = true;

        /// Only interested in odd packets
        auto w = whOther->installWatch(BuildTopicMask("odd"));

        for (int i = 0, last = 0; i < 200 && last < 81; ++i)
        {
            for (auto i: whOther->checkWatch(w))
            {
                Message msg = whOther->envelope(i).open();
                last = RLP(msg.payload()).toInt<unsigned>();
                cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
                result += last;
            }
            this_thread::sleep_for(chrono::milliseconds(50));
        }
    });

    Host ph("Test", NetworkPreferences(30300, "127.0.0.1", true, true));
    auto wh = ph.registerCapability(new WhisperHost());
    ph.start();
    ph.addNode(phOther.id(), "127.0.0.1", 30303, 30303);

    this_thread::sleep_for(chrono::milliseconds(300));
    while (!started)
        this_thread::sleep_for(chrono::milliseconds(25));

    
    KeyPair us = KeyPair::create();
    for (int i = 0; i < 10; ++i)
    {
        wh->post(us.sec(), RLPStream().append(i * i).out(), BuildTopic(i)(i % 2 ? "odd" : "even"));
        this_thread::sleep_for(chrono::milliseconds(250));
    }

    listener.join();
    g_logVerbosity = oldLogVerbosity;

    BOOST_REQUIRE_EQUAL(result, 1 + 9 + 25 + 49 + 81);
}

BOOST_AUTO_TEST_SUITE_END()