Provide extend interface for SingleRawExtAttribute.

1 files changed, 9 insertions, 9 deletions

diff --git a/searchlib/src/tests/attribute/extendattributes/extendattribute.cpp b/searchlib/src/tests/attribute/extendattributes/extendattribute.cpp
index c2285a99599..8f056323733 100644
--- a/searchlib/src/tests/attribute/extendattributes/extendattribute.cpp
+++ b/searchlib/src/tests/attribute/extendattributes/extendattribute.cpp
@@ -26,8 +26,7 @@ protected:
     void testExtendFloat(Attribute & attr);
     template <typename Attribute>
     void testExtendString(Attribute & attr);
-    template <typename Attribute>
-    void testExtendRaw(Attribute & attr);
+    void testExtendRaw(AttributeVector& attr);
 };
 
 template <typename Attribute>
@@ -145,33 +144,34 @@ void ExtendAttributeTest::testExtendString(Attribute & attr)
     }
 }
 
-template <typename Attribute>
-void ExtendAttributeTest::testExtendRaw(Attribute & attr)
+void ExtendAttributeTest::testExtendRaw(AttributeVector& attr)
 {
     std::vector<char> empty;
     std::vector<char> zeros{10, 0, 0, 11};
+    auto* ext_attr = attr.getExtendInterface();
+    EXPECT_NE(nullptr, ext_attr);
     uint32_t docId(0);
     EXPECT_EQ(0u, attr.getNumDocs());
     attr.addDoc(docId);
     EXPECT_EQ(0u, docId);
     EXPECT_EQ(1u, attr.getNumDocs());
-    attr.add(as_vector("1.7"));
+    ext_attr->add(as_vector("1.7"));
     auto buf = attr.get_raw(0);
     EXPECT_EQ(as_vector("1.7"), as_vector(buf));
-    attr.add(vespalib::ConstArrayRef<char>(as_vector("2.3")));
+    ext_attr->add(vespalib::ConstArrayRef<char>(as_vector("2.3")));
     buf = attr.get_raw(0);
     EXPECT_EQ(as_vector("2.3"), as_vector(buf));
     attr.addDoc(docId);
     EXPECT_EQ(1u, docId);
     EXPECT_EQ(attr.getNumDocs(), 2u);
-    attr.add(as_vector("3.6"));
+    ext_attr->add(as_vector("3.6"));
     buf = attr.get_raw(1);
     EXPECT_EQ(as_vector("3.6"), as_vector(buf));
     buf = attr.get_raw(0);
     EXPECT_EQ(as_vector("2.3"), as_vector(buf));
     attr.addDoc(docId);
     EXPECT_EQ(2u, docId);
-    attr.add(zeros);
+    ext_attr->add(zeros);
     buf = attr.get_raw(2);
     EXPECT_EQ(zeros, as_vector(buf));
     attr.addDoc(docId);
@@ -180,7 +180,7 @@ void ExtendAttributeTest::testExtendRaw(Attribute & attr)
     EXPECT_EQ(empty, as_vector(buf));
     attr.addDoc(docId);
     EXPECT_EQ(4u, docId);
-    attr.add(empty);
+    ext_attr->add(empty);
     buf = attr.get_raw(4);
     EXPECT_EQ(empty, as_vector(buf));
 }