lisyarus/psemek
1
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add tests output coloring

Browse source
This commit is contained in:
Nikita Lisitsa 2022-05-19 16:25:01 +03:00
parent 43c18eb4d2
commit fd336dfde2
1 changed files with 36 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
tools/test/source/main.cpp
View file

@ -10,6 +10,7 @@
#include <algorithm> #include <algorithm>
#include <cmath> #include <cmath>
#include <chrono> #include <chrono>
#include <cstdlib>
namespace psemek::test namespace psemek::test
{ {
@ -58,8 +59,34 @@ namespace psemek::test
} }
static bool terminal_has_color()
{
std::string_view term;
if (auto t = std::getenv("TERM"))
term = t;
return term == "linux" || term.starts_with("xterm");
}
int main(int argc, char ** argv) int main(int argc, char ** argv)
{ {
std::string_view bold;
std::string_view red;
std::string_view green;
std::string_view yellow;
std::string_view cyan;
std::string_view normal;
if (terminal_has_color())
{
bold = "\033[1m";
red = "\033[1;31m";
green = "\033[1;32m";
yellow = "\033[33m";
cyan = "\033[36m";
normal = "\033[0m";
}
std::set<std::string> tests; std::set<std::string> tests;
if (argc == 1) if (argc == 1)
{ {
@ -113,8 +140,8 @@ int main(int argc, char ** argv)
++i; ++i;
std::cout std::cout
<< '[' << std::setfill(' ') << std::setw(test_index_len) << std::right << i << '/' << test_count << "] " << bold << '[' << std::setfill(' ') << std::setw(test_index_len) << std::right << i << '/' << test_count << "] " << name
<< std::left << std::setfill('.') << std::setw(max_name_length + 5) << name; << normal << std::string(max_name_length + 5 - name.size(), '.');
psemek::test::context ctx; psemek::test::context ctx;
@ -127,7 +154,7 @@ int main(int argc, char ** argv)
{ {
psemek::test::tests[name](ctx); psemek::test::tests[name](ctx);
auto end = clock::now(); auto end = clock::now();
std::cout << "ok " << psemek::util::pretty(end - start, std::chrono::milliseconds{1}); std::cout << green << "ok " << normal << psemek::util::pretty(end - start, std::chrono::milliseconds{1});
auto filter = [](auto const & m){ return m.first >= psemek::log::level::info; }; auto filter = [](auto const & m){ return m.first >= psemek::log::level::info; };
if (std::count_if(sink_ptr->messages.begin(), sink_ptr->messages.end(), filter) == 1) if (std::count_if(sink_ptr->messages.begin(), sink_ptr->messages.end(), filter) == 1)
@ -142,19 +169,19 @@ int main(int argc, char ** argv)
catch (psemek::test::failure const & e) catch (psemek::test::failure const & e)
{ {
auto end = clock::now(); auto end = clock::now();
std::cout << "failure " << psemek::util::pretty(end - start, std::chrono::milliseconds{1}) << std::endl; std::cout << red << "failure " << normal << psemek::util::pretty(end - start, std::chrono::milliseconds{1}) << std::endl;
std::cout << indent << "Reason: " << e.message() << std::endl; std::cout << indent << "Reason: " << yellow << e.message() << normal << std::endl;
std::cout << indent << "Location: " << e.location() << std::endl; std::cout << indent << "Location: " << cyan << e.location() << normal << std::endl;
} }
catch (std::exception const & e) catch (std::exception const & e)
{ {
auto end = clock::now(); auto end = clock::now();
std::cout << "failure: " << e.what() << " " << psemek::util::pretty(end - start, std::chrono::milliseconds{1}) << std::endl; std::cout << red << "failure: " << e.what() << normal << " " << psemek::util::pretty(end - start, std::chrono::milliseconds{1}) << std::endl;
} }
catch (...) catch (...)
{ {
auto end = clock::now(); auto end = clock::now();
std::cout << "failure: (unknown exception) " << psemek::util::pretty(end - start, std::chrono::milliseconds{1}) << std::endl; std::cout << red << "failure: (unknown exception) " << normal << psemek::util::pretty(end - start, std::chrono::milliseconds{1}) << std::endl;
} }
sink = psemek::log::remove_sink(sink_ptr); sink = psemek::log::remove_sink(sink_ptr);
@ -183,7 +210,7 @@ int main(int argc, char ** argv)
auto all_end = clock::now(); auto all_end = clock::now();
std::cout << std::endl; std::cout << std::endl;
std::cout << (success == test_count ? "Success: " : "Failure: ") << success << '/' << test_count << " passed (" std::cout << (success == test_count ? green : red) << (success == test_count ? "Success: " : "Failure: ") << normal << success << '/' << test_count << " passed ("
<< psemek::util::pretty(all_end - all_start, std::chrono::milliseconds{1}) << ")" << std::endl; << psemek::util::pretty(all_end - all_start, std::chrono::milliseconds{1}) << ")" << std::endl;
if (success <= test_count) if (success <= test_count)