PointLookup: added a point lookup query for unique indexes, and some tests

2 files changed, 113 insertions, 2 deletions

diff --git a/tests/fst_tests.cpp b/tests/fst_tests.cpp
index 4c8da4a..778f64a 100644
--- a/tests/fst_tests.cpp
+++ b/tests/fst_tests.cpp
@@ -20,14 +20,14 @@ typedef StringRec R;
 typedef FSTrie<R> Shard;
 
 #include "include/shard_string.h"
-//#include "include/rangequery.h"
+#include "include/pointlookup.h"
 
 Suite *unit_testing()
 {
     Suite *unit = suite_create("Fast-succinct Trie Shard Unit Testing");
 
- //   inject_rangequery_tests(unit);
     inject_shard_tests(unit);
+    inject_pointlookup_tests(unit);
 
     return unit;
 }
diff --git a/tests/include/pointlookup.h b/tests/include/pointlookup.h
new file mode 100644
index 0000000..bf4810b
--- /dev/null
+++ b/tests/include/pointlookup.h
@@ -0,0 +1,111 @@
+/*
+ * tests/include/pointlookup.h
+ *
+ * Standardized unit tests for point lookups against supporting
+ * shard types (must be unique for the moment)
+ *
+ * Copyright (C) 2024 Douglas Rumbaugh <drumbaugh@psu.edu> 
+ *
+ * Distributed under the Modified BSD License.
+ *
+ * WARNING: This file must be included in the main unit test set
+ *          after the definition of an appropriate Shard and R
+ *          type. In particular, R needs to implement the key-value
+ *          pair interface and Shard needs to support lower_bound. 
+ *          For other types of record and shard, you'll need to
+ *          use a different set of unit tests.
+ */
+#pragma once
+
+/*
+ * Uncomment these lines temporarily to remove errors in this file
+ * temporarily for development purposes. They should be removed prior
+ * to building, to ensure no duplicate definitions. These includes/defines
+ * should be included in the source file that includes this one, above the
+ * include statement.
+ */
+
+#include "shard/FSTrie.h"
+#include "query/pointlookup.h"
+#include "testing.h"
+
+#include <check.h>
+
+using namespace de;
+typedef StringRec R;
+typedef FSTrie<R> Shard;
+
+START_TEST(t_point_lookup_query)
+{
+    auto buffer = create_test_mbuffer<R>(1000);
+    auto shard = Shard(buffer->get_buffer_view());
+
+    pl::Parms<R> parms;
+    {
+        auto bv = buffer->get_buffer_view();
+        for (size_t i=0; i<bv.get_record_count(); i++) {
+            auto key = bv.get(i)->rec.key;
+
+            parms.search_key = key;
+            auto state = pl::Query<R, Shard>::get_query_state(&shard, &parms);
+            auto result = pl::Query<R, Shard>::query(&shard, state, &parms);
+            pl::Query<R, Shard>::delete_query_state(state);
+
+            ck_assert_int_eq(result.size(), 1);
+            ck_assert_str_eq(result[0].rec.key.c_str(), key.c_str());
+            ck_assert_int_eq(result[0].rec.value, bv.get(i)->rec.value);
+        }
+
+        /* point lookup miss; result size should be 0 */
+        parms.search_key = "computer";
+        auto state = pl::Query<R, Shard>::get_query_state(&shard, &parms);
+        auto result = pl::Query<R, Shard>::query(&shard, state, &parms);
+        pl::Query<R, Shard>::delete_query_state(state);
+
+        ck_assert_int_eq(result.size(), 0);
+    }
+
+    delete buffer;
+}
+END_TEST
+
+
+START_TEST(t_buffer_point_lookup)
+{
+
+    auto buffer = create_test_mbuffer<R>(1000);
+    pl::Parms<R> parms;
+    {
+        auto view = buffer->get_buffer_view();
+        for (int i=view.get_record_count()-1; i>=0; i--) {
+            parms.search_key = view.get(i)->rec.key;
+
+            auto state = pl::Query<R, Shard>::get_buffer_query_state(&view, &parms);
+            auto result = pl::Query<R, Shard>::buffer_query(state, &parms);
+            pl::Query<R, Shard>::delete_buffer_query_state(state);
+
+            ck_assert_int_eq(result.size(), 1);
+            ck_assert_str_eq(result[0].rec.key.c_str(), view.get(i)->rec.key.c_str());
+            ck_assert_int_eq(result[0].rec.value, view.get(i)->rec.value);
+        }
+
+        /* point lookup miss; result size should be 0 */
+        parms.search_key = "computer";
+        auto state = pl::Query<R, Shard>::get_buffer_query_state(&view, &parms);
+        auto result = pl::Query<R, Shard>::buffer_query(state, &parms);
+        pl::Query<R, Shard>::delete_buffer_query_state(state);
+
+        ck_assert_int_eq(result.size(), 0);
+    }
+
+    delete buffer;
+}
+END_TEST
+
+
+static void inject_pointlookup_tests(Suite *suite) {
+    TCase *point_lookup_query = tcase_create("Point Lookup Testing"); 
+    tcase_add_test(point_lookup_query, t_point_lookup_query); 
+    tcase_add_test(point_lookup_query, t_buffer_point_lookup); 
+    suite_add_tcase(suite, point_lookup_query);
+}