diff options
Diffstat (limited to 'library/cpp/packers/ut/packers_ut.cpp')
-rw-r--r-- | library/cpp/packers/ut/packers_ut.cpp | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/library/cpp/packers/ut/packers_ut.cpp b/library/cpp/packers/ut/packers_ut.cpp new file mode 100644 index 0000000000..18ce2150d1 --- /dev/null +++ b/library/cpp/packers/ut/packers_ut.cpp @@ -0,0 +1,110 @@ +#include <library/cpp/testing/unittest/registar.h> + +#include <util/stream/output.h> +#include <utility> + +#include <util/charset/wide.h> +#include <util/generic/algorithm.h> +#include <util/generic/buffer.h> +#include <util/generic/map.h> +#include <util/generic/vector.h> +#include <util/generic/ptr.h> +#include <util/generic/ylimits.h> + +#include <util/folder/dirut.h> + +#include <util/random/random.h> + +#include <util/string/hex.h> + +#include "packers.h" + +#include <array> +#include <iterator> + +class TPackersTest: public TTestBase { +private: + UNIT_TEST_SUITE(TPackersTest); + UNIT_TEST(TestPackers); + UNIT_TEST_SUITE_END(); + + template <class TData, class TPacker> + void TestPacker(const TData& data); + + template <class TData, class TPacker> + void TestPacker(const TData* test, size_t size); + +public: + void TestPackers(); +}; + +UNIT_TEST_SUITE_REGISTRATION(TPackersTest); + +template <class TData, class TPacker> +void TPackersTest::TestPacker(const TData& data) { + size_t len = TPacker().MeasureLeaf(data); + size_t bufLen = len * 3; + + TArrayHolder<char> buf(new char[bufLen]); + memset(buf.Get(), -1, bufLen); + + TPacker().PackLeaf(buf.Get(), data, len); + + UNIT_ASSERT(TPacker().SkipLeaf(buf.Get()) == len); + + TData dataTmp; + TPacker().UnpackLeaf(buf.Get(), dataTmp); + UNIT_ASSERT(data == dataTmp); +} + +template <class TData, class TPacker> +void TPackersTest::TestPacker(const TData* test, size_t size) { + for (size_t i = 0; i < size; ++i) { + TestPacker<TData, TPacker>(test[i]); + } +} + +void TPackersTest::TestPackers() { + { + const TString test[] = {"", + "a", "b", "c", "d", + "aa", "ab", "ac", "ad", + "aaa", "aab", "aac", "aad", + "aba", "abb", "abc", "abd", + "asdfjjmk.gjilsjgilsjilgjildsajgfilsjdfilgjm ldsa8oq43u 583uq4905 -q435 jiores u893q 5oiju fd-KE 89536 9Q2URE 12AI894T3 89 Q*(re43"}; + + TestPacker<TString, NPackers::TPacker<TString>>(test, Y_ARRAY_SIZE(test)); + + for (size_t i = 0; i != Y_ARRAY_SIZE(test); ++i) { + TestPacker<TUtf16String, NPackers::TPacker<TUtf16String>>(UTF8ToWide(test[i])); + } + } + { + const ui64 test[] = { + 0, 1, 2, 3, 4, 5, 6, 76, 100000, Max<ui64>()}; + + TestPacker<ui64, NPackers::TPacker<ui64>>(test, Y_ARRAY_SIZE(test)); + } + { + const int test[] = { + 0, 1, 2, 3, 4, 5, 6, 76, 100000, -1, -2, -3, -4, -5, -6, -76, -10000, Min<int>(), Max<int>()}; + + TestPacker<int, NPackers::TPacker<int>>(test, Y_ARRAY_SIZE(test)); + } + { + const float test[] = { + 2.f, 3.f, 4.f, 0.f, -0.f, 1.f, -1.f, 1.1f, -1.1f, + std::numeric_limits<float>::min(), -std::numeric_limits<float>::min(), + std::numeric_limits<float>::max(), -std::numeric_limits<float>::max()}; + + TestPacker<float, NPackers::TFloatPacker>(test, Y_ARRAY_SIZE(test)); + } + { + const double test[] = { + 0., -0., 1., -1., 1.1, -1.1, + std::numeric_limits<double>::min(), -std::numeric_limits<double>::min(), + std::numeric_limits<double>::max(), -std::numeric_limits<double>::max()}; + + TestPacker<double, NPackers::TDoublePacker>(test, Y_ARRAY_SIZE(test)); + } +} |