library/cpp/getopt/ut/ygetopt_ut.cpp


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

#include <library/cpp/getopt/ygetopt.h>
 
#include <library/cpp/testing/unittest/registar.h>
 
class TGetOptTest: public TTestBase { 
    UNIT_TEST_SUITE(TGetOptTest); 
    UNIT_TEST(TestGetOpt); 
    UNIT_TEST_EXCEPTION(TestZeroArgC, yexception); 
    UNIT_TEST_SUITE_END(); 
 
public: 
    void TestGetOpt(); 
    void TestZeroArgC(); 
}; 
 
UNIT_TEST_SUITE_REGISTRATION(TGetOptTest); 
 
void TGetOptTest::TestZeroArgC() { 
    TGetOpt opt(0, nullptr, "");
} 
 
void TGetOptTest::TestGetOpt() { 
    const char* argv[] = { 
        "/usr/bin/bash", 
        "-f", 
        "-p", 
        "qwerty123", 
        "-z", 
        "-q", 
        nullptr}; 
 
    TString res;
    const TString format = "qzp:f";
    TGetOpt opt(sizeof(argv) / sizeof(*argv) - 1, argv, format); 
 
    for (TGetOpt::TIterator it = opt.Begin(); it != opt.End(); ++it) { 
        res += it->Key(); 
 
        if (it->HaveArg()) { 
            res += it->Arg(); 
        } 
    } 
 
    UNIT_ASSERT_EQUAL(res, "fpqwerty123zq"); 
}