class ControllerTests(HiveBasicTest): UUID = None def setUp(self): self.controller = Controller(self.config) def handshake(self): data = {} data["action"] = "HANDSHAKE" data["to"] = "agentmanager" data["from"] = "Unknown" data["data"] = "" data["machineid"] = "testmachine" data["reply_to"] = "ASBDLHABSDJLHBASD" resp = RPCResponse() self.controller.handshake(data, resp) ControllerTests.uuid = resp.data def getsingle(self): data = {} data["action"] = "SINGLEAGENT" data["to"] = "agentmanager" data["from"] = ControllerTests.uuid data["data"] = ControllerTests.uuid data["machineid"] = "testmachine" resp = RPCResponse() self.controller.get_single_agent(data, resp) result = resp.data self.assertTrue(result[ControllerTests.uuid]["UUID"] == ControllerTests.uuid) def heartbeat(self): data = {} data["action"] = "HEARTBEAT" data["to"] = "agentmanager" data["from"] = ControllerTests.uuid data["data"] = "" data["machineid"] = "testmachine" resp = RPCResponse() self.controller.heartbeat(data, resp) def goodbye(self): data = {} data["action"] = "GOODBYE" data["to"] = "agentmanager" data["from"] = ControllerTests.uuid data["data"] = "" data["machineid"] = "testmachine" resp = RPCResponse() self.controller.goodbye(data, resp) self.assertTrue(resp.data == "GOODBYE!") def getagents(self): data = {} data["action"] = "ALLAGENTS" data["to"] = "agentmanager" data["from"] = ControllerTests.uuid data["data"] = ControllerTests.uuid data["machineid"] = "testmachine" resp = RPCResponse() self.controller.get_agents(data, resp) result = resp.data self.assertTrue(not (ControllerTests.uuid in result))
def setUp(self): self.controller = Controller(self.config)