#include "meshkit-test.h"

#define MOCK_CC_SCRIPT "./mockcc.py"
#define MOCK_CC_TR_SCRIPT "./mockcctr.py"
//#define CCRESPONSE_PY "../../src/heartbeat-sender/CCResponse.py"
#define CCRESPONSE_PY "./CCResponse.updated.py"

/** Tests for the command and control class */
class MKControlTests : public Test::Suite {
public:

	MKControlTests() {
		TEST_ADD(MKControlTests::testStart);
		TEST_ADD(MKControlTests::testSendTraceRouteRequest);
//		TEST_ADD(MKControlTests::testUpdateBSON1);
		TEST_ADD(MKControlTests::testUpdateBSONRoute1);
		TEST_ADD(MKControlTests::testCCListen1);
	}

	static bool respondMockLatency(int requestID, int latency);
	static bool respondMockTraceRoute(int requestID, char *target);
//	static bool receiveMockResponse(int requestID, vector<ip4_t> route);
//	static bool sendMockResponse(int requestID, vector<ip4_t> route);
	static pid_t setupResponder();
	static void killResponder(pid_t respondProc);

protected:

	void setup();
	void tear_down();
	void testStart();
	void testSendTraceRouteRequest();
	void testUpdateBSONLatency1();
	void testUpdateBSONRoute1();
	void testCCListen1(); // Latency
	void testCCListen2(); // TraceRoute


	MKCommModel *cm;

};