diff options
Diffstat (limited to 'test/utils/remote_execution/main-remote-execution.cpp')
-rwxr-xr-x | test/utils/remote_execution/main-remote-execution.cpp | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/test/utils/remote_execution/main-remote-execution.cpp b/test/utils/remote_execution/main-remote-execution.cpp new file mode 100755 index 00000000..3f435dcc --- /dev/null +++ b/test/utils/remote_execution/main-remote-execution.cpp @@ -0,0 +1,50 @@ +#define CATCH_CONFIG_RUNNER +#include <chrono> +#include <cstdlib> +#include <iostream> +#include <sstream> +#include <thread> + +#include "catch2/catch.hpp" +#include "src/buildtool/execution_api/remote/config.hpp" +#include "test/utils/logging/log_config.hpp" +#include "test/utils/test_env.hpp" + +namespace { + +void wait_for_grpc_to_shutdown() { + // grpc_shutdown_blocking(); // not working + std::this_thread::sleep_for(std::chrono::seconds(1)); +} + +/// \brief Configure remote execution from test environment. In case the +/// environment variable is malformed, we write a message and stop execution. +/// \returns true If remote execution was successfully configured. +[[nodiscard]] auto ConfigureRemoteExecution() -> bool { + auto address = ReadRemoteAddressFromEnv(); + auto& config = RemoteExecutionConfig::Instance(); + if (address and not config.SetAddress(*address)) { + Logger::Log(LogLevel::Error, "parsing address '{}' failed.", *address); + std::exit(EXIT_FAILURE); + } + return config.IsValidAddress(); +} + +} // namespace + +auto main(int argc, char* argv[]) -> int { + ConfigureLogging(); + + // In case remote execution address is not valid, we skip tests. This is in + // order to avoid tests being dependent on the environment. + if (not ConfigureRemoteExecution()) { + return EXIT_SUCCESS; + } + + int result = Catch::Session().run(argc, argv); + + // valgrind fails if we terminate before grpc's async shutdown threads exit + wait_for_grpc_to_shutdown(); + + return result; +} |