Cleanup multi value mapping unit test.

1 files changed, 6 insertions, 6 deletions

diff --git a/searchlib/src/tests/attribute/multi_value_mapping/multi_value_mapping_test.cpp b/searchlib/src/tests/attribute/multi_value_mapping/multi_value_mapping_test.cpp
index 17e852c8b92..4b01808e855 100644
--- a/searchlib/src/tests/attribute/multi_value_mapping/multi_value_mapping_test.cpp
+++ b/searchlib/src/tests/attribute/multi_value_mapping/multi_value_mapping_test.cpp
@@ -91,21 +91,21 @@ public:
           _maxSmallArraySize()
     {
     }
-    void setup(uint32_t maxSmallArraySize, bool enable_free_lists = true) {
-        ArrayStoreConfig config(maxSmallArraySize,
+    void setup(uint32_t max_array_store_type_id, bool enable_free_lists = true) {
+        ArrayStoreConfig config(max_array_store_type_id,
                                 ArrayStoreConfig::AllocSpec(0, RefType::offsetSize(), 8_Ki, ALLOC_GROW_FACTOR));
         config.enable_free_lists(enable_free_lists);
         _mvMapping = std::make_unique<MvMapping>(config, vespalib::GrowStrategy(), std::make_unique<MemoryAllocatorObserver>(_stats));
         _attr = std::make_unique<AttributeType>(*_mvMapping);
-        _maxSmallArraySize = maxSmallArraySize;
+        _maxSmallArraySize = _mvMapping->get_mapper().get_array_size(max_array_store_type_id);
     }
-    void setup(uint32_t maxSmallArraySize, size_t min_entries, size_t max_entries, size_t num_entries_for_new_buffer, bool enable_free_lists = true) {
-        ArrayStoreConfig config(maxSmallArraySize,
+    void setup(uint32_t max_array_store_type_id, size_t min_entries, size_t max_entries, size_t num_entries_for_new_buffer, bool enable_free_lists = true) {
+        ArrayStoreConfig config(max_array_store_type_id,
                                 ArrayStoreConfig::AllocSpec(min_entries, max_entries, num_entries_for_new_buffer, ALLOC_GROW_FACTOR));
         config.enable_free_lists(enable_free_lists);
         _mvMapping = std::make_unique<MvMapping>(config, vespalib::GrowStrategy(), std::make_unique<MemoryAllocatorObserver>(_stats));
         _attr = std::make_unique<AttributeType>(*_mvMapping);
-        _maxSmallArraySize = maxSmallArraySize;
+        _maxSmallArraySize = _mvMapping->get_mapper().get_array_size(max_array_store_type_id);
     }
     ~MappingTestBase() { }