34738d9c90
This CL introduces TestNetLogManager that dumps NetLog events for net_unittests when `--log-net-log` command line flag is specified. Specifically: * When a file path is given, use FileNetLogObserver and dump events to the file. * When no file path is given, use VLOG(1) to dump events. Bug: 336167322 Change-Id: I9385d134ade4e28feef63567903fffbbf1f3c94b Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5884794 Reviewed-by: Adam Rice <ricea@chromium.org> Commit-Queue: Kenichi Ishibashi <bashi@chromium.org> Cr-Commit-Position: refs/heads/main@{#1370969}
94 lines
3.1 KiB
C++
94 lines
3.1 KiB
C++
// Copyright 2024 The Chromium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "net/test/test_net_log_manager.h"
|
|
|
|
#include "base/command_line.h"
|
|
#include "base/files/file.h"
|
|
#include "base/logging.h"
|
|
#include "base/memory/raw_ptr.h"
|
|
#include "base/run_loop.h"
|
|
#include "base/strings/utf_string_conversions.h"
|
|
#include "base/values.h"
|
|
#include "net/log/file_net_log_observer.h"
|
|
#include "net/log/net_log.h"
|
|
#include "net/log/net_log_util.h"
|
|
|
|
namespace net {
|
|
|
|
// A simple NetLog::ThreadSafeObserver that dumps NetLog entries to VLOG.
|
|
class TestNetLogManager::VlogNetLogObserver
|
|
: public NetLog::ThreadSafeObserver {
|
|
public:
|
|
VlogNetLogObserver(NetLog* net_log, NetLogCaptureMode capture_mode)
|
|
: net_log_(net_log) {
|
|
LOG_IF(INFO, !VLOG_IS_ON(1))
|
|
<< "Use --vmodule=test_net_log_manager=1 to see NetLog messages";
|
|
net_log_->AddObserver(this, capture_mode);
|
|
}
|
|
|
|
VlogNetLogObserver(const VlogNetLogObserver&) = delete;
|
|
VlogNetLogObserver& operator=(const VlogNetLogObserver&) = delete;
|
|
|
|
~VlogNetLogObserver() override { net_log_->RemoveObserver(this); }
|
|
|
|
void OnAddEntry(const NetLogEntry& entry) override {
|
|
VLOG(1) << "NetLog: id=" << entry.source.id
|
|
<< " source=" << NetLog::SourceTypeToString(entry.source.type)
|
|
<< "\n"
|
|
<< "event=" << NetLogEventTypeToString(entry.type)
|
|
<< " phase=" << NetLog::EventPhaseToString(entry.phase) << "\n"
|
|
<< entry.params.DebugString();
|
|
}
|
|
|
|
private:
|
|
const raw_ptr<NetLog> net_log_;
|
|
};
|
|
|
|
TestNetLogManager::TestNetLogManager(NetLog* net_log,
|
|
NetLogCaptureMode capture_mode) {
|
|
const base::CommandLine* command_line =
|
|
base::CommandLine::ForCurrentProcess();
|
|
base::FilePath log_file_path =
|
|
command_line->GetSwitchValuePath(kLogNetLogSwitch);
|
|
if (log_file_path.empty()) {
|
|
vlog_net_log_observer_ =
|
|
std::make_unique<VlogNetLogObserver>(net_log, capture_mode);
|
|
return;
|
|
}
|
|
|
|
base::File file = base::File(
|
|
log_file_path, base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE);
|
|
if (!file.IsValid()) {
|
|
return;
|
|
}
|
|
|
|
auto constants = std::make_unique<base::Value::Dict>(GetNetConstants());
|
|
base::Value::Dict client_info;
|
|
client_info.Set("name", "net_unittests");
|
|
base::CommandLine::StringType command_line_string =
|
|
command_line->GetCommandLineString();
|
|
#if BUILDFLAG(IS_WIN)
|
|
client_info.Set("command_line", base::WideToUTF8(command_line_string));
|
|
#else
|
|
client_info.Set("command_line", command_line_string);
|
|
#endif
|
|
constants->Set("clientInfo", std::move(client_info));
|
|
|
|
file_net_log_observer_ = FileNetLogObserver::CreateUnboundedPreExisting(
|
|
std::move(file), capture_mode, std::move(constants));
|
|
file_net_log_observer_->StartObserving(net_log);
|
|
}
|
|
|
|
TestNetLogManager::~TestNetLogManager() {
|
|
if (file_net_log_observer_) {
|
|
base::RunLoop run_loop;
|
|
file_net_log_observer_->StopObserving(/*polled_data=*/nullptr,
|
|
run_loop.QuitClosure());
|
|
run_loop.Run();
|
|
}
|
|
}
|
|
|
|
} // namespace net
|