aboutsummaryrefslogtreecommitdiffstats
path: root/storage/src/tests/distributor/persistence_metrics_set_test.cpp
blob: 7d8fe6d096d4bac4c0155ab28343198673f9a28e (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/storage/distributor/distributormetricsset.h>
#include <vespa/storageapi/messageapi/returncode.h>
#include <vespa/vespalib/gtest/gtest.h>

using namespace ::testing;

namespace storage::distributor {

struct PersistenceMetricsSetTest : Test {
    void assert_failure_is_counted(PersistenceOperationMetricSet& metrics,
                                   api::ReturnCode::Result failure_code,
                                   const metrics::LongCountMetric& checked)
    {
        metrics.updateFromResult(api::ReturnCode(failure_code));
        EXPECT_EQ(1, checked.getLongValue("count"));
        EXPECT_EQ(0, metrics.ok.getLongValue("count"));
    }
};

TEST_F(PersistenceMetricsSetTest, successful_return_codes_are_counted_as_ok) {
    PersistenceOperationMetricSet metrics("foo");
    metrics.updateFromResult(api::ReturnCode());
    EXPECT_EQ(1, metrics.ok.getLongValue("count"));
}

TEST_F(PersistenceMetricsSetTest, wrong_distribution_failure_is_counted) {
    PersistenceOperationMetricSet metrics("foo");
    assert_failure_is_counted(metrics, api::ReturnCode::WRONG_DISTRIBUTION,
                              metrics.failures.wrongdistributor);
}

TEST_F(PersistenceMetricsSetTest, timeout_failure_is_counted) {
    PersistenceOperationMetricSet metrics("foo");
    assert_failure_is_counted(metrics, api::ReturnCode::TIMEOUT,
                              metrics.failures.timeout);
}

// Note for these tests: busy, connection failures et al are sets of
// failure codes and not just a single code. We only test certain members
// of these sets here. See api::ReturnCode implementation for an exhaustive
// list.
TEST_F(PersistenceMetricsSetTest, busy_failure_is_counted) {
    PersistenceOperationMetricSet metrics("foo");
    assert_failure_is_counted(metrics, api::ReturnCode::BUSY,
                              metrics.failures.busy);
}

TEST_F(PersistenceMetricsSetTest, connection_failure_is_counted) {
    PersistenceOperationMetricSet metrics("foo");
    // This is dirty enum value coercion, but this is how "parent protocol"
    // error codes are handled already.
    api::ReturnCode::Result error_code(static_cast<api::ReturnCode::Result>(
            mbus::ErrorCode::CONNECTION_ERROR));
    assert_failure_is_counted(metrics, error_code,
                              metrics.failures.notconnected);
}

TEST_F(PersistenceMetricsSetTest, inconsistent_bucket_is_counted) {
    PersistenceOperationMetricSet metrics("foo");
    assert_failure_is_counted(metrics, api::ReturnCode::BUCKET_NOT_FOUND,
                              metrics.failures.inconsistent_bucket);
}

TEST_F(PersistenceMetricsSetTest, non_special_cased_failure_codes_are_catchall_counted) {
    PersistenceOperationMetricSet metrics("foo");
    assert_failure_is_counted(metrics, api::ReturnCode::REJECTED,
                              metrics.failures.storagefailure);
}

}