catch2/include/internal/catch_test_case_registry_impl.hpp

117 lines
4.1 KiB
C++
Raw Normal View History

2011-01-07 11:22:24 +01:00
/*
* Created by Phil on 7/1/2011
* Copyright 2010 Two Blue Cubes Ltd. All rights reserved.
*
* Distributed under the Boost Software License, Version 1.0. (See accompanying
* file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
*/
#include "catch_test_registry.hpp"
#include "catch_test_case_info.hpp"
#include "catch_context.h"
2011-01-07 11:22:24 +01:00
#include <vector>
#include <set>
#include <sstream>
#include <iostream>
2011-01-07 11:22:24 +01:00
2012-05-16 09:02:20 +02:00
namespace Catch {
2012-05-10 22:46:46 +02:00
class TestRegistry : public ITestCaseRegistry {
2011-01-07 11:22:24 +01:00
public:
2012-05-10 22:46:46 +02:00
TestRegistry() : m_unnamedCount( 0 ) {}
2011-01-07 11:22:24 +01:00
2012-05-10 22:46:46 +02:00
virtual void registerTest( const TestCaseInfo& testInfo ) {
if( testInfo.getName() == "" ) {
std::ostringstream oss;
oss << testInfo.getName() << "unnamed/" << ++m_unnamedCount;
return registerTest( TestCaseInfo( testInfo, oss.str() ) );
}
2012-05-10 22:46:46 +02:00
if( m_functions.find( testInfo ) == m_functions.end() ) {
2011-01-07 11:22:24 +01:00
m_functions.insert( testInfo );
m_functionsInOrder.push_back( testInfo );
}
2012-05-10 22:46:46 +02:00
else {
const TestCaseInfo& prev = *m_functions.find( testInfo );
std::cerr << "error: TEST_CASE( \"" << testInfo.getName() << "\" ) already defined.\n"
<< "\tFirst seen at " << SourceLineInfo( prev.getLineInfo() ) << "\n"
<< "\tRedefined at " << SourceLineInfo( testInfo.getLineInfo() ) << std::endl;
exit(1);
}
2011-01-07 11:22:24 +01:00
}
2012-05-10 22:46:46 +02:00
virtual const std::vector<TestCaseInfo>& getAllTests() const {
2011-01-07 11:22:24 +01:00
return m_functionsInOrder;
}
2012-05-10 22:46:46 +02:00
virtual std::vector<TestCaseInfo> getMatchingTestCases( const std::string& rawTestSpec ) {
TestSpec testSpec( rawTestSpec );
std::vector<TestCaseInfo> testList;
std::vector<TestCaseInfo>::const_iterator it = m_functionsInOrder.begin();
std::vector<TestCaseInfo>::const_iterator itEnd = m_functionsInOrder.end();
2012-05-10 22:46:46 +02:00
for(; it != itEnd; ++it ) {
if( testSpec.matches( it->getName() ) ) {
testList.push_back( *it );
}
}
return testList;
}
2011-01-07 11:22:24 +01:00
private:
std::set<TestCaseInfo> m_functions;
std::vector<TestCaseInfo> m_functionsInOrder;
size_t m_unnamedCount;
2011-01-07 11:22:24 +01:00
};
2011-01-28 19:56:26 +01:00
///////////////////////////////////////////////////////////////////////////
2011-01-07 11:22:24 +01:00
2012-05-10 22:46:46 +02:00
class FreeFunctionTestCase : public ITestCase {
public:
FreeFunctionTestCase( TestFunction fun ) : m_fun( fun ) {}
virtual void invoke() const {
2011-01-31 11:10:20 +01:00
m_fun();
2011-01-07 11:22:24 +01:00
}
2012-05-10 22:46:46 +02:00
virtual ITestCase* clone() const {
2011-01-31 11:10:20 +01:00
return new FreeFunctionTestCase( m_fun );
2011-01-07 11:22:24 +01:00
}
2012-05-10 22:46:46 +02:00
virtual bool operator == ( const ITestCase& other ) const {
2011-01-07 11:22:24 +01:00
const FreeFunctionTestCase* ffOther = dynamic_cast<const FreeFunctionTestCase*> ( &other );
2011-01-31 11:10:20 +01:00
return ffOther && m_fun == ffOther->m_fun;
2011-01-07 11:22:24 +01:00
}
2012-05-10 22:46:46 +02:00
virtual bool operator < ( const ITestCase& other ) const {
2011-01-07 11:22:24 +01:00
const FreeFunctionTestCase* ffOther = dynamic_cast<const FreeFunctionTestCase*> ( &other );
2011-01-31 11:10:20 +01:00
return ffOther && m_fun < ffOther->m_fun;
2011-01-07 11:22:24 +01:00
}
private:
2011-01-31 11:10:20 +01:00
TestFunction m_fun;
2011-01-07 11:22:24 +01:00
};
///////////////////////////////////////////////////////////////////////////
2012-05-10 22:46:46 +02:00
AutoReg::AutoReg( TestFunction function,
const char* name,
const char* description,
const SourceLineInfo& lineInfo ) {
registerTestCase( new FreeFunctionTestCase( function ), name, description, lineInfo );
2011-01-07 11:22:24 +01:00
}
2012-05-10 22:46:46 +02:00
AutoReg::~AutoReg() {}
2011-01-07 11:22:24 +01:00
2012-05-10 22:46:46 +02:00
void AutoReg::registerTestCase( ITestCase* testCase,
const char* name,
const char* description,
const SourceLineInfo& lineInfo ) {
getCurrentContext().getTestCaseRegistry().registerTest( TestCaseInfo( testCase, name, description, lineInfo ) );
2011-01-07 11:22:24 +01:00
}
} // end namespace Catch