searchlib/src/vespa/searchlib/test/mock_attribute_context.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include <vespa/searchcommon/attribute/iattributecontext.h>
#include <map>

namespace search::attribute::test {

class MockAttributeContext : public IAttributeContext
{
private:
    using Map = std::map<string, IAttributeVector *>;
    Map _vectors;

public:
    ~MockAttributeContext() override;
    void add(IAttributeVector *attr);

    const IAttributeVector *get(const string &name) const;
    const IAttributeVector * getAttribute(const string &name) const override;
    const IAttributeVector * getAttributeStableEnum(const string &name) const override;
    void getAttributeList(std::vector<const IAttributeVector *> & list) const override;
    void asyncForAttribute(const vespalib::string &, std::unique_ptr<IAttributeFunctor>) const override;
};

}