webrtc/test/testsupport/test_artifacts.cc
Dor Hen 4efc830e53 Provide test output path with OutputPathWithRandomDirectory 1/n
First commit in a series of commits to wire up the test output path utility that adds a random directory in the path, for problematic tests that run in concurrent execution environments.

Bug: webrtc:15833
Change-Id: I5e5b3940007be773d77dbbfc953efa810e4e3ea9
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/339522
Reviewed-by: Artem Titov <titovartem@webrtc.org>
Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org>
Reviewed-by: Harald Alvestrand <hta@webrtc.org>
Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org>
Cr-Commit-Position: refs/heads/main@{#41738}
2024-02-15 07:35:00 +00:00

74 lines
2.2 KiB
C++

/*
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "test/testsupport/test_artifacts.h"
#include <string.h>
#include "absl/flags/flag.h"
#include "absl/flags/parse.h"
#include "rtc_base/logging.h"
#include "rtc_base/system/file_wrapper.h"
#include "test/testsupport/file_utils.h"
namespace {
const std::string& DefaultArtifactPath() {
static const std::string path = webrtc::test::OutputPathWithRandomDirectory();
return path;
}
} // namespace
ABSL_FLAG(std::string,
test_artifacts_dir,
DefaultArtifactPath().c_str(),
"The output folder where test output should be saved.");
namespace webrtc {
namespace test {
bool GetTestArtifactsDir(std::string* out_dir) {
if (absl::GetFlag(FLAGS_test_artifacts_dir).empty()) {
RTC_LOG(LS_WARNING) << "No test_out_dir defined.";
return false;
}
*out_dir = absl::GetFlag(FLAGS_test_artifacts_dir);
return true;
}
bool WriteToTestArtifactsDir(const char* filename,
const uint8_t* buffer,
size_t length) {
if (absl::GetFlag(FLAGS_test_artifacts_dir).empty()) {
RTC_LOG(LS_WARNING) << "No test_out_dir defined.";
return false;
}
if (filename == nullptr || strlen(filename) == 0) {
RTC_LOG(LS_WARNING) << "filename must be provided.";
return false;
}
std::string full_path =
JoinFilename(absl::GetFlag(FLAGS_test_artifacts_dir), filename);
FileWrapper output = FileWrapper::OpenWriteOnly(full_path);
RTC_LOG(LS_INFO) << "Writing test artifacts in: " << full_path;
return output.is_open() && output.Write(buffer, length);
}
bool WriteToTestArtifactsDir(const char* filename, const std::string& content) {
return WriteToTestArtifactsDir(
filename, reinterpret_cast<const uint8_t*>(content.c_str()),
content.length());
}
} // namespace test
} // namespace webrtc