iutest_tap_printer.hpp

[詳解]

//======================================================================
//-----------------------------------------------------------------------
//-----------------------------------------------------------------------
//======================================================================
#ifndef INCG_IRIS_IUTEST_TAP_PRINTER_HPP_77055C2B_AAE1_4944_A61C_26C58B04B37B_
#define INCG_IRIS_IUTEST_TAP_PRINTER_HPP_77055C2B_AAE1_4944_A61C_26C58B04B37B_
 
//======================================================================
// include
#include "../iutest_core.hpp"
#include "../internal/iutest_log_stream.hpp"
#include "../internal/iutest_filepath.hpp"
 
namespace iutest
{
 
//======================================================================
// class
class TAPPrintListener : public EmptyTestEventListener
{
public:
    TAPPrintListener() {}
public:
    virtual void OnTestProgramStart(const UnitTest& test) IUTEST_CXX_OVERRIDE;
    virtual void OnTestProgramEnd(const UnitTest& test) IUTEST_CXX_OVERRIDE;
 
protected:
    void OnReportTestCase(detail::IOutStream* const stream, const TestCase& test_case, int top=1);
 
public:
    static TestEventListener* SetUp()
    {
        TestEventListeners& listeners = UnitTest::GetInstance()->listeners();
        delete listeners.Release(listeners.default_result_printer());
        TestEventListener* p = new TAPPrintListener;
        listeners.Append(p);
        return p;
    }
};
 
class TAPFileGeneratorListener : public TAPPrintListener
{
    ::std::string m_output_path;
public:
    explicit TAPFileGeneratorListener(const char* directory=NULL)
    {
        SetFilePath(directory);
    }
public:
    const ::std::string& GetFilePath() const { return m_output_path; }
private:
    void SetFilePath(const char* directory)
    {
        if( directory == NULL || *directory == '\0' )
        {
            m_output_path = internal::posix::GetCWD();
        }
        else
        {
            detail::iuFilePath path(directory);
            if( path.IsDirectory() )
            {
                m_output_path = directory;
            }
            else
            {
                m_output_path = path.RemoveFileName().ToString();
            }
        }
    }
public:
    virtual void OnTestProgramStart(const UnitTest& test) IUTEST_CXX_OVERRIDE
    {
        IUTEST_UNUSED_VAR(test);
    }
    virtual void OnTestRecordProperty(const TestProperty& test_property) IUTEST_CXX_OVERRIDE
    {
        IUTEST_UNUSED_VAR(test_property);
    }
    virtual void OnTestProgramEnd(const UnitTest& test) IUTEST_CXX_OVERRIDE;
 
public:
    static TestEventListener* SetUp()
    {
        TestEventListeners& listeners = UnitTest::GetInstance()->listeners();
        const ::std::string& output =  TestEnv::get_output_option();
        TestEventListener* p = new TAPFileGeneratorListener(output.c_str());
        listeners.Append(p);
        return p;
    }
};
 
inline void TAPPrintListener::OnTestProgramStart(const UnitTest& test)
{
    IUTEST_UNUSED_VAR(test);
    // フィルタリング
    if( TestFlag::IsEnableFlag(TestFlag::FILTERING_TESTS) )
    {
        detail::iuConsole::color_output(detail::iuConsole::yellow
            , "# Note: iutest filter = %s\n", TestEnv::test_filter());
    }
}
inline void TAPPrintListener::OnTestProgramEnd(const UnitTest& test)
{
    detail::LogStream stream;
 
    int number = 1;
    for( int k=0, count=test.total_test_case_count(); k < count; ++k )
    {
        const TestCase& test_case = *test.GetTestCase(k);
 
        OnReportTestCase(&stream, test_case, number);
 
        number += test_case.total_test_count();
    }
    detail::iuConsole::output("1..%d\n", number-1);
}
inline void TAPPrintListener::OnReportTestCase(detail::IOutStream* const stream, const TestCase& test_case, int top)
{
    stream->Printf("# %s started.\n", test_case.name());
 
    for( int i=0, test_count=test_case.total_test_count(); i < test_count; ++i )
    {
        const TestInfo* test_info = test_case.GetTestInfo(i);
        if( !test_info->should_run() )
        {
            // スキップ
            stream->Printf("ok %d # SKIP - %s\n", top+i, test_info->name());
        }
        else
        {
            if( test_info->HasFailure() )
            {
                stream->Printf("not ok %d - %s ", top+i, test_info->name() );
                for( int j=0, part_count=test_info->result()->total_part_count(); j < part_count; ++j )
                {
                    const TestPartResult& part = test_info->result()->GetTestPartResult(j);
                    ::std::string summary = part.make_message();
                    // 改行を置き換え
                    detail::StringReplace(summary, '\n', ", ");
                    stream->Printf("%s. ", summary.c_str() );
                }
                stream->Printf("\n");
            }
            else
            {
                stream->Printf("ok %d - %s\n", top+i, test_info->name() );
            }
        }
    }
 
    stream->Printf("# %s ended.\n", test_case.name());
}
 
inline void TAPFileGeneratorListener::OnTestProgramEnd(const UnitTest& test)
{
    for( int k=0, count=test.total_test_case_count(); k < count; ++k )
    {
        const TestCase& test_case = *test.GetTestCase(k);
 
        IUTEST_UNUSED_VAR(test);
        IFile* fp = detail::IFileSystem::New();
        if( fp == NULL )
        {
            return;
        }
 
        ::std::string filepath = m_output_path;
        ::std::string name = test_case.name();
        detail::StringReplace(name, '/', "_");
        filepath += detail::GetPathSeparator();
        filepath += name;
        filepath += ".tap";
 
        if( !fp->Open(filepath.c_str(), IFile::OpenWrite) )
        {
            fprintf(stderr, "Unable to open file \"%s\".\n", filepath.c_str());
            fflush(stderr);
        }
 
        OnReportTestCase(fp, test_case);
 
        fp->Printf("1..%d\n", test_case.total_test_count());
 
        fp->Close();
        detail::IFileSystem::Free(fp);
    }
}
 
}   // end of namespace iutest
 
#endif // INCG_IRIS_IUTEST_TAP_PRINTER_HPP_77055C2B_AAE1_4944_A61C_26C58B04B37B_