blob: 8b093be08b77b0fb6aaeec56f35659b81d248234 (
plain) (
blame)
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
|
// 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/exclusive_attribute_read_accessor.h>
#include <vespa/searchcommon/attribute/config.h>
#include <vespa/searchlib/attribute/attributefactory.h>
#include <vespa/searchlib/attribute/attributevector.h>
#include <vespa/vespalib/util/sequencedtaskexecutor.h>
#include <vespa/vespalib/util/gate.h>
using namespace proton;
using namespace search;
using namespace vespalib;
using ReadGuard = ExclusiveAttributeReadAccessor::Guard;
VESPA_THREAD_STACK_TAG(test_executor)
AttributeVector::SP
createAttribute()
{
attribute::Config cfg(attribute::BasicType::INT32, attribute::CollectionType::SINGLE);
return search::AttributeFactory::createAttribute("myattr", cfg);
}
struct Fixture
{
AttributeVector::SP attribute;
std::unique_ptr<ISequencedTaskExecutor> writer;
ExclusiveAttributeReadAccessor accessor;
Fixture()
: attribute(createAttribute()),
writer(SequencedTaskExecutor::create(test_executor, 1)),
accessor(attribute, *writer)
{}
};
TEST_F("require that attribute write thread is blocked while guard is held", Fixture)
{
ReadGuard::UP guard = f.accessor.takeGuard();
Gate gate;
f.writer->execute(f.writer->getExecutorIdFromName(f.attribute->getNamePrefix()), [&gate]() { gate.countDown(); });
bool reachedZero = gate.await(100ms);
EXPECT_FALSE(reachedZero);
EXPECT_EQUAL(1u, gate.getCount());
guard.reset();
gate.await();
EXPECT_EQUAL(0u, gate.getCount());
}
TEST_MAIN()
{
TEST_RUN_ALL();
}
|