Fixed a few concurrency bugs

1 files changed, 9 insertions, 9 deletions

diff --git a/tests/vptree_tests.cpp b/tests/vptree_tests.cpp
index 49964e5..3bbf7c1 100644
--- a/tests/vptree_tests.cpp
+++ b/tests/vptree_tests.cpp
@@ -33,7 +33,7 @@ START_TEST(t_mbuffer_init)
         buffer->append({i, i});
     }
 
-    Shard* shard = new Shard(buffer->get_buffer_view());
+    Shard* shard = new Shard(buffer->get_buffer_view(buffer->debug_get_head()));
     ck_assert_uint_eq(shard->get_record_count(), n);
 
     delete buffer;
@@ -48,9 +48,9 @@ START_TEST(t_wss_init)
     auto mbuffer2 = create_test_mbuffer<R>(n);
     auto mbuffer3 = create_test_mbuffer<R>(n);
 
-    auto shard1 = new Shard(mbuffer1->get_buffer_view());
-    auto shard2 = new Shard(mbuffer2->get_buffer_view());
-    auto shard3 = new Shard(mbuffer3->get_buffer_view());
+    auto shard1 = new Shard(mbuffer1->get_buffer_view(mbuffer1->debug_get_head()));
+    auto shard2 = new Shard(mbuffer2->get_buffer_view(mbuffer2->debug_get_head()));
+    auto shard3 = new Shard(mbuffer3->get_buffer_view(mbuffer3->debug_get_head()));
 
     std::vector<const Shard *> shards = {shard1, shard2, shard3};
     auto shard4 = new Shard(shards);
@@ -74,10 +74,10 @@ START_TEST(t_point_lookup)
     size_t n = 16;
 
     auto buffer = create_sequential_mbuffer<R>(0, n);
-    auto wss = Shard(buffer->get_buffer_view());
+    auto wss = Shard(buffer->get_buffer_view(buffer->debug_get_head()));
 
     {
-        auto bv = buffer->get_buffer_view();
+        auto bv = buffer->get_buffer_view(buffer->debug_get_head());
 
         for (size_t i=0; i<n; i++) {
             PRec r;
@@ -102,7 +102,7 @@ START_TEST(t_point_lookup_miss)
     size_t n = 10000;
 
     auto buffer = create_sequential_mbuffer<R>(0, n);
-    auto wss = Shard(buffer->get_buffer_view());
+    auto wss = Shard(buffer->get_buffer_view(buffer->debug_get_head()));
 
     for (size_t i=n + 100; i<2*n; i++) {
         PRec r;
@@ -131,7 +131,7 @@ START_TEST(t_buffer_query)
     p.point = target;
 
     {
-        auto bv = buffer->get_buffer_view();
+        auto bv = buffer->get_buffer_view(buffer->debug_get_head());
         auto query = Q::local_preproc_buffer(&bv, &p);
         auto result = Q::local_query_buffer(query);
         delete query;
@@ -151,7 +151,7 @@ START_TEST(t_knn_query)
     size_t n = 1000;
     auto buffer = create_sequential_mbuffer<R>(0, n);
 
-    auto vptree = VPTree<PRec>(buffer->get_buffer_view());
+    auto vptree = VPTree<PRec>(buffer->get_buffer_view(buffer->debug_get_head()));
 
     Q::Parameters p;