blob: 597bb4b07ffbd69741768ba95af75e01bd6ffbf0 (
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
56
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/storageapi/message/bucket.h>
#include <vespa/storageapi/message/state.h>
#include <vespa/persistence/spi/test.h>
#include <tests/persistence/common/persistenceproviderwrapper.h>
#include <vespa/persistence/dummyimpl/dummypersistence.h>
#include <tests/persistence/common/filestortestfixture.h>
#include <vespa/vdslib/state/clusterstate.h>
using storage::spi::test::makeSpiBucket;
using namespace ::testing;
namespace storage {
struct DeactivateBucketsTest : FileStorTestFixture {
bool isActive(const document::BucketId&) const;
};
bool
DeactivateBucketsTest::isActive(const document::BucketId& bucket) const
{
StorBucketDatabase::WrappedEntry entry(
_node->getStorageBucketDatabase().get(bucket, "foo"));
assert(entry.exist());
return entry->info.isActive();
}
TEST_F(DeactivateBucketsTest, buckets_in_database_deactivated_when_node_down_in_cluster_state) {
TestFileStorComponents c(*this);
// Must set state to up first, or down-edge case won't trigger.
std::string upState("storage:2 distributor:2");
_node->getStateUpdater().setClusterState(
lib::ClusterState::CSP(new lib::ClusterState(upState)));
document::BucketId bucket(8, 123);
createBucket(bucket);
api::BucketInfo serviceLayerInfo(1, 2, 3, 4, 5, true, true);
{
StorBucketDatabase::WrappedEntry entry(
_node->getStorageBucketDatabase().get(bucket, "foo",
StorBucketDatabase::CREATE_IF_NONEXISTING));
entry->info = serviceLayerInfo;
entry.write();
}
EXPECT_TRUE(isActive(bucket));
std::string downState("storage:2 .1.s:d distributor:2");
_node->getStateUpdater().setClusterState(
lib::ClusterState::CSP(new lib::ClusterState(downState)));
// Buckets should have been deactivated in content layer
EXPECT_FALSE(isActive(bucket));
}
} // namespace storage
|
