136 lines
3.4 KiB
C++
136 lines
3.4 KiB
C++
//========= 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;
|
|
}
|