mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-05 03:36:08 +03:00
118 lines
4.5 KiB
C++
118 lines
4.5 KiB
C++
// Copyright 2014 The Chromium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
//
|
|
// Testing utilities that extend gtest.
|
|
|
|
#ifndef NET_TEST_GTEST_UTIL_H_
|
|
#define NET_TEST_GTEST_UTIL_H_
|
|
|
|
#include <string>
|
|
#include <string_view>
|
|
|
|
#include "base/test/mock_log.h"
|
|
#include "net/base/net_errors.h"
|
|
#include "net/test/scoped_disable_exit_on_dfatal.h"
|
|
#include "testing/gmock/include/gmock/gmock-matchers.h"
|
|
#include "testing/gmock/include/gmock/gmock.h"
|
|
#include "testing/gtest/include/gtest/gtest.h"
|
|
|
|
namespace net::test {
|
|
|
|
// A GMock matcher that checks whether the argument is the expected net::Error.
|
|
// On failure, the expected and actual net::Error names will be printed.
|
|
// Usage: EXPECT_THAT(foo(), IsError(net::ERR_INVALID_ARGUMENT));
|
|
MATCHER_P(IsError,
|
|
expected,
|
|
std::string(negation ? "not " : "") + net::ErrorToString(expected)) {
|
|
if (arg <= 0)
|
|
*result_listener << net::ErrorToString(arg);
|
|
return arg == expected;
|
|
}
|
|
|
|
// Shorthand for IsError(net::OK).
|
|
// Usage: EXPECT_THAT(foo(), IsOk());
|
|
MATCHER(IsOk,
|
|
std::string(negation ? "not " : "") + net::ErrorToString(net::OK)) {
|
|
if (arg <= 0)
|
|
*result_listener << net::ErrorToString(arg);
|
|
return arg == net::OK;
|
|
}
|
|
|
|
// A gMock matcher for std::string_view arguments.
|
|
// gMock's built-in HasSubstrMatcher does not work,
|
|
// because std::string_view cannot be implicitly converted to std::string.
|
|
class StringPieceHasSubstrMatcher {
|
|
public:
|
|
explicit StringPieceHasSubstrMatcher(const std::string& substring)
|
|
: substring_(substring) {}
|
|
StringPieceHasSubstrMatcher(const StringPieceHasSubstrMatcher&) = default;
|
|
StringPieceHasSubstrMatcher& operator=(const StringPieceHasSubstrMatcher&) =
|
|
default;
|
|
|
|
bool MatchAndExplain(std::string_view s,
|
|
::testing::MatchResultListener* listener) const {
|
|
return s.find(substring_) != std::string::npos;
|
|
}
|
|
|
|
// Describe what this matcher matches.
|
|
void DescribeTo(std::ostream* os) const {
|
|
*os << "has substring " << substring_;
|
|
}
|
|
|
|
void DescribeNegationTo(std::ostream* os) const {
|
|
*os << "has no substring " << substring_;
|
|
}
|
|
|
|
private:
|
|
std::string substring_;
|
|
};
|
|
|
|
// Internal implementation for the EXPECT_DFATAL and ASSERT_DFATAL
|
|
// macros. Do not use this directly.
|
|
#define GTEST_DFATAL_(statement, matcher, fail) \
|
|
do { \
|
|
::base::test::MockLog gtest_log; \
|
|
::net::test::ScopedDisableExitOnDFatal gtest_disable_exit; \
|
|
using ::testing::_; \
|
|
EXPECT_CALL(gtest_log, Log(_, _, _, _, _)) \
|
|
.WillRepeatedly(::testing::Return(false)); \
|
|
EXPECT_CALL(gtest_log, Log(::logging::LOGGING_DFATAL, _, _, _, matcher)) \
|
|
.Times(::testing::AtLeast(1)) \
|
|
.WillOnce(::testing::Return(false)); \
|
|
gtest_log.StartCapturingLogs(); \
|
|
{ statement; } \
|
|
gtest_log.StopCapturingLogs(); \
|
|
if (!testing::Mock::VerifyAndClear(>est_log)) \
|
|
fail(""); \
|
|
} while (false)
|
|
|
|
// The EXPECT_DFATAL and ASSERT_DFATAL macros are lightweight
|
|
// alternatives to EXPECT_DEBUG_DEATH and ASSERT_DEBUG_DEATH. They
|
|
// are appropriate for testing that your code logs a message at the
|
|
// DFATAL level.
|
|
//
|
|
// Unlike EXPECT_DEBUG_DEATH and ASSERT_DEBUG_DEATH, these macros
|
|
// execute the given statement in the current process, not a forked
|
|
// one. This works because we disable exiting the program for
|
|
// LOG(DFATAL). This makes the tests run more quickly.
|
|
//
|
|
// The _WITH() variants allow one to specify any matcher for the
|
|
// DFATAL log message, whereas the other variants assume a regex.
|
|
|
|
#define EXPECT_DFATAL_WITH(statement, matcher) \
|
|
GTEST_DFATAL_(statement, matcher, GTEST_NONFATAL_FAILURE_)
|
|
|
|
#define ASSERT_DFATAL_WITH(statement, matcher) \
|
|
GTEST_DFATAL_(statement, matcher, GTEST_FATAL_FAILURE_)
|
|
|
|
#define EXPECT_DFATAL(statement, regex) \
|
|
EXPECT_DFATAL_WITH(statement, ::testing::ContainsRegex(regex))
|
|
|
|
#define ASSERT_DFATAL(statement, regex) \
|
|
ASSERT_DFATAL_WITH(statement, ::testing::ContainsRegex(regex))
|
|
|
|
} // namespace net::test
|
|
|
|
#endif // NET_TEST_GTEST_UTIL_H_
|