Ejemplo n.º 1
0
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))
Ejemplo n.º 2
0
 def setUp(self):
     self.controller = Controller(self.config)