iutest_progress_printer.hpp

[詳解]

//======================================================================
//-----------------------------------------------------------------------
//-----------------------------------------------------------------------
//======================================================================
#ifndef INCG_IRIS_IUTEST_PROGRESS_PRINTER_HPP_D31CE626_EA82_4265_B2B6_50EC5C63CFB6_
#define INCG_IRIS_IUTEST_PROGRESS_PRINTER_HPP_D31CE626_EA82_4265_B2B6_50EC5C63CFB6_
 
//======================================================================
// include
#include "../iutest_core.hpp"
 
namespace iutest
{
 
//======================================================================
// class
class ProgressPrintListener : public EmptyTestEventListener
{
public:
    ProgressPrintListener() : m_should_run_num(0), m_ran_num(0) {}
public:
    virtual void OnTestProgramStart(const UnitTest& test) IUTEST_CXX_OVERRIDE;
    virtual void OnTestIterationStart(const UnitTest& test
                                    , int iteration) IUTEST_CXX_OVERRIDE;
    virtual void OnTestEnd(const TestInfo& test_info) IUTEST_CXX_OVERRIDE;
 
private:
    int m_should_run_num;
    int m_ran_num;
 
public:
    static TestEventListener* SetUp()
    {
        TestEventListeners& listeners = UnitTest::GetInstance()->listeners();
        delete listeners.Release(listeners.default_result_printer());
        TestEventListener* p = new ProgressPrintListener;
        listeners.Append(p);
        return p;
    }
};
 
inline void ProgressPrintListener::OnTestProgramStart(const UnitTest& test)
{
    IUTEST_UNUSED_VAR(test);
 
    detail::iuConsole::output("0%%   10   20   30   40   50   60   70   80   90   100%%\n");
    detail::iuConsole::output("|----|----|----|----|----|----|----|----|----|----|   \n");
}
inline void ProgressPrintListener::OnTestIterationStart(const UnitTest& test
                                , int iteration)
{
    IUTEST_UNUSED_VAR(iteration);
 
    m_should_run_num = test.test_to_run_count();
    m_ran_num = 0;
}
inline void ProgressPrintListener::OnTestEnd(const TestInfo& test_info)
{
    IUTEST_UNUSED_VAR(test_info);
 
    ++m_ran_num;
    const int star_count = 51;
    const int n = (m_ran_num * star_count) / m_should_run_num;
    char progress[star_count + 2] = { 0 };
    for( int i=0; i < n; ++i )
    {
        progress[i] = '*';
    }
    progress[n] = m_ran_num == m_should_run_num ? '\n' : '\r';
    progress[n + 1] = '\0';
    detail::iuConsole::output("%s", progress);
}
 
}   // end of namespace iutest
 
#endif // INCG_IRIS_IUTEST_PROGRESS_PRINTER_HPP_D31CE626_EA82_4265_B2B6_50EC5C63CFB6_