ain/src/test/oracles_tests.cpp at master · MarcinSPro/ain · GitHub

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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
// Copyright (c) DeFi Blockchain Developers
// Distributed under the MIT software license, see the accompanying
// file LICENSE or http://www.opensource.org/licenses/mit-license.php.

#include <test/setup_common.h>
#include <string>
#include <boost/test/unit_test.hpp>

#include <masternodes/oracles.h>
#include <rpc/rawtransaction_util.h>
#include <masternodes/masternodes.h>

#include <boost/algorithm/string/join.hpp>
#include <boost/range/adaptor/transformed.hpp>

struct OraclesTestingSetup : public TestingSetup {
    const std::string data1 =
            "\x9c\x52\x4a\xdb\xcf\x56\x11\x12\x2b\x29\x12\x5e\x5d\x35\xd2\xd2"
            "\x22\x81\xaa\xb5\x33\xf0\x08\x32\xd5\x56\xb1\xf9\xea\xe5\x1d\x7d";

    const std::string data2 =
            "\x9c\x52\x4a\xac\xcf\x56\x11\x12\x2b\x29\x12\x5e\x5d\x35\xd2\xd2"
            "\x22\x81\xaa\xb5\x24\xf0\x08\x32\xd5\x56\xb1\xf9\xea\xe5\x1d\x7d";

    const std::string hex1 = "7d1de5eaf9b156d53208f033b5aa8122d2d2355d5e12292b121156cfdb4a529c";
    const std::vector<unsigned char> rawVector1{data1.begin(), data1.end()};
    const std::vector<unsigned char> rawVector2{data2.begin(), data2.end()};
    const std::string address1 = "mhWzxsS5aDfmNY2EpPuM2xQZx7Ju3yjkQ4";

    std::string JoinOracles(const std::vector<COracleId> &oracles) {
        auto list = boost::algorithm::join(
                oracles | boost::adaptors::transformed(
                        [](const COracleId &x) -> std::string {
                            return x.GetHex();
                        }),
                ", ");
        return "[" + list + "]";
    }
};

BOOST_FIXTURE_TEST_SUITE(oracles_tests, OraclesTestingSetup)

    BOOST_AUTO_TEST_CASE(check_oracleid_compare_operator) {
        COracleId id1{rawVector1}, id2{}, id3{rawVector1};
        BOOST_ASSERT_MSG(id1 != id2, "compare unequal failed");
        BOOST_ASSERT_MSG(id1 == id3, "compare equal failed");
    }

    BOOST_AUTO_TEST_CASE(oracleid_serialization_test) {

        CDataStream stream(std::vector<unsigned char>{}, SER_NETWORK, PROTOCOL_VERSION);
        COracleId oracleId{rawVector1};
        stream << oracleId;
        COracleId oracleId1{};
        stream >> oracleId1;
        BOOST_ASSERT_MSG(oracleId == oracleId1, "failed to serialize/deserialize COracleId");
    }

    BOOST_AUTO_TEST_CASE(oracle_serialization_test) {
        COracleId oracleId1{rawVector1};
        COracleId oracleId2{rawVector2};

        BOOST_ASSERT_MSG(oracleId1 != oracleId2, "bad test data");

        std::vector<unsigned char> tmp{'a', 'b', 'c'};
        CScript oracleAddress1{tmp.begin(), tmp.end()};
        uint8_t weightage = 15;
        std::set<CTokenCurrencyPair> availableTokens = {
                {"DFI", "USD"},
                {"TOK", "USD"},
        };
        CAppointOracleMessage msg{oracleAddress1, weightage, availableTokens};
        COracle oracle;
        static_cast<CAppointOracleMessage&>(oracle) = msg;

        CCustomCSView mnview(*pcustomcsview);
        auto res = mnview.AppointOracle(oracleId1, oracle);
        BOOST_ASSERT_MSG(res.ok, res.msg.c_str());

        res = mnview.AppointOracle(oracleId2, oracle);
        BOOST_ASSERT_MSG(res.ok, res.msg.c_str());
    }

    BOOST_AUTO_TEST_CASE(remove_oracle_test) {

        COracleId oracleId1{rawVector1};
        COracleId oracleId2{rawVector2};
        uint8_t weightage = 15;
        std::vector<unsigned char> tmp{'a', 'b', 'c'};
        CScript oracleAddress1{tmp.begin(), tmp.end()};
        std::set<CTokenCurrencyPair> availableTokens = {
                {"DFI", "USD"},
                {"TOK", "USD"},
        };
        CAppointOracleMessage msg{oracleAddress1, weightage, availableTokens};
        COracle oracle1;
        static_cast<CAppointOracleMessage&>(oracle1) = msg;

        CCustomCSView mnview(*pcustomcsview);
        auto res = mnview.AppointOracle(oracleId1, oracle1);
        BOOST_ASSERT_MSG(res.ok, res.msg.c_str());

        res = mnview.AppointOracle(oracleId2, oracle1);
        BOOST_ASSERT_MSG(res.ok, res.msg.c_str());

        auto res1 = mnview.RemoveOracle(oracleId1);
        BOOST_ASSERT_MSG(res1.ok, "failed to remove oracle");

        auto res2 = mnview.RemoveOracle(oracleId2);
        BOOST_ASSERT_MSG(res2.ok, "failed to remove oracle");
    }

    BOOST_AUTO_TEST_CASE(update_oracle_test) {

        COracleId oracleId1{rawVector1};
        COracleId oracleId2{rawVector2};
        uint8_t weightage = 15;
        std::vector<unsigned char> tmp{'a', 'b', 'c'};
        CScript oracleAddress1{tmp.begin(), tmp.end()};
        std::set<CTokenCurrencyPair> availableTokens = {
                {"DFI", "USD"},
                {"TOK", "USD"},
        };
        CAppointOracleMessage msg1{oracleAddress1, weightage, availableTokens};
        COracle oracle1;
        static_cast<CAppointOracleMessage&>(oracle1) = msg1;

        uint8_t weightage2 = weightage + 2;
        availableTokens.insert({"DFI", "EUR"});
        CAppointOracleMessage msg2{oracleAddress1, weightage2, availableTokens};
        COracle oracle2;
        static_cast<CAppointOracleMessage&>(oracle2) = msg2;

        CCustomCSView mnview(*pcustomcsview);
        auto res = mnview.AppointOracle(oracleId1, oracle1);
        BOOST_ASSERT_MSG(res.ok, res.msg.c_str());

        res = mnview.UpdateOracle(oracleId1, std::move(oracle2));
        BOOST_ASSERT_MSG(res.ok, res.msg.c_str());

        auto dataRes = mnview.GetOracleData(oracleId1);
        BOOST_ASSERT_MSG(dataRes.ok, dataRes.msg.c_str());
    }

BOOST_AUTO_TEST_SUITE_END()