Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

182 lines
4.4 KiB

5 years ago
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: Unit test program
//
// $NoKeywords: $
//=============================================================================//
#include "unitlib/unitlib.h"
#include "appframework/IAppSystemGroup.h"
#include "appframework/AppFramework.h"
5 years ago
#include "tier0/dbg.h"
#include <stdio.h>
#include "vstdlib/iprocessutils.h"
#include "tier1/interface.h"
#include "vstdlib/cvar.h"
#if defined(POSIX)
#include <dirent.h>
#elif defined(WIN32)
5 years ago
#pragma warning (disable:4100)
#include <windows.h>
#endif
static int g_TestResult = 0;
5 years ago
SpewRetval_t UnitTestSpew( SpewType_t type, char const *pMsg )
{
switch( type )
{
case SPEW_WARNING:
5 years ago
printf( "UnitTest Warning:\n" );
break;
case SPEW_ASSERT:
5 years ago
printf( "UnitTest Assert:\n" );
g_TestResult = 1;
5 years ago
break;
case SPEW_ERROR:
5 years ago
printf( "UnitTest Error:\n" );
g_TestResult = 1;
5 years ago
break;
}
printf( "%s", pMsg );
fflush(stdout);
5 years ago
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.so", PROCESS_UTILS_INTERFACE_VERSION },
5 years ago
{ "", "" } // 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.
#ifdef WIN32
5 years ago
WIN32_FIND_DATA findFileData;
HANDLE hFind= FindFirstFile("tests/*.dll", &findFileData);
5 years ago
while (hFind != INVALID_HANDLE_VALUE)
{
static char path[2048];
snprintf(path, sizeof(path), "tests/%s", findFileData.cFileName);
CSysModule* hLib = Sys_LoadModule(path);
5 years ago
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;
}
#elif POSIX
DIR *d;
struct dirent *dir;
d = opendir("tests");
if (d)
{
while ((dir = readdir(d)) != NULL)
{
int len = strlen(dir->d_name);
if( len > 2 && strcmp(dir->d_name+len-strlen(DLL_EXT_STRING), DLL_EXT_STRING) == 0)
{
static char path[2048];
snprintf(path, sizeof(path), "tests/%s", dir->d_name);
CSysModule* hLib = Sys_LoadModule(path);
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 );
}
}
}
}
closedir(d);
}
#else
#error "Implement me!"
#endif
5 years ago
return true;
}
void CUnitTestApp::Destroy()
{
}
//-----------------------------------------------------------------------------
// The application object
//-----------------------------------------------------------------------------
int CUnitTestApp::Main()
{
printf( "Valve Software - unittest (%s)\n", __DATE__ );
5 years ago
int nTestCount = UnitTestCount();
for ( int i = 0; i < nTestCount; ++i )
{
ITestCase* pTestCase = GetUnitTest(i);
printf("Starting test %s....\n", pTestCase->GetName() );
pTestCase->RunTest();
}
return g_TestResult;
5 years ago
}