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
|
#pragma once
#include <absl/status/status.h>
#include <absl/status/statusor.h>
#include <absl/strings/str_cat.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
namespace testing {
/*
* This file contains gtest matchers for absl::Status.
*
* Example usage:
*
* EXPECT_OK(status); -- fails the test if 'status' is an error
* ASSERT_OK(status); -- instantly fails the test if error
*
* using ::testing::IsStatusCode;
* EXPECT_THAT(status, IsStatusCode(absl::StatusCode::kInternal));
*/
namespace nix_internal {
using ::testing::MakeMatcher;
using ::testing::Matcher;
using ::testing::MatcherInterface;
using ::testing::MatchResultListener;
MATCHER_P(IsStatusCode, code, "") { return arg.code() == code; }
class StatusCodeMatcher {
public:
StatusCodeMatcher(absl::StatusCode code) : code_(code) {}
// Match on absl::Status.
template <class T,
typename std::enable_if<std::is_same<T, absl::Status>::value,
int>::type int_ = 0>
bool MatchAndExplain(const T& status,
MatchResultListener* /* listener */) const {
return status.code() == code_;
}
// Match on absl::StatusOr.
//
// note: I check for the return value of ConsumeValueOrDie because it's the
// only non-overloaded member I could figure out how to select. Checking for
// the presence of .status() didn't work because it's overloaded, so
// std::invoke_result can't pick which overload to use.
template <class T,
typename std::enable_if<
std::is_same<typename std::invoke_result<
decltype(&T::ConsumeValueOrDie), T>::type,
typename T::value_type>::value,
int>::type int_ = 0>
bool MatchAndExplain(const T& statusor,
MatchResultListener* /* listener */) const {
return statusor.status().code() == code_;
}
void DescribeTo(std::ostream* os) const { *os << "is " << code_; }
void DescribeNegationTo(std::ostream* os) const { *os << "isn't " << code_; }
private:
absl::StatusCode code_;
};
} // namespace nix_internal
PolymorphicMatcher<nix_internal::StatusCodeMatcher> IsStatusCode(
absl::StatusCode code) {
return MakePolymorphicMatcher(nix_internal::StatusCodeMatcher(code));
}
#define EXPECT_OK(status) \
EXPECT_THAT((status), testing::IsStatusCode(absl::StatusCode::kOk))
#define ASSERT_OK(status) \
ASSERT_THAT((status), testing::IsStatusCode(absl::StatusCode::kOk))
} // namespace testing
|
