libs/sqplus/testSqPlus2unit/testEnv.cpp

130 lines
2.9 KiB
C++
Raw Permalink Normal View History

2017-01-10 04:08:49 +00:00
#include "testEnv.hpp"
#include <iostream>
#include <sqplus.h>
#include <sqstdaux.h>
#define SQDBG_DEBUG_HOOK _SC("_sqdebughook_")
#define SQDBG_ERROR_HANDLER _SC("_sqerrorhandler_")
#define PRINTBUFSIZE 4096
int
my_error_handler(HSQUIRRELVM v)
{
const SQChar *sErr = NULL;
if(sq_gettop(v) >= 1) {
if(SQ_SUCCEEDED(sq_getstring(v, 2, &sErr))) {
scprintf(_SC("Error detected: [%s]\n"), sErr);
} else{
scprintf(_SC("Error detected: [unknown]\n"));
}
/* no callstack printing */
}
return 0;
}
int
debug_hook(HSQUIRRELVM v)
{
struct TestEnv::debug_info_t &d = TestEnv::debug();
sq_getinteger(v, 2, &d.event_type);
sq_getstring(v, 3, &d.src);
sq_getinteger(v, 4, &d.line);
sq_getstring(v, 5, &d.func);
sq_getuserpointer(v, -1, &d.up);
return 0;
}
TestEnv::TestEnv(void)
{
scprintf(_SC("TEST BEGINS -------------------------------------->\n"));
SquirrelVM::Init();
HSQUIRRELVM _v = SquirrelVM::GetVMPtr();
sq_pushregistrytable(_v);
sq_pushstring(_v, SQDBG_DEBUG_HOOK, -1);
sq_pushuserpointer(_v, this);
sq_newclosure(_v, debug_hook, 1);
sq_createslot(_v, -3);
sq_pushstring(_v, SQDBG_DEBUG_HOOK, -1);
sq_rawget(_v, -2);
sq_setdebughook(_v);
sq_pop(_v, 1);
sq_pushregistrytable(_v);
sq_pushstring(_v, SQDBG_ERROR_HANDLER, -1);
sq_pushuserpointer(_v, this);
sq_newclosure(_v, my_error_handler, 1);
sq_createslot(_v, -3);
sq_pushstring(_v, SQDBG_ERROR_HANDLER, -1);
sq_rawget(_v, -2);
sq_seterrorhandler(_v);
sq_pop(_v, 1);
sq_enabledebuginfo(_v, SQTrue);
}
TestEnv::~TestEnv()
{
// See if we can trigger a bug by collecting garbage
static int st_cnt;
if( (st_cnt&1) )
sq_collectgarbage( SquirrelVM::GetVMPtr() );
SquirrelVM::Shutdown();
debug().reset();
scprintf(_SC("------------------------------------> TEST FINISHES\n\n"));
}
struct TestEnv::debug_info_t &
TestEnv::debug(void)
{
static struct debug_info_t d;
return d;
}
void
TestEnv::report_exception(void)
{
try {
throw;
}
catch (const SquirrelError &e) {
scprintf(
_SC("\nSQUIRREL ERROR!\n[%s]\n"
) _SC("At line (%d) in script buffer:\n>>>>>>>>>>>>>>\n"),
e.desc,
debug().line
);
const SQChar *c = debug().script;
for (int i = 1; i < debug().line; ++i) {
c = scstrstr(c, _SC("\n"));
if (!c) goto FINISH_OUTPUT_LINE;
++c;
}
for (; *c != _SC('\0') && *c != _SC('\n'); ++c) {
putchar(*c);
}
FINISH_OUTPUT_LINE:
scprintf(_SC("\n<<<<<<<<<<<<<<\n"));
}
}
const char *
toMultibyte(const wchar_t *orig)
{
using namespace std;
static char dst[BUFSIZ];
const size_t length = min(wcslen(orig) + 1, size_t(BUFSIZ));
wcstombs(dst, orig, length);
return dst;
}