mirror of
https://github.com/maxDcb/C2Core
synced 2026-06-08 15:48:01 +00:00
maxdcb
33 lines
805 B
C++
33 lines
805 B
C++
#include "../StealToken.hpp"
|
|
#include "../../tests/TestHelpers.hpp"
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
using namespace test_helpers;
|
|
|
|
int main()
|
|
{
|
|
bool ok = true;
|
|
|
|
{
|
|
StealToken module;
|
|
std::vector<std::string> cmd = {"stealToken", "1234"};
|
|
C2Message message;
|
|
|
|
ok &= expect(module.init(cmd, message) == 0, "numeric pid should be accepted");
|
|
ok &= expect(message.instruction() == "stealToken", "instruction should be set");
|
|
ok &= expect(message.pid() == 1234, "pid should be packed");
|
|
}
|
|
|
|
{
|
|
StealToken module;
|
|
std::vector<std::string> cmd = {"stealToken", "not-a-pid"};
|
|
C2Message message;
|
|
|
|
ok &= expect(module.init(cmd, message) == -1, "non-numeric pid should be rejected");
|
|
}
|
|
|
|
return ok ? 0 : 1;
|
|
}
|