intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 154 insertions, 0 deletions

diff --git a/library/cpp/http/misc/httpreqdata_ut.cpp b/library/cpp/http/misc/httpreqdata_ut.cpp
new file mode 100644
index 0000000000..e7f16ef27c
--- /dev/null
+++ b/library/cpp/http/misc/httpreqdata_ut.cpp
@@ -0,0 +1,154 @@
+#include "httpreqdata.h"
+
+#include <library/cpp/testing/unittest/registar.h>
+
+Y_UNIT_TEST_SUITE(TRequestServerDataTest) {
+    Y_UNIT_TEST(Headers) {
+        TServerRequestData sd;
+
+        sd.AddHeader("x-xx", "y-yy");
+        sd.AddHeader("x-Xx", "y-yy");
+
+        UNIT_ASSERT_VALUES_EQUAL(sd.HeadersCount(), 1);
+
+        sd.AddHeader("x-XxX", "y-yyy");
+        UNIT_ASSERT_VALUES_EQUAL(sd.HeadersCount(), 2);
+        UNIT_ASSERT_VALUES_EQUAL(TStringBuf(sd.HeaderIn("X-XX")), TStringBuf("y-yy"));
+        UNIT_ASSERT_VALUES_EQUAL(TStringBuf(sd.HeaderIn("X-XXX")), TStringBuf("y-yyy"));
+    }
+
+    Y_UNIT_TEST(ComplexHeaders) {
+        TServerRequestData sd;
+        sd.SetHost("zzz", 1);
+
+        sd.AddHeader("x-Xx", "y-yy");
+        UNIT_ASSERT_VALUES_EQUAL(sd.HeadersCount(), 1);
+        UNIT_ASSERT_VALUES_EQUAL(TStringBuf(sd.HeaderIn("X-XX")), TStringBuf("y-yy"));
+
+        sd.AddHeader("x-Xz", "y-yy");
+        UNIT_ASSERT_VALUES_EQUAL(sd.HeadersCount(), 2);
+        UNIT_ASSERT_VALUES_EQUAL(TStringBuf(sd.HeaderIn("X-Xz")), TStringBuf("y-yy"));
+
+        UNIT_ASSERT_VALUES_EQUAL(sd.ServerName(), "zzz");
+        UNIT_ASSERT_VALUES_EQUAL(sd.ServerPort(), "1");
+        sd.AddHeader("Host", "1234");
+        UNIT_ASSERT_VALUES_EQUAL(sd.HeadersCount(), 3);
+        UNIT_ASSERT_VALUES_EQUAL(TStringBuf(sd.HeaderIn("Host")), TStringBuf("1234"));
+        UNIT_ASSERT_VALUES_EQUAL(sd.ServerName(), "1234");
+        sd.AddHeader("Host", "12345:678");
+        UNIT_ASSERT_VALUES_EQUAL(sd.HeadersCount(), 3);
+        UNIT_ASSERT_VALUES_EQUAL(TStringBuf(sd.HeaderIn("Host")), TStringBuf("12345:678"));
+        UNIT_ASSERT_VALUES_EQUAL(sd.ServerName(), "12345");
+        UNIT_ASSERT_VALUES_EQUAL(sd.ServerPort(), "678");
+    }
+
+    Y_UNIT_TEST(ParseScan) {
+        TServerRequestData rd;
+
+        // Parse parses url without host
+        UNIT_ASSERT(!rd.Parse(" http://yandex.ru/yandsearch?&gta=fake&haha=da HTTP 1.1 OK"));
+
+        // This should work
+        UNIT_ASSERT(rd.Parse(" /yandsearch?&gta=fake&haha=da HTTP 1.1 OK"));
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), "&gta=fake&haha=da");
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), rd.OrigQueryStringBuf());
+
+        rd.Scan();
+        UNIT_ASSERT(rd.CgiParam.Has("gta", "fake"));
+        UNIT_ASSERT(rd.CgiParam.Has("haha", "da"));
+        UNIT_ASSERT(!rd.CgiParam.Has("no-param"));
+
+        rd.Clear();
+    }
+
+    Y_UNIT_TEST(Ctor) {
+        const TString qs("gta=fake&haha=da");
+        TServerRequestData rd(qs.c_str());
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), qs);
+        UNIT_ASSERT_STRINGS_EQUAL(rd.OrigQueryStringBuf(), qs);
+
+        UNIT_ASSERT(rd.CgiParam.Has("gta"));
+        UNIT_ASSERT(rd.CgiParam.Has("haha"));
+        UNIT_ASSERT(!rd.CgiParam.Has("no-param"));
+    }
+
+    Y_UNIT_TEST(HashCut) {
+        const TString qs("&gta=fake&haha=da");
+        const TString header = " /yandsearch?" + qs + "#&uberParam=yes&q=? HTTP 1.1 OK";
+
+        TServerRequestData rd;
+        rd.Parse(header.c_str());
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), qs);
+        UNIT_ASSERT_STRINGS_EQUAL(rd.OrigQueryStringBuf(), qs);
+
+        rd.Scan();
+        UNIT_ASSERT(rd.CgiParam.Has("gta"));
+        UNIT_ASSERT(rd.CgiParam.Has("haha"));
+        UNIT_ASSERT(!rd.CgiParam.Has("uberParam"));
+    }
+
+    Y_UNIT_TEST(MisplacedHashCut) {
+        TServerRequestData rd;
+        rd.Parse(" /y#ndsearch?&gta=fake&haha=da&uberParam=yes&q=? HTTP 1.1 OK");
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), "");
+        UNIT_ASSERT_STRINGS_EQUAL(rd.OrigQueryStringBuf(), "");
+
+        rd.Scan();
+        UNIT_ASSERT(rd.CgiParam.empty());
+    }
+
+    Y_UNIT_TEST(CornerCase) {
+        TServerRequestData rd;
+        rd.Parse(" /yandsearch?#");
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), "");
+        UNIT_ASSERT_STRINGS_EQUAL(rd.OrigQueryStringBuf(), "");
+
+        rd.Scan();
+        UNIT_ASSERT(rd.CgiParam.empty());
+    }
+
+    Y_UNIT_TEST(AppendQueryString) {
+        const TString qs("gta=fake&haha=da");
+        TServerRequestData rd(qs.c_str());
+
+        UNIT_ASSERT(rd.CgiParam.Has("gta", "fake"));
+        UNIT_ASSERT(rd.CgiParam.Has("haha", "da"));
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), qs);
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), rd.OrigQueryStringBuf());
+
+        constexpr TStringBuf appendix = "gta=true&gta=new";
+        rd.AppendQueryString(appendix.data(), appendix.size());
+
+        UNIT_ASSERT_STRINGS_EQUAL(rd.QueryStringBuf(), qs + '&' + appendix);
+        UNIT_ASSERT_STRINGS_EQUAL(rd.OrigQueryStringBuf(), qs);
+
+        rd.Scan();
+
+        UNIT_ASSERT(rd.CgiParam.Has("gta", "true"));
+        UNIT_ASSERT(rd.CgiParam.Has("gta", "new"));
+    }
+
+    Y_UNIT_TEST(SetRemoteAddrSimple) {
+        static const TString TEST = "abacaba.search.yandex.net";
+
+        TServerRequestData rd;
+        rd.SetRemoteAddr(TEST);
+        UNIT_ASSERT_STRINGS_EQUAL(TEST, rd.RemoteAddr());
+    }
+
+    Y_UNIT_TEST(SetRemoteAddrRandom) {
+        for (size_t size = 0; size < 2 * INET6_ADDRSTRLEN; ++size) {
+            const TString test = NUnitTest::RandomString(size, size);
+            TServerRequestData rd;
+            rd.SetRemoteAddr(test);
+            UNIT_ASSERT_STRINGS_EQUAL(test.substr(0, INET6_ADDRSTRLEN - 1), rd.RemoteAddr());
+        }
+    }
+
+} // TRequestServerDataTest