1HEADmaster

1 files changed, 135 insertions, 0 deletions

diff --git a/utils/unittest/unittest.cpp b/utils/unittest/unittest.cpp
new file mode 100644
index 0000000..31b7101
--- /dev/null
+++ b/utils/unittest/unittest.cpp
@@ -0,0 +1,135 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: Unit test program
+//
+// $NoKeywords: $
+//=============================================================================//
+
+#include "unitlib/unitlib.h"
+#include "appframework/iappsystemgroup.h"
+#include "appframework/appframework.h"
+#include "tier0/dbg.h"
+#include <stdio.h>
+#include <windows.h>
+#include "vstdlib/iprocessutils.h"
+#include "tier1/interface.h"
+#include "vstdlib/cvar.h"
+
+#pragma warning (disable:4100)
+
+SpewRetval_t UnitTestSpew( SpewType_t type, char const *pMsg )
+{
+	switch( type )
+	{
+	case 	SPEW_WARNING:
+		printf( "UnitTest Warning:\n" );
+		break;
+	case	SPEW_ASSERT:
+		printf( "UnitTest Assert:\n" );
+		break;
+	case	SPEW_ERROR:
+		printf( "UnitTest Error:\n" );
+		break;
+	}
+	printf( "%s", pMsg );
+	OutputDebugString( pMsg );
+
+	if ( Sys_IsDebuggerPresent() )
+		return ( type == SPEW_ASSERT || type == SPEW_ERROR ) ? SPEW_DEBUGGER : SPEW_CONTINUE;
+	return SPEW_CONTINUE;
+}
+
+
+//-----------------------------------------------------------------------------
+// The application object
+//-----------------------------------------------------------------------------
+class CUnitTestApp : public CDefaultAppSystemGroup<CSteamAppSystemGroup>
+{
+public:
+	// Methods of IApplication
+	virtual bool Create();
+	virtual int Main();
+	virtual void Destroy();
+
+private:
+};
+
+DEFINE_CONSOLE_STEAM_APPLICATION_OBJECT( CUnitTestApp );
+
+
+//-----------------------------------------------------------------------------
+// The application object
+//-----------------------------------------------------------------------------
+bool CUnitTestApp::Create()
+{
+	// Install a special Spew handler that ignores all assertions and lets us
+	// run for as long as possible
+	SpewOutputFunc( UnitTestSpew );
+
+	// FIXME: This list of dlls should come from the unittests themselves
+	AppSystemInfo_t appSystems[] = 
+	{
+		{ "vstdlib.dll",			PROCESS_UTILS_INTERFACE_VERSION },
+		{ "", "" }	// Required to terminate the list
+	};
+
+	if ( !AddSystems( appSystems ) ) 
+		return false;
+
+	// Very simple... just iterate over all .DLLs in the current directory 
+	// see if they export UNITTEST_INTERFACE_VERSION. If not, then unload them
+	// just as quick.
+
+	// We may want to make this more sophisticated, giving it a search path,
+	// or giving test DLLs special extensions, or statically linking the test DLLs
+	// to this program.
+
+	WIN32_FIND_DATA findFileData;
+	HANDLE hFind= FindFirstFile("*.dll", &findFileData);
+
+	while (hFind != INVALID_HANDLE_VALUE)
+	{
+		CSysModule* hLib = Sys_LoadModule(findFileData.cFileName);
+		if ( hLib )
+		{
+			CreateInterfaceFn factory = Sys_GetFactory( hLib );
+			if ( factory && factory( UNITTEST_INTERFACE_VERSION, NULL ) )
+			{
+				AppModule_t module = LoadModule( factory );
+				AddSystem( module, UNITTEST_INTERFACE_VERSION );
+			}
+			else
+			{
+				Sys_UnloadModule( hLib );
+			}
+		}
+
+		if (!FindNextFile( hFind, &findFileData ))
+			break;
+	}
+
+	return true;
+}
+
+void CUnitTestApp::Destroy()
+{
+}
+
+
+//-----------------------------------------------------------------------------
+// The application object
+//-----------------------------------------------------------------------------
+int CUnitTestApp::Main()
+{
+	printf( "Valve Software - unittest.exe (%s)\n", __DATE__ );
+
+	int nTestCount = UnitTestCount();
+	for ( int i = 0; i < nTestCount; ++i )
+	{
+		ITestCase* pTestCase = GetUnitTest(i);
+		printf("Starting test %s....\n", pTestCase->GetName() );
+		pTestCase->RunTest();
+	}
+
+	return 0;
+}