From b138d9b9bdb25373cdfae2af5567bc51528a2764 Mon Sep 17 00:00:00 2001
From: Lucas <junsuny789@gmail.com>
Date: Fri, 27 Sep 2024 00:13:10 -0400
Subject: [PATCH] art

---
 src/bliss/bench_ART.h         |  5 +----
 tests/test_art/CMakeLists.txt |  9 +++++++++
 tests/test_art/art_tests.cpp  | 31 +++++++++++++++++++++++++++++++
 3 files changed, 41 insertions(+), 4 deletions(-)
 create mode 100644 tests/test_art/CMakeLists.txt
 create mode 100644 tests/test_art/art_tests.cpp

diff --git a/src/bliss/bench_ART.h b/src/bliss/bench_ART.h
index c5e2be9..4c0c675 100644
--- a/src/bliss/bench_ART.h
+++ b/src/bliss/bench_ART.h
@@ -1,7 +1,6 @@
 #ifndef BLISS_BENCH_ART
 #define BLISS_BENCH_ART
 
-#include "ART.h"
 #include "ART.h"
 
 #include <vector>
@@ -41,9 +40,7 @@ class BlissARTIndex : public BlissIndex<KEY_TYPE, VALUE_TYPE> {
     void put(KEY_TYPE key, VALUE_TYPE value) override {
         uint8_t ARTkey[KEY_SIZE];
         ART::loadKey(key, ARTkey);
-
-        uint8_t depth = 0;
-        ART::insert(_index, &_index, ARTkey, depth, value, VALUE_SIZE);
+        ART::insert(_index, &_index, ARTkey, 0, key, 8);
     }
 
     void end_routine() override {}
diff --git a/tests/test_art/CMakeLists.txt b/tests/test_art/CMakeLists.txt
new file mode 100644
index 0000000..b44a9ac
--- /dev/null
+++ b/tests/test_art/CMakeLists.txt
@@ -0,0 +1,9 @@
+get_filename_component(EXEC ${CMAKE_CURRENT_SOURCE_DIR} NAME)
+file(GLOB_RECURSE CPP_TESTS "*_tests.cpp")
+add_executable(${EXEC} ${CPP_TESTS})
+target_link_libraries(${EXEC} PRIVATE 
+bliss
+bliss_test_infra 
+GTest::gtest_main)
+include(GoogleTest)
+gtest_discover_tests(${EXEC})
\ No newline at end of file
diff --git a/tests/test_art/art_tests.cpp b/tests/test_art/art_tests.cpp
new file mode 100644
index 0000000..9bcefea
--- /dev/null
+++ b/tests/test_art/art_tests.cpp
@@ -0,0 +1,31 @@
+#include "bliss_index_tests.h"
+
+class BTreeTest : public BlissIndexTest {};
+
+TEST_F(BTreeTest, TestBTree_Sorted) {
+    index.reset(new bliss::BlissARTIndex<key_type, key_type>());
+    std::vector<key_type> data;
+    GenerateData(data, num_keys);
+
+    auto insert_start = data.begin();
+    auto insert_end = data.end();
+    executor::execute_inserts(*index, insert_start, insert_end);
+
+    for (auto key : data) {
+        EXPECT_TRUE(index->get(key));
+    }
+}
+
+TEST_F(BTreeTest, TestBTree_Random) {
+    index.reset(new bliss::BlissARTIndex<key_type, key_type>());
+    std::vector<key_type> data;
+    GenerateData(data, num_keys, false);
+
+    auto insert_start = data.begin();
+    auto insert_end = data.end();
+    executor::execute_inserts(*index, insert_start, insert_end);
+
+    for (auto key : data) {
+        EXPECT_TRUE(index->get(key));
+    }
+}
\ No newline at end of file