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
|
#include "common.hh"
#include <cstdarg>
#include <iostream>
#include "args.hh"
using namespace stuff;
namespace {
bool compare(const std::string& in, const std::vector<std::string>& out,
va_list args) {
for (const auto& arg : out) {
const char* str = va_arg(args, const char*);
if (!str) {
std::cerr << "expected less arguments: " << in
<< ", first extra: " << arg << std::endl;
return false;
}
if (arg.compare(str) != 0) {
std::cerr << "expected " << str << " in " << in
<< ", got: " << arg << std::endl;
return false;
}
}
const char* str = va_arg(args, const char*);
if (str) {
std::cerr << "expected more arguments: " << in
<< ", missing: " << str << std::endl;
return false;
}
return true;
}
bool test(const char* in, ...) {
std::vector<std::string> out;
if (!Args::parse(in, &out, false)) {
std::cerr << "expected success: " << in << std::endl;
return false;
}
va_list args;
va_start(args, in);
bool ret = compare(in, out, args);
va_end(args);
return ret;
}
bool test_nice(const char* in, ...) {
std::vector<std::string> out;
if (!Args::parse(in, &out, true)) {
std::cerr << "expected success: " << in << std::endl;
return false;
}
va_list args;
va_start(args, in);
bool ret = compare(in, out, args);
va_end(args);
return ret;
}
bool test_fail(const std::string& in) {
std::vector<std::string> out;
if (Args::parse(in, &out, false)) {
std::cerr << "expected fail: " << in << std::endl;
return false;
}
return true;
}
} // namespace
int main() {
unsigned int ok = 0, tot = 0;
tot++; if (test("", NULL)) ok++;
tot++; if (test("foo", "foo", NULL)) ok++;
tot++; if (test("hello world", "hello", "world", NULL)) ok++;
tot++; if (test("\"hello world\"", "hello world", NULL)) ok++;
tot++; if (test("'hello world'", "hello world", NULL)) ok++;
tot++; if (test("a' b'", "a b", NULL)) ok++;
tot++; if (test("a'b'ba", "abba", NULL)) ok++;
tot++; if (test("a'b'b '' a", "abb", "", "a", NULL)) ok++;
tot++; if (test("'\"'", "\"", NULL)) ok++;
tot++; if (test("'\\\"'", "\"", NULL)) ok++;
tot++; if (test("'\\\\'", "\\", NULL)) ok++;
tot++; if (test("'\\''", "'", NULL)) ok++;
tot++; if (test("'a\\'b'", "a'b", NULL)) ok++;
tot++; if (test("a'\\''b", "a'b", NULL)) ok++;
tot++; if (test_nice("a'b", "a'b", NULL)) ok++;
tot++; if (test_nice("a'b foo", "a'b", "foo", NULL)) ok++;
tot++; if (test_nice("'\\", "'\\", NULL)) ok++;
tot++; if (test_nice("'\\'", "'\\'", NULL)) ok++;
tot++; if (test_fail("'")) ok++;
tot++; if (test_fail("'''")) ok++;
tot++; if (test_fail("\"")) ok++;
tot++; if (test_fail("'\\")) ok++;
tot++; if (test_fail("'\\'")) ok++;
std::cout << "OK " << ok << "/" << tot << std::endl;
return ok == tot ? EXIT_SUCCESS : EXIT_FAILURE;
}
|
