Line data Source code
1 : // Copyright (c) 2012-2016 The Bitcoin Core developers
2 : // Copyright (c) 2017-2021 The PIVX Core developers
3 : // Distributed under the MIT/X11 software license, see the accompanying
4 : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
5 :
6 : #include "util/system.h"
7 : #include "test/test_pivx.h"
8 :
9 : #include <string>
10 : #include <vector>
11 :
12 : #include <boost/algorithm/string.hpp>
13 : #include <boost/test/unit_test.hpp>
14 :
15 : BOOST_FIXTURE_TEST_SUITE(getarg_tests, BasicTestingSetup)
16 :
17 26 : static void ResetArgs(const std::string& strArg)
18 : {
19 26 : std::vector<std::string> vecArg;
20 26 : if (strArg.size())
21 48 : boost::split(vecArg, strArg, boost::is_space(), boost::token_compress_on);
22 :
23 : // Insert dummy executable name:
24 26 : vecArg.insert(vecArg.begin(), "testbitcoin");
25 :
26 : // Convert to char*:
27 52 : std::vector<const char*> vecChar;
28 86 : for (std::string& s : vecArg)
29 60 : vecChar.push_back(s.c_str());
30 :
31 26 : gArgs.ParseParameters(vecChar.size(), vecChar.data());
32 26 : }
33 :
34 2 : BOOST_AUTO_TEST_CASE(boolarg)
35 : {
36 1 : ResetArgs("-foo");
37 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", false));
38 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", true));
39 :
40 2 : BOOST_CHECK(!gArgs.GetBoolArg("-fo", false));
41 2 : BOOST_CHECK(gArgs.GetBoolArg("-fo", true));
42 :
43 2 : BOOST_CHECK(!gArgs.GetBoolArg("-fooo", false));
44 2 : BOOST_CHECK(gArgs.GetBoolArg("-fooo", true));
45 :
46 1 : ResetArgs("-foo=0");
47 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
48 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
49 :
50 1 : ResetArgs("-foo=1");
51 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", false));
52 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", true));
53 :
54 : // New 0.6 feature: auto-map -nosomething to !-something:
55 1 : ResetArgs("-nofoo");
56 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
57 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
58 :
59 1 : ResetArgs("-nofoo=1");
60 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
61 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
62 :
63 1 : ResetArgs("-foo -nofoo"); // -nofoo should win
64 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
65 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
66 :
67 1 : ResetArgs("-foo=1 -nofoo=1"); // -nofoo should win
68 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
69 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
70 :
71 1 : ResetArgs("-foo=0 -nofoo=0"); // -nofoo=0 should win
72 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", false));
73 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", true));
74 :
75 : // New 0.6 feature: treat -- same as -:
76 1 : ResetArgs("--foo=1");
77 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", false));
78 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", true));
79 :
80 1 : ResetArgs("--nofoo=1");
81 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
82 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
83 :
84 1 : }
85 :
86 2 : BOOST_AUTO_TEST_CASE(stringarg)
87 : {
88 1 : ResetArgs("");
89 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", ""), "");
90 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", "eleven"), "eleven");
91 :
92 1 : ResetArgs("-foo -bar");
93 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", ""), "");
94 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", "eleven"), "");
95 :
96 1 : ResetArgs("-foo=");
97 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", ""), "");
98 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", "eleven"), "");
99 :
100 1 : ResetArgs("-foo=11");
101 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", ""), "11");
102 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", "eleven"), "11");
103 :
104 1 : ResetArgs("-foo=eleven");
105 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", ""), "eleven");
106 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", "eleven"), "eleven");
107 :
108 1 : }
109 :
110 2 : BOOST_AUTO_TEST_CASE(intarg)
111 : {
112 1 : ResetArgs("");
113 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", 11), 11);
114 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", 0), 0);
115 :
116 1 : ResetArgs("-foo -bar");
117 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", 11), 0);
118 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-bar", 11), 0);
119 :
120 1 : ResetArgs("-foo=11 -bar=12");
121 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", 0), 11);
122 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-bar", 11), 12);
123 :
124 1 : ResetArgs("-foo=NaN -bar=NotANumber");
125 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", 1), 0);
126 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-bar", 11), 0);
127 1 : }
128 :
129 2 : BOOST_AUTO_TEST_CASE(doubledash)
130 : {
131 1 : ResetArgs("--foo");
132 1 : BOOST_CHECK_EQUAL(gArgs.GetBoolArg("-foo", false), true);
133 :
134 1 : ResetArgs("--foo=verbose --bar=1");
135 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-foo", ""), "verbose");
136 1 : BOOST_CHECK_EQUAL(gArgs.GetArg("-bar", 0), 1);
137 1 : }
138 :
139 2 : BOOST_AUTO_TEST_CASE(boolargno)
140 : {
141 1 : ResetArgs("-nofoo");
142 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
143 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
144 :
145 1 : ResetArgs("-nofoo=1");
146 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
147 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
148 :
149 1 : ResetArgs("-nofoo=0");
150 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", true));
151 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", false));
152 :
153 1 : ResetArgs("-foo --nofoo"); // --nofoo should win
154 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", true));
155 2 : BOOST_CHECK(!gArgs.GetBoolArg("-foo", false));
156 :
157 1 : ResetArgs("-nofoo -foo"); // foo always wins:
158 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", true));
159 2 : BOOST_CHECK(gArgs.GetBoolArg("-foo", false));
160 1 : }
161 :
162 : BOOST_AUTO_TEST_SUITE_END()
|