split test registry code out

internal/catch_test_case_registry_impl.hpp

@@ -0,0 +1,111 @@
+/*
+ *  catch_test_case_registry_impl.hpp
+ *  Catch
+ *
+ *  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_hub.h"
+
+#include <vector>
+#include <set>
+
+namespace Catch
+{
+    class TestRegistry : public ITestCaseRegistry
+    {
+    public:
+        
+        virtual void registerTest( const TestCaseInfo& testInfo )
+        {
+            if( m_functions.find( testInfo ) == m_functions.end() )
+            {
+                m_functions.insert( testInfo );
+                m_functionsInOrder.push_back( testInfo );
+            }
+        }
+        
+        virtual const std::vector<TestCaseInfo>& getAllTests() const
+        {
+            return m_functionsInOrder;
+        }
+        
+    private:
+        
+        std::set<TestCaseInfo> m_functions;
+        std::vector<TestCaseInfo> m_functionsInOrder;
+    };
+    
+    typedef void(*TestFunction)();
+    
+    struct FreeFunctionTestCase : ITestCase
+    {
+        FreeFunctionTestCase( TestFunction fun )
+        : fun( fun )
+        {}
+        
+        virtual void invoke() const
+        {
+            fun();
+        }
+        
+        virtual ITestCase* clone() const
+        {
+            return new FreeFunctionTestCase( fun );
+        }
+        
+        virtual bool operator == ( const ITestCase& other ) const
+        {
+            const FreeFunctionTestCase* ffOther = dynamic_cast<const FreeFunctionTestCase*> ( &other );
+            return ffOther && fun == ffOther->fun;
+        }
+        
+        virtual bool operator < ( const ITestCase& other ) const
+        {
+            const FreeFunctionTestCase* ffOther = dynamic_cast<const FreeFunctionTestCase*> ( &other );
+            return ffOther && fun < ffOther->fun;
+        }
+        
+    private:
+        TestFunction fun;
+    };
+        
+    ///////////////////////////////////////////////////////////////////////////
+    ///////////////////////////////////////////////////////////////////////////
+    
+    ///////////////////////////////////////////////////////////////////////////
+    AutoReg::AutoReg
+    (
+     TestFunction function, 
+     const char* name,
+     const char* description
+     )
+    {
+        registerTestCase( new FreeFunctionTestCase( function ), name, description );
+    }    
+    
+    ///////////////////////////////////////////////////////////////////////////
+    AutoReg::~AutoReg
+    ()
+    {
+    }
+    
+    ///////////////////////////////////////////////////////////////////////////
+    void AutoReg::registerTestCase
+    (
+     ITestCase* testCase, 
+     const char* name, 
+     const char* description 
+     )
+    {
+        Hub::getTestCaseRegistry().registerTest( TestCaseInfo( testCase, name, description ) );
+    }
+    
+} // end namespace Catch
+