1 files changed, 42 insertions, 0 deletions
diff --git a/storage/src/tests/persistence/has_mask_remapper_test.cpp b/storage/src/tests/persistence/has_mask_remapper_test.cpp
new file mode 100644
index 00000000000..58f6f5b3fe7
--- /dev/null
+++ b/storage/src/tests/persistence/has_mask_remapper_test.cpp
@@ -0,0 +1,42 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include <vespa/storage/persistence/filestorage/has_mask_remapper.h>
+#include <gtest/gtest.h>
+
+namespace storage {
+
+using NodeList = std::vector<api::MergeBucketCommand::Node>;
+
+const NodeList merge_operation_nodes{{0, true}, {1, true}, {2, false}, {3, false}, {4, false}};
+
+TEST(HasMaskRemapperTest, test_remap_none)
+{
+    HasMaskRemapper remap_mask(merge_operation_nodes, merge_operation_nodes);
+    for (uint32_t i = 0; i < (1u << merge_operation_nodes.size()); ++i) {
+        EXPECT_EQ(i, remap_mask(i));
+    }
+}
+
+TEST(HasMaskRemapperTest, test_remap_subset)
+{
+    NodeList reply_nodes{{0, true}, {1, true}, {3, false}};
+    HasMaskRemapper remap_mask(merge_operation_nodes, reply_nodes);
+    std::vector<uint16_t> remapped;
+    for (uint32_t i = 0; i < (1u << reply_nodes.size()); ++i) {
+        remapped.push_back(remap_mask(i));
+    }
+    EXPECT_EQ((std::vector<uint16_t>{0u, 1u, 2u, 3u, 8u, 9u, 10u, 11u}), remapped);
+}
+
+TEST(HasMaskRemapperTest, test_remap_swapped_subset)
+{
+    NodeList reply_nodes{{1, true}, {0, true}};
+    HasMaskRemapper remap_mask(merge_operation_nodes, reply_nodes);
+    std::vector<uint16_t> remapped;
+    for (uint32_t i = 0; i < (1u << reply_nodes.size()); ++i) {
+        remapped.push_back(remap_mask(i));
+    }
+    EXPECT_EQ((std::vector<uint16_t>{0u, 2u, 1u, 3u}), remapped);
+}
+
+}