blob: 8b1ef88aa2c93c0b82975f68b0e291fc1089d9d5 (
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
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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/fastos/fastos.h>
#include <vespa/vdstestlib/cppunit/macros.h>
#include <tests/persistence/persistencetestutils.h>
#include <tests/persistence/common/persistenceproviderwrapper.h>
#include <vespa/storage/persistence/providershutdownwrapper.h>
namespace storage {
class ProviderShutdownWrapperTest : public SingleDiskPersistenceTestUtils
{
public:
CPPUNIT_TEST_SUITE(ProviderShutdownWrapperTest);
CPPUNIT_TEST(testShutdownOnFatalError);
CPPUNIT_TEST_SUITE_END();
void testShutdownOnFatalError();
};
CPPUNIT_TEST_SUITE_REGISTRATION(ProviderShutdownWrapperTest);
namespace {
class TestShutdownListener
: public framework::defaultimplementation::ShutdownListener
{
public:
TestShutdownListener() : _reason() {}
void requestShutdown(vespalib::stringref reason) override {
_reason = reason;
}
bool shutdownRequested() const { return !_reason.empty(); }
const vespalib::string& getReason() const { return _reason; }
private:
vespalib::string _reason;
};
}
void
ProviderShutdownWrapperTest::testShutdownOnFatalError()
{
// We wrap the wrapper. It's turtles all the way down!
PersistenceProviderWrapper providerWrapper(
getPersistenceProvider());
TestServiceLayerApp app;
ServiceLayerComponent component(app.getComponentRegister(), "dummy");
ProviderShutdownWrapper shutdownWrapper(providerWrapper, component);
TestShutdownListener shutdownListener;
app.getComponentRegister().registerShutdownListener(shutdownListener);
providerWrapper.setResult(
spi::Result(spi::Result::FATAL_ERROR, "eject! eject!"));
providerWrapper.setFailureMask(
PersistenceProviderWrapper::FAIL_ALL_OPERATIONS);
CPPUNIT_ASSERT(!shutdownListener.shutdownRequested());
// This should cause the node to implicitly be shut down
shutdownWrapper.getBucketInfo(
spi::Bucket(document::BucketId(16, 1234),
spi::PartitionId(0)));
CPPUNIT_ASSERT(shutdownListener.shutdownRequested());
CPPUNIT_ASSERT_EQUAL(vespalib::string("eject! eject!"),
shutdownListener.getReason());
// Triggering a new error should not cause shutdown to be requested twice.
providerWrapper.setResult(
spi::Result(spi::Result::FATAL_ERROR, "boom!"));
shutdownWrapper.getBucketInfo(
spi::Bucket(document::BucketId(16, 1234),
spi::PartitionId(0)));
CPPUNIT_ASSERT_EQUAL(vespalib::string("eject! eject!"),
shutdownListener.getReason());
}
} // ns storage
|