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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/testkit/testapp.h>
#include <vespa/searchcore/proton/attribute/imported_attributes_repo.h>
#include <vespa/searchcore/proton/reference/document_db_reference.h>
#include <vespa/searchcore/proton/test/mock_attribute_manager.h>
#include <vespa/searchlib/attribute/attributefactory.h>
#include <vespa/searchlib/attribute/imported_attribute_vector.h>
#include <vespa/log/log.h>
LOG_SETUP("document_db_reference_test");
using namespace proton;
using search::AttributeFactory;
using search::AttributeVector;
using search::IDocumentMetaStoreContext;
using search::attribute::BasicType;
using search::attribute::ImportedAttributeVector;
using search::attribute::ReadableAttributeVector;
using search::attribute::ReferenceAttribute;
ImportedAttributesRepo::UP
makeImportedAttributesRepo()
{
ImportedAttributesRepo::UP result = std::make_unique<ImportedAttributesRepo>();
ImportedAttributeVector::SP attr = std::make_shared<ImportedAttributeVector>("imported",
std::shared_ptr<ReferenceAttribute>(),
std::shared_ptr<IDocumentMetaStoreContext>(),
std::shared_ptr<ReadableAttributeVector>(),
std::shared_ptr<const IDocumentMetaStoreContext>(), false);
result->add("imported", std::move(attr));
return result;
}
struct Fixture {
std::shared_ptr<test::MockAttributeManager> attrMgr;
DocumentDBReference ref;
Fixture()
: attrMgr(std::make_shared<test::MockAttributeManager>()),
ref(attrMgr, std::shared_ptr<const IDocumentMetaStoreContext>(), std::shared_ptr<IGidToLidChangeHandler>())
{
attrMgr->addAttribute("regular", AttributeFactory::createAttribute("regular", {BasicType::INT32}));
attrMgr->setImportedAttributes(makeImportedAttributesRepo());
}
};
TEST_F("regular attribute vector can be retrieved", Fixture)
{
auto attr = f.ref.getAttribute("regular");
EXPECT_TRUE(attr.get());
const AttributeVector *attrPtr = dynamic_cast<const AttributeVector *>(attr.get());
EXPECT_TRUE(attrPtr != nullptr);
}
TEST_F("imported attribute vector can be retrieved", Fixture)
{
auto attr = f.ref.getAttribute("imported");
EXPECT_TRUE(attr.get());
const ImportedAttributeVector *importedPtr = dynamic_cast<const ImportedAttributeVector *>(attr.get());
EXPECT_TRUE(importedPtr != nullptr);
}
TEST_F("nullptr is returned for non-existing attribute vector", Fixture)
{
auto attr = f.ref.getAttribute("non-existing");
EXPECT_FALSE(attr.get());
}
TEST_MAIN()
{
TEST_RUN_ALL();
}
|