intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 30 insertions, 0 deletions

diff --git a/library/cpp/containers/comptrie/loader/loader_ut.cpp b/library/cpp/containers/comptrie/loader/loader_ut.cpp
new file mode 100644
index 00000000000..345063a31e4
--- /dev/null
+++ b/library/cpp/containers/comptrie/loader/loader_ut.cpp
@@ -0,0 +1,30 @@
+#include <library/cpp/testing/unittest/registar.h>
+#include <library/cpp/containers/comptrie/comptrie.h>
+#include <library/cpp/containers/comptrie/loader/loader.h>
+
+using TDummyTrie = TCompactTrie<char, i32>;
+
+namespace {
+    const unsigned char DATA[] = {
+#include "data.inc"
+    };
+}
+
+Y_UNIT_TEST_SUITE(ArchiveLoaderTests) {
+    Y_UNIT_TEST(BaseTest) {
+        TDummyTrie trie = LoadTrieFromArchive<TDummyTrie>("/dummy.trie", DATA, true);
+        UNIT_ASSERT_EQUAL(trie.Size(), 3);
+
+        const TString TrieKyes[3] = {
+            "zero", "one", "two"};
+        i32 val = -1;
+        for (i32 i = 0; i < 3; ++i) {
+            UNIT_ASSERT(trie.Find(TrieKyes[i].data(), TrieKyes[i].size(), &val));
+            UNIT_ASSERT_EQUAL(i, val);
+        }
+
+        UNIT_CHECK_GENERATED_EXCEPTION(
+            LoadTrieFromArchive<TDummyTrie>("/noname.trie", DATA),
+            yexception);
+    }
+}