Files
scylladb/test/lib/test_utils.cc
Pavel Emelyanov fe70333c19 test: Auto-skip object-storage test cases if run from shell
In case an sstable unit test case is run individually, it would fail
with exception saying that S3_... environment is not set. It's better to
skip the test-case rather than fail. If someone wants to run it from
shell, it will have to prepare S3 server (minio/AWS public bucket) and
provide proper environment for the test-case.

refs: #13569

Signed-off-by: Pavel Emelyanov <xemul@scylladb.com>

Closes #13755
2023-05-04 14:15:18 +03:00

86 lines
2.4 KiB
C++

/*
* Copyright (C) 2020-present ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#include "test/lib/test_utils.hh"
#include <boost/range/adaptor/map.hpp>
#include <boost/range/algorithm/sort.hpp>
#include <seastar/core/print.hh>
#include <seastar/util/backtrace.hh>
#include "test/lib/log.hh"
#include "test/lib/simple_schema.hh"
#include "seastarx.hh"
#include <random>
namespace tests {
namespace {
std::string format_msg(std::string_view test_function_name, bool ok, std::source_location sl, std::string_view msg) {
return fmt::format("{}(): {} @ {}() {}:{:d}{}{}", test_function_name, ok ? "OK" : "FAIL", sl.function_name(), sl.file_name(), sl.line(), msg.empty() ? "" : ": ", msg);
}
}
bool do_check(bool condition, std::source_location sl, std::string_view msg) {
if (condition) {
testlog.trace("{}", format_msg(__FUNCTION__, condition, sl, msg));
} else {
testlog.error("{}", format_msg(__FUNCTION__, condition, sl, msg));
}
return condition;
}
void do_require(bool condition, std::source_location sl, std::string_view msg) {
if (condition) {
testlog.trace("{}", format_msg(__FUNCTION__, condition, sl, msg));
} else {
auto formatted_msg = format_msg(__FUNCTION__, condition, sl, msg);
testlog.error("{}", formatted_msg);
throw_with_backtrace<std::runtime_error>(std::move(formatted_msg));
}
}
void fail(std::string_view msg, std::source_location sl) {
throw_with_backtrace<std::runtime_error>(format_msg(__FUNCTION__, false, sl, msg));
}
extern boost::test_tools::assertion_result has_scylla_test_env(boost::unit_test::test_unit_id) {
if (::getenv("SCYLLA_TEST_ENV")) {
return true;
}
testlog.info("Test environment is not configured. "
"Check test/pylib/minio_server.py for an example of how to configure the environment for it to run.");
return false;
}
}
sstring make_random_string(size_t size) {
static thread_local std::default_random_engine rng;
std::uniform_int_distribution<char> dist;
sstring str = uninitialized_string(size);
for (auto&& b : str) {
b = dist(rng);
}
return str;
}
sstring make_random_numeric_string(size_t size) {
static thread_local std::default_random_engine rng;
std::uniform_int_distribution<char> dist('0', '9');
sstring str = uninitialized_string(size);
for (auto&& b : str) {
b = dist(rng);
}
return str;
}