#include "cgiparam.h" #include Y_UNIT_TEST_SUITE(TCgiParametersTest) { Y_UNIT_TEST(TestScan1) { TCgiParameters C; C.Scan("aaa=b%62b&ccc=ddd&ag0="); UNIT_ASSERT_EQUAL(C.Get("aaa") == "bbb", true); UNIT_ASSERT_EQUAL(C.NumOfValues("ag0") == 1, true); UNIT_ASSERT(C.Has("ccc", "ddd")); UNIT_ASSERT(C.Has("ag0", "")); UNIT_ASSERT(!C.Has("a", "bbb")); UNIT_ASSERT(!C.Has("aaa", "bb")); UNIT_ASSERT(C.Has("ccc")); UNIT_ASSERT(!C.Has("zzzzzz")); } Y_UNIT_TEST(TestQuick) { TQuickCgiParam C("aaa=b%62b&ccc=ddd&ag0="); UNIT_ASSERT_EQUAL(C.Get("aaa") == "bbb", true); UNIT_ASSERT(C.Has("ccc", "ddd")); UNIT_ASSERT(C.Has("ag0", "")); UNIT_ASSERT(!C.Has("a", "bbb")); UNIT_ASSERT(!C.Has("aaa", "bb")); UNIT_ASSERT(C.Has("ccc")); UNIT_ASSERT(!C.Has("zzzzzz")); TQuickCgiParam D = std::move(C); UNIT_ASSERT(D.Has("aaa")); TQuickCgiParam E(""); UNIT_ASSERT(!E.Has("aaa")); C = std::move(E); UNIT_ASSERT(!C.Has("aaa")); } Y_UNIT_TEST(TestScan2) { const TString parsee("=000&aaa=bbb&ag0=&ccc=ddd"); TCgiParameters c; c.Scan(parsee); UNIT_ASSERT_VALUES_EQUAL(c.Print(), parsee); } Y_UNIT_TEST(TestScan3) { const TString parsee("aaa=bbb&ag0=&ccc=ddd"); TCgiParameters c; c.Scan(parsee); c.InsertUnescaped("d", "x=x"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), parsee + "&d=x\%3Dx"); } Y_UNIT_TEST(TestScanAddAll1) { TCgiParameters c; c.ScanAddAll("qw"); UNIT_ASSERT_VALUES_EQUAL(c.size(), 1u); UNIT_ASSERT(c.Get("qw").empty()); } Y_UNIT_TEST(TestScanAddAll2) { TCgiParameters c; c.ScanAddAll("qw&"); UNIT_ASSERT_VALUES_EQUAL(c.size(), 1u); UNIT_ASSERT(c.Get("qw").empty()); } Y_UNIT_TEST(TestScanAddAll3) { TCgiParameters c; c.ScanAddAll("qw=1&x"); UNIT_ASSERT_VALUES_EQUAL(c.size(), 2u); UNIT_ASSERT_VALUES_EQUAL(c.Get("qw"), "1"); UNIT_ASSERT(c.Get("x").empty()); } Y_UNIT_TEST(TestScanAddAll4) { TCgiParameters c; c.ScanAddAll("ccc=1&aaa=1&ccc=3&bbb&ccc=2"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&ccc=1&ccc=3&ccc=2"); } Y_UNIT_TEST(TestScanAddAllUnescaped1) { TCgiParameters c; c.ScanAddAllUnescaped("ccc=1&aaa=1&ccc=3&bbb&ccc=2"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&ccc=1&ccc=3&ccc=2"); } Y_UNIT_TEST(TestScanAddAllUnescaped2) { TCgiParameters c; c.ScanAddAllUnescaped("text=something&null"); UNIT_ASSERT_VALUES_EQUAL(c.size(), 2u); UNIT_ASSERT_VALUES_EQUAL(c.Get("text"), "something"); UNIT_ASSERT(c.Get("null").empty()); } Y_UNIT_TEST(TestScanAddAllUnescaped3) { TCgiParameters c; c.ScanAddAllUnescaped("text=%D0%9F%D1%80%D0%B8%D0%B2%D0%B5%D1%82%2C"); UNIT_ASSERT_VALUES_EQUAL(c.Get("text"), "%D0%9F%D1%80%D0%B8%D0%B2%D0%B5%D1%82%2C"); } Y_UNIT_TEST(TestEraseAll) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb&par=3"); c.EraseAll("par"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb="); } Y_UNIT_TEST(TestErase) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb&par=3&par=1"); c.Erase("par", 1); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&par=1&par=3&par=1"); c.Erase("par", "1"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&par=3"); } Y_UNIT_TEST(TestReplaceUnescaped1) { TCgiParameters c; c.ScanAddAll("many_keys=1&aaa=1&many_keys=2&bbb&many_keys=3"); c.ReplaceUnescaped("many_keys", "new_value"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&many_keys=new_value"); } Y_UNIT_TEST(TestReplaceUnescaped2) { TCgiParameters c; c.ScanAddAll("par=1&only_one=1&par=2&bbb&par=3"); c.ReplaceUnescaped("only_one", "new_value"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "bbb=&only_one=new_value&par=1&par=2&par=3"); } Y_UNIT_TEST(TestReplaceUnescaped3) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb&par=3"); c.ReplaceUnescaped("no_such_key", "new_value"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&no_such_key=new_value&par=1&par=2&par=3"); } Y_UNIT_TEST(TestReplaceUnescapedRange1) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb&par=3"); c.ReplaceUnescaped("par", {"x", "y", "z"}); // 3 old values, 3 new values UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&par=x&par=y&par=z"); } Y_UNIT_TEST(TestReplaceUnescapedRange2) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb"); c.ReplaceUnescaped("par", {"x", "y", "z"}); // 2 old values, 3 new values UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&par=x&par=y&par=z"); } Y_UNIT_TEST(TestReplaceUnescapedRange3) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb&par=3"); c.ReplaceUnescaped("par", {"x", "y"}); // 3 old values, 2 new values UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=1&bbb=&par=x&par=y"); } Y_UNIT_TEST(TestNumOfValues) { TCgiParameters c; c.ScanAddAll("par=1&aaa=1&par=2&bbb&par=3"); UNIT_ASSERT_VALUES_EQUAL(c.NumOfValues("par"), 3u); } Y_UNIT_TEST(TestUnscape) { TCgiParameters c("f=1&t=%84R%84%7C%84%80%84%7E&reqenc=SHIFT_JIS&p=0"); UNIT_ASSERT_VALUES_EQUAL(c.Get("t"), "\x84R\x84\x7C\x84\x80\x84\x7E"); } Y_UNIT_TEST(TestEmpty) { UNIT_ASSERT(TCgiParameters().Print().empty()); } Y_UNIT_TEST(TestJoinUnescaped) { TCgiParameters c; c.Scan("foo=1&foo=2"); c.JoinUnescaped("foo", ';', "0"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "foo=1;2;0"); } Y_UNIT_TEST(TestContInit) { TCgiParameters c = {std::make_pair("a", "a1"), std::make_pair("b", "b1"), std::make_pair("a", "a2")}; UNIT_ASSERT_VALUES_EQUAL(c.NumOfValues("a"), 2u); UNIT_ASSERT_VALUES_EQUAL(c.NumOfValues("b"), 1u); UNIT_ASSERT_VALUES_EQUAL(c.Get("b"), "b1"); UNIT_ASSERT_VALUES_EQUAL(c.Get("a", 0), "a1"); UNIT_ASSERT_VALUES_EQUAL(c.Get("a", 1), "a2"); UNIT_ASSERT_VALUES_EQUAL(c.Print(), "a=a1&a=a2&b=b1"); } Y_UNIT_TEST(TestPrintAsQuote) { TCgiParameters c = { std::make_pair("aaa", "value/with/slashes"), std::make_pair("b/b/b", "value_without_slashes"), std::make_pair("ccc", "value")}; UNIT_ASSERT_VALUES_EQUAL(c.Print(), "aaa=value/with/slashes&b/b/b=value_without_slashes&ccc=value"); UNIT_ASSERT_VALUES_EQUAL(c.QuotedPrint(""), "aaa=value%2Fwith%2Fslashes&b%2Fb%2Fb=value_without_slashes&ccc=value"); } Y_UNIT_TEST(TestPrintAsQuoteEmpty) { TCgiParameters c = {}; UNIT_ASSERT_VALUES_EQUAL(c.QuotedPrint(""), ""); } Y_UNIT_TEST(TestPrintAsQuoteEmptyKeyOrValue) { TCgiParameters c = { std::make_pair("", "value/of/empty"), std::make_pair("key/for/empty", "")}; UNIT_ASSERT_VALUES_EQUAL(c.Print(), "=value/of/empty&key/for/empty="); UNIT_ASSERT_VALUES_EQUAL(c.QuotedPrint(""), "=value%2Fof%2Fempty&key%2Ffor%2Fempty="); } }