Add const attributes to mock attribute context.

2 files changed, 3 insertions, 3 deletions

diff --git a/searchlib/src/vespa/searchlib/test/mock_attribute_context.cpp b/searchlib/src/vespa/searchlib/test/mock_attribute_context.cpp
index 75e85d9d828..1003160249d 100644
--- a/searchlib/src/vespa/searchlib/test/mock_attribute_context.cpp
+++ b/searchlib/src/vespa/searchlib/test/mock_attribute_context.cpp
@@ -28,7 +28,7 @@ MockAttributeContext::getAttributeList(std::vector<const IAttributeVector *> & l
 MockAttributeContext::~MockAttributeContext() = default;
 
 void
-MockAttributeContext::add(std::shared_ptr<IAttributeVector> attr) {
+MockAttributeContext::add(std::shared_ptr<const IAttributeVector> attr) {
     _vectors[attr->getName()] = attr;
 }
 
diff --git a/searchlib/src/vespa/searchlib/test/mock_attribute_context.h b/searchlib/src/vespa/searchlib/test/mock_attribute_context.h
index 5b522e907b1..165a37994b9 100644
--- a/searchlib/src/vespa/searchlib/test/mock_attribute_context.h
+++ b/searchlib/src/vespa/searchlib/test/mock_attribute_context.h
@@ -11,12 +11,12 @@ namespace search::attribute::test {
 class MockAttributeContext : public IAttributeContext
 {
 private:
-    using Map = std::map<string, std::shared_ptr<IAttributeVector>>;
+    using Map = std::map<string, std::shared_ptr<const IAttributeVector>>;
     Map _vectors;
 
 public:
     ~MockAttributeContext() override;
-    void add(std::shared_ptr<IAttributeVector> attr);
+    void add(std::shared_ptr<const IAttributeVector> attr);
 
     const IAttributeVector *get(const string &name) const;
     const IAttributeVector * getAttribute(const string &name) const override;