mirror of
https://github.com/abseil/abseil-cpp.git
synced 2026-06-04 12:07:05 +08:00
4825ef4095
In C++20, this happens by default, but in C++17, it's left uninitialized if no logs are sent and StartCapturingLogs is never called. The destructor then makes an attempt to access it, which is both unsafe and not very useful while it is uninitialized. Also add a test that expects the appropriate CHECK fail when StartCapturingLogs is not called. PiperOrigin-RevId: 511865440 Change-Id: Ie23ff3f901e926761d5f487e10d33c21c3bd43d3
87 lines
2.8 KiB
C++
87 lines
2.8 KiB
C++
//
|
|
// Copyright 2022 The Abseil Authors.
|
|
//
|
|
// 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.
|
|
|
|
#include "absl/log/scoped_mock_log.h"
|
|
|
|
#include <atomic>
|
|
#include <string>
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "absl/base/config.h"
|
|
#include "absl/base/internal/raw_logging.h"
|
|
#include "absl/log/log_entry.h"
|
|
#include "absl/log/log_sink.h"
|
|
#include "absl/log/log_sink_registry.h"
|
|
#include "absl/strings/string_view.h"
|
|
|
|
namespace absl {
|
|
ABSL_NAMESPACE_BEGIN
|
|
|
|
ScopedMockLog::ScopedMockLog(MockLogDefault default_exp)
|
|
: sink_(this), is_capturing_logs_(false), is_triggered_(false) {
|
|
if (default_exp == MockLogDefault::kIgnoreUnexpected) {
|
|
// Ignore all calls to Log we did not set expectations for.
|
|
EXPECT_CALL(*this, Log).Times(::testing::AnyNumber());
|
|
} else {
|
|
// Disallow all calls to Log we did not set expectations for.
|
|
EXPECT_CALL(*this, Log).Times(0);
|
|
}
|
|
// By default Send mock forwards to Log mock.
|
|
EXPECT_CALL(*this, Send)
|
|
.Times(::testing::AnyNumber())
|
|
.WillRepeatedly([this](const absl::LogEntry& entry) {
|
|
is_triggered_.store(true, std::memory_order_relaxed);
|
|
Log(entry.log_severity(), std::string(entry.source_filename()),
|
|
std::string(entry.text_message()));
|
|
});
|
|
|
|
// By default We ignore all Flush calls.
|
|
EXPECT_CALL(*this, Flush).Times(::testing::AnyNumber());
|
|
}
|
|
|
|
ScopedMockLog::~ScopedMockLog() {
|
|
ABSL_RAW_CHECK(is_triggered_.load(std::memory_order_relaxed),
|
|
"Did you forget to call StartCapturingLogs()?");
|
|
|
|
if (is_capturing_logs_) StopCapturingLogs();
|
|
}
|
|
|
|
void ScopedMockLog::StartCapturingLogs() {
|
|
ABSL_RAW_CHECK(!is_capturing_logs_,
|
|
"StartCapturingLogs() can be called only when the "
|
|
"absl::ScopedMockLog object is not capturing logs.");
|
|
|
|
is_capturing_logs_ = true;
|
|
is_triggered_.store(true, std::memory_order_relaxed);
|
|
absl::AddLogSink(&sink_);
|
|
}
|
|
|
|
void ScopedMockLog::StopCapturingLogs() {
|
|
ABSL_RAW_CHECK(is_capturing_logs_,
|
|
"StopCapturingLogs() can be called only when the "
|
|
"absl::ScopedMockLog object is capturing logs.");
|
|
|
|
is_capturing_logs_ = false;
|
|
absl::RemoveLogSink(&sink_);
|
|
}
|
|
|
|
absl::LogSink& ScopedMockLog::UseAsLocalSink() {
|
|
is_triggered_.store(true, std::memory_order_relaxed);
|
|
return sink_;
|
|
}
|
|
|
|
ABSL_NAMESPACE_END
|
|
} // namespace absl
|