70 lines
1.3 KiB
C++
70 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#define TEST_SERVER_AND_PORT L"localhost", 19980
|
|
#define TEST_SERVER_VERSION L"1.1.0"
|
|
|
|
using namespace Sia::Api;
|
|
|
|
class Daemon
|
|
{
|
|
private:
|
|
PROCESS_INFORMATION pi;
|
|
STARTUPINFO si;
|
|
private:
|
|
void Cleanup()
|
|
{
|
|
ZeroMemory(&si, sizeof(si));
|
|
si.cb = sizeof(si);
|
|
ZeroMemory(&pi, sizeof(pi));
|
|
|
|
String szCmdline = L"cmd.exe /c del /s /q .\\data";
|
|
CreateProcess(nullptr, &szCmdline[0], nullptr, nullptr, TRUE, 0, nullptr, nullptr, &si, &pi);
|
|
if (pi.hProcess)
|
|
{
|
|
WaitForSingleObject(pi.hProcess, INFINITE);
|
|
|
|
CloseHandle(pi.hProcess);
|
|
CloseHandle(pi.hThread);
|
|
}
|
|
}
|
|
|
|
public:
|
|
void Start()
|
|
{
|
|
Cleanup();
|
|
ZeroMemory(&si, sizeof(si));
|
|
si.cb = sizeof(si);
|
|
ZeroMemory(&pi, sizeof(pi));
|
|
|
|
String szCmdline = L"\"..\\..\\3rd-party\\Sia-v1.1.0-windows-amd64\\siad.exe\" -d .\\data --api-addr localhost:19980";
|
|
CreateProcess(nullptr, &szCmdline[0], nullptr, nullptr, TRUE, 0, nullptr, nullptr, &si, &pi);
|
|
}
|
|
|
|
void Stop()
|
|
{
|
|
if (pi.hProcess)
|
|
{
|
|
TerminateProcess(pi.hProcess, 0);
|
|
|
|
CloseHandle(pi.hProcess);
|
|
CloseHandle(pi.hThread);
|
|
|
|
Cleanup();
|
|
}
|
|
}
|
|
};
|
|
|
|
#define DAEMON_TEST() \
|
|
private:\
|
|
static Daemon _daemon; \
|
|
\
|
|
public:\
|
|
TEST_CLASS_INITIALIZE(StartDaemon)\
|
|
{\
|
|
_daemon.Start();\
|
|
}\
|
|
\
|
|
TEST_CLASS_CLEANUP(StopDaemon)\
|
|
{\
|
|
_daemon.Stop();\
|
|
} |