sandboxed-api/sandboxed_api/util/status_matchers.h
Copybara-Service 97b3a9e325 Merge pull request #147 from oshogbo:matcher2
PiperOrigin-RevId: 435267982
Change-Id: I8f877da70282df9192be4cfe43d74d1539b9824f
2022-03-17 00:13:21 -07:00

131 lines
4.1 KiB
C++

// Copyright 2019 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#ifndef SANDBOXED_API_UTIL_STATUS_MATCHERS_H_
#define SANDBOXED_API_UTIL_STATUS_MATCHERS_H_
#include <type_traits>
#include "gmock/gmock.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/types/optional.h"
#include "sandboxed_api/util/status_macros.h"
#define SAPI_ASSERT_OK(expr) ASSERT_THAT(expr, ::sapi::IsOk())
#define SAPI_ASSERT_OK_AND_ASSIGN(lhs, rexpr) \
SAPI_ASSERT_OK_AND_ASSIGN_IMPL( \
SAPI_MACROS_IMPL_CONCAT(_sapi_statusor, __LINE__), lhs, rexpr)
#define SAPI_ASSERT_OK_AND_ASSIGN_IMPL(statusor, lhs, rexpr) \
auto statusor = (rexpr); \
ASSERT_THAT(statusor.status(), ::sapi::IsOk()); \
lhs = std::move(statusor).value();
namespace sapi {
namespace internal {
class IsOkMatcher {
public:
template <typename StatusT>
bool MatchAndExplain(const StatusT& status_container,
::testing::MatchResultListener* listener) const {
if (!status_container.ok()) {
*listener << "which is not OK";
return false;
}
return true;
}
void DescribeTo(std::ostream* os) const { *os << "is OK"; }
void DescribeNegationTo(std::ostream* os) const { *os << "is not OK"; }
};
class StatusIsMatcher {
public:
StatusIsMatcher(const StatusIsMatcher&) = default;
StatusIsMatcher(absl::StatusCode code,
absl::optional<absl::string_view> message)
: code_(code), message_(message) {}
template <typename T>
bool MatchAndExplain(const T& value,
::testing::MatchResultListener* listener) const {
auto status = GetStatus(value);
if (code_ != status.code()) {
*listener << "whose error code is "
<< absl::StatusCodeToString(status.code());
return false;
}
if (message_.has_value() && status.message() != message_.value()) {
*listener << "whose error message is '" << message_.value() << "'";
return false;
}
return true;
}
void DescribeTo(std::ostream* os) const {
*os << "has a status code that is " << absl::StatusCodeToString(code_);
if (message_.has_value()) {
*os << ", and has an error message that is '" << message_.value() << "'";
}
}
void DescribeNegationTo(std::ostream* os) const {
*os << "has a status code that is not " << absl::StatusCodeToString(code_);
if (message_.has_value()) {
*os << ", and has an error message that is not '" << message_.value()
<< "'";
}
}
private:
template <typename StatusT,
typename std::enable_if<
!std::is_void<decltype(std::declval<StatusT>().code())>::value,
int>::type = 0>
static const StatusT& GetStatus(const StatusT& status) {
return status;
}
template <typename StatusOrT,
typename StatusT = decltype(std::declval<StatusOrT>().status())>
static StatusT GetStatus(const StatusOrT& status_or) {
return status_or.status();
}
const absl::StatusCode code_;
const absl::optional<std::string> message_;
};
} // namespace internal
inline ::testing::PolymorphicMatcher<internal::IsOkMatcher> IsOk() {
return ::testing::MakePolymorphicMatcher(internal::IsOkMatcher{});
}
inline ::testing::PolymorphicMatcher<internal::StatusIsMatcher> StatusIs(
absl::StatusCode code,
absl::optional<absl::string_view> message = absl::nullopt) {
return ::testing::MakePolymorphicMatcher(
internal::StatusIsMatcher(code, message));
}
} // namespace sapi
#endif // SANDBOXED_API_UTIL_STATUS_MATCHERS_H_