ostream_event_listener.h

Go to the documentation of this file.

#pragma once
#include "__console_color.h"
#include "event_listener.h"
#include <ostream>

namespace tunit {
  class ostream_event_listener : public tunit::event_listener {
  public:
    ostream_event_listener(std::ostream& os) noexcept : os_(os) {}
    
    void on_test_aborted(const tunit::test_event_args& e) const override {
      this->event_listener::on_test_aborted(e);
      this->os_ << __foreground_color(__console_color::magenta);
      this->os_ << "  ABORTED ";
      this->os_ << __reset_color();
      this->os_ << e.test_class().name() << "." << e.test().name();
      if (tunit::settings::default_settings().show_duration())
        this->os_ << " (" << e.test().elapsed_time().count() << " ms total)";
      this->os_ << std::endl;
      if (e.test().message() != "")
        this->os_ << "    " << e.test().message() << std::endl;
      if (e.test().expect() != "")
        this->os_ << "    Expected: " << e.test().expect() << std::endl;
      if (e.test().actual() != "")
        this->os_ << "    But was:  " << e.test().actual() << std::endl;
      if ((e.test().expect() != "" || e.test().actual() != "") && e.test().line_info() != tunit::line_info::empty()) {
        this->os_ << "    Stack Trace: in " << e.test().line_info().file_path();
        if (e.test().line_info().line_number() != 0) this->os_ << ":" << e.test().line_info().line_number();
      }
      this->os_ << std::endl;
   }

    void on_test_failed(const tunit::test_event_args& e) const override {
      this->event_listener::on_test_failed(e);
      this->os_ << __foreground_color(__console_color::red);
      this->os_ << "  FAILED  ";
      this->os_ << __reset_color();
      this->os_ << e.test_class().name() << "." << e.test().name();
      if (tunit::settings::default_settings().show_duration())
        this->os_ << " (" << e.test().elapsed_time().count() << " ms total)";
      this->os_ << std::endl;
      if (e.test().message() != "")
        this->os_ << "    " << e.test().message() << std::endl;
      if (e.test().expect() != "")
        this->os_ << "    Expected: " << e.test().expect() << std::endl;
      if (e.test().actual() != "")
        this->os_ << "    But was:  " << e.test().actual() << std::endl;
      if (e.test().line_info() != tunit::line_info::empty()) {
        this->os_ << "    Stack Trace: in " << e.test().line_info().file_path();
        if (e.test().line_info().line_number() != 0) this->os_ << ":" << e.test().line_info().line_number();
      }
      this->os_ << std::endl;
    }
    
    void on_test_ignored(const tunit::test_event_args& e) const override {
      this->event_listener::on_test_ignored(e);
      this->os_ << __foreground_color(__console_color::yellow);
      this->os_ << "  IGNORED ";
      this->os_ << __reset_color();
      this->os_ << e.test_class().name() << "." << e.test().name();
      if (tunit::settings::default_settings().show_duration())
        this->os_ << " (" << e.test().elapsed_time().count() << " ms total)";
      this->os_ << std::endl;
      if (e.test().message() != "")
        this->os_ << "    " << e.test().message() << std::endl;
    }

    void on_test_succeed(const tunit::test_event_args& e) const override {
      this->event_listener::on_test_succeed(e);
      this->os_ << __foreground_color(__console_color::green);
      this->os_ << "  SUCCEED ";
      this->os_ << __reset_color();
      this->os_ << e.test_class().name() << "." << e.test().name();
      if (tunit::settings::default_settings().show_duration())
        this->os_ << " (" << e.test().elapsed_time().count() << " ms total)";
      this->os_ << std::endl;
      if (e.test().message() != "")
        this->os_ << "    " << e.test().message() << std::endl;
    }
    
    void on_unit_test_end(const tunit::tunit_event_args& e) const override {
      this->event_listener::on_unit_test_end(e);
      this->os_ << std::endl;
      this->os_ << "Test results:" << std::endl;
      if (e.unit_test().succeed_test_count()) {
        this->os_ << __foreground_color(__console_color::green);
        this->os_ << "  SUCCEED ";
        this->os_ << __reset_color();
        this->os_ << e.unit_test().succeed_test_count() << " test" << (e.unit_test().succeed_test_count() < 2 ? "" : "s") << "." << std::endl;
      }
      if (e.unit_test().aborted_test_count()) {
        this->os_ << __foreground_color(__console_color::magenta);
        this->os_ << "  ABORTED ";
        this->os_ << __reset_color();
        this->os_ << e.unit_test().aborted_test_count() << " test" << (e.unit_test().aborted_test_count() < 2 ? "" : "s") << "." << std::endl;
      }
      if (e.unit_test().failed_test_count()) {
        this->os_ << __foreground_color(__console_color::red);
        this->os_ << "  FAILED  ";
        this->os_ << __reset_color();
        this->os_ << e.unit_test().failed_test_count() << " test" << (e.unit_test().failed_test_count() < 2 ? "" : "s") << "." << std::endl;
      }
      if (e.unit_test().ignored_test_count()) {
        this->os_ << __foreground_color(__console_color::yellow);
        this->os_ << "  IGNORED ";
        this->os_ << __reset_color();
        this->os_ << e.unit_test().ignored_test_count() << " test" << (e.unit_test().ignored_test_count() < 2 ? "" : "s") << "." << std::endl;
      }
      
      this->os_ << "End " << e.unit_test().test_count() << " test" << (e.unit_test().test_count() < 2 ? "" : "s") << " from " << e.unit_test().test_cases_count() << " test case" << (e.unit_test().test_cases_count() < 2 ? "" : "s") << " ran.";
      if (tunit::settings::default_settings().show_duration())
        this->os_ << " (" << e.unit_test().elapsed_time().count() << " ms total)";
      this->os_ << std::endl << std::endl;
    }
    
    void on_unit_test_start(const tunit::tunit_event_args& e) const override {
      if (e.unit_test().repeat_tests()) {
        this->os_ << __foreground_color(__console_color::cyan);
        this->os_ << "Repeating all tests (iteration " << e.unit_test().repeat_iteration() << ") . . ." << std::endl << std::endl;
        this->os_ << __reset_color();
      }
      this->os_ << "Start " << e.unit_test().test_count() << " test" << (e.unit_test().test_count() < 2 ? "" : "s") << " from " << e.unit_test().test_cases_count() << " test case" << (e.unit_test().test_cases_count() < 2 ? "" : "s") << std::endl;
      this->os_ << "Run tests:" << std::endl;
      this->event_listener::on_unit_test_start(e);
    }
    
  private:
    std::ostream& os_;
  };
}