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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/log/log.h>
LOG_SETUP("attribute_populator_test");
#include <vespa/vespalib/testkit/testapp.h>
#include <vespa/searchcommon/common/schema.h>
#include <vespa/searchcore/proton/attribute/attribute_populator.h>
#include <vespa/searchcore/proton/attribute/attributemanager.h>
#include <vespa/searchcore/proton/common/hw_info.h>
#include <vespa/searchcore/proton/test/test.h>
#include <vespa/searchlib/common/foregroundtaskexecutor.h>
#include <vespa/searchlib/index/docbuilder.h>
#include <vespa/searchlib/index/dummyfileheadercontext.h>
#include <vespa/searchlib/test/directory_handler.h>
#include <vespa/vespalib/util/stringfmt.h>
using namespace document;
using namespace proton;
using namespace search;
using namespace search::index;
using search::test::DirectoryHandler;
using AVBasicType = search::attribute::BasicType;
using AVConfig = search::attribute::Config;
const vespalib::string TEST_DIR = "testdir";
const uint64_t CREATE_SERIAL_NUM = 8u;
Schema
createSchema()
{
Schema schema;
schema.addAttributeField(Schema::AttributeField("a1", Schema::DataType::INT32));
return schema;
}
struct DocContext
{
Schema _schema;
DocBuilder _builder;
DocContext()
: _schema(createSchema()),
_builder(_schema)
{
}
Document::UP create(uint32_t id, int64_t fieldValue) {
vespalib::string docId =
vespalib::make_string("id:searchdocument:searchdocument::%u", id);
return _builder.startDocument(docId).
startAttributeField("a1").addInt(fieldValue).endField().
endDocument();
}
};
struct Fixture
{
DirectoryHandler _testDir;
DummyFileHeaderContext _fileHeader;
ForegroundTaskExecutor _attributeFieldWriter;
HwInfo _hwInfo;
AttributeManager::SP _mgr;
std::unique_ptr<AttributePopulator> _pop;
DocContext _ctx;
Fixture()
: _testDir(TEST_DIR),
_fileHeader(),
_attributeFieldWriter(),
_hwInfo(),
_mgr(new AttributeManager(TEST_DIR, "test.subdb",
TuneFileAttributes(),
_fileHeader, _attributeFieldWriter, _hwInfo)),
_pop(),
_ctx()
{
_mgr->addAttribute({ "a1", AVConfig(AVBasicType::INT32)},
CREATE_SERIAL_NUM);
_pop = std::make_unique<AttributePopulator>(_mgr, 1, "test", CREATE_SERIAL_NUM);
}
AttributeGuard::UP getAttr() {
return _mgr->getAttribute("a1");
}
};
TEST_F("require that reprocess with document populates attribute", Fixture)
{
AttributeGuard::UP attr = f.getAttr();
EXPECT_EQUAL(1u, attr->get()->getNumDocs());
f._pop->handleExisting(5, *f._ctx.create(0, 33));
EXPECT_EQUAL(6u, attr->get()->getNumDocs());
EXPECT_EQUAL(33, attr->get()->getInt(5));
EXPECT_EQUAL(1u, attr->get()->getStatus().getLastSyncToken());
f._pop->handleExisting(6, *f._ctx.create(1, 44));
EXPECT_EQUAL(7u, attr->get()->getNumDocs());
EXPECT_EQUAL(44, attr->get()->getInt(6));
EXPECT_EQUAL(2u, attr->get()->getStatus().getLastSyncToken());
f._pop->done();
EXPECT_EQUAL(CREATE_SERIAL_NUM, attr->get()->getStatus().getLastSyncToken());
}
TEST_MAIN()
{
vespalib::rmdir(TEST_DIR, true);
TEST_RUN_ALL();
}
|