1
0
mirror of https://github.com/alliedmodders/metamod-source.git synced 2024-12-04 16:24:16 +01:00
HLMetaModOfficial/sourcehook/test/test4.cpp
Pavol Marko d0e26565dd Subtle bug fixed + tests now more resistant against compiler
optimizations

--HG--
extra : convert_revision : svn%3Ac2935e3e-5518-0410-8daf-afa5dab7d4e3/trunk%40472
2007-10-08 14:20:51 +00:00

316 lines
6.7 KiB
C++

#include <string>
#include "sourcehook_test.h"
#include "testevents.h"
// TEST4
// Tests of plugin management system
namespace
{
StateList g_States;
SourceHook::ISourceHook *g_SHPtr;
SourceHook::Plugin g_PLID;
MAKE_STATE(State_Func1_Called);
MAKE_STATE(State_Func2_Called);
MAKE_STATE(State_Func3_Called);
MAKE_STATE(State_Func1H_Called);
MAKE_STATE(State_Func2H_Called);
MAKE_STATE(State_Func3H_Called);
class Test
{
public:
virtual void Func1()
{
ADD_STATE(State_Func1_Called);
}
virtual void Func2()
{
ADD_STATE(State_Func2_Called);
}
virtual void Func3()
{
ADD_STATE(State_Func3_Called);
}
};
SH_DECL_HOOK0_void(Test, Func1, SH_NOATTRIB, 1);
SH_DECL_HOOK0_void(Test, Func2, SH_NOATTRIB, 1);
SH_DECL_HOOK0_void(Test, Func3, SH_NOATTRIB, 1);
void Handler_Func1()
{
ADD_STATE(State_Func1H_Called);
}
void Handler_Func2()
{
ADD_STATE(State_Func2H_Called);
}
void Handler_Func3()
{
ADD_STATE(State_Func3H_Called);
}
Test *MyTestFactory()
{
return new Test;
}
}
bool TestPlugSys(std::string &error)
{
GET_SHPTR(g_SHPtr);
g_PLID = 1;
Test *pInst = MyTestFactory();
CAutoPtrDestruction<Test> apd(pInst);
// 1)
// Add hooks, then issue a complete shutdown
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 1.1");
Test_CompleteShutdown(g_SHPtr);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1_Called,
new State_Func2_Called,
new State_Func3_Called,
NULL), "Part 1.2");
// 2)
// Add hooks from "different plugins", then shutdown the plugins
g_PLID = 1;
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
g_PLID = 2;
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
g_PLID = 3;
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
g_PLID = 1;
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
g_PLID = 2;
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
g_PLID = 3;
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 2.1");
// Unload plugins one by one
Test_UnloadPlugin(g_SHPtr, 3);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 2.3.1");
Test_UnloadPlugin(g_SHPtr, 2);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 2.4.1");
Test_UnloadPlugin(g_SHPtr, 1);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1_Called,
new State_Func2_Called,
new State_Func3_Called,
NULL), "Part 2.5.1");
// 3)
// Add hooks from "different plugins", then pause the plugins
g_PLID = 1;
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
g_PLID = 2;
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
g_PLID = 3;
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
g_PLID = 1;
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
g_PLID = 2;
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Test, Func3, pInst, SH_STATIC(Handler_Func3), false);
g_PLID = 3;
SH_ADD_HOOK(Test, Func1, pInst, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Test, Func2, pInst, SH_STATIC(Handler_Func2), true);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 3.1");
// Unload plugins one by one
Test_PausePlugin(g_SHPtr, 3);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 3.3.1");
Test_PausePlugin(g_SHPtr, 2);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 3.4.1");
Test_PausePlugin(g_SHPtr, 1);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1_Called,
new State_Func2_Called,
new State_Func3_Called,
NULL), "Part 3.5.1");
Test_UnpausePlugin(g_SHPtr, 1);
Test_UnpausePlugin(g_SHPtr, 2);
Test_UnpausePlugin(g_SHPtr, 3);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1H_Called,
new State_Func1_Called,
new State_Func2_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func2H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3H_Called,
new State_Func3_Called,
NULL), "Part 3.7");
// 4) Shutdown :)
Test_CompleteShutdown(g_SHPtr);
pInst->Func1();
pInst->Func2();
pInst->Func3();
CHECK_STATES((&g_States,
new State_Func1_Called,
new State_Func2_Called,
new State_Func3_Called,
NULL), "Part 4");
return true;
}