def test_message_post(self):
input_stream = {"message": "Reached chunk limit for rotate0.txt"}
input_stream = json.dumps(input_stream).encode('utf-8')
sc = CallLogic(input_stream)
result = sc.message_post()
self.assertTrue(result, {
"result": "success",
"message": "Message received successfully"
})
def test_start_post(self):
input_stream = {"start": "Starting Client1"}
input_stream = json.dumps(input_stream).encode('utf-8')
sc = CallLogic(input_stream)
result = sc.start_post()
self.assertTrue(result, {
"result": "success",
"message": "Message received successfully"
})
def test_metric_post(self, mockdbops):
input_stream = {
"value": {
"name": "Client1",
"cpu": "0.0",
"memory": "4.0",
"disk": "./"
},
"metric": "Metric for Client1 is CPU:: 0.0 , Memory:: 4.0"
}
input_stream = json.dumps(input_stream).encode('utf-8')
sc = CallLogic(input_stream)
result = sc.metric_post()
self.assertTrue(result, {
"result": "success",
"message": "Metric received successfully"
})
def test_shutdown_post(self, mockdbops):
input_stream = {
"value": {
"name": "Client1",
"heartbeat_time": "",
"status": "offline"
},
"shutdown": "Shutting down: Client1"
}
input_stream = json.dumps(input_stream).encode('utf-8')
sc = CallLogic(input_stream)
result = sc.shutdown_post()
self.assertTrue(
result, {
"result": "success",
"message": "Shutdown message received successfully"
})
def test_heartbeat_post(self, mock_dbops):
input_stream = {
"value": {
"name": "Client1",
"heartbeat_time": "",
"status": "alive"
},
"heartbeat": "Heartbeat Client1 is alive"
}
input_stream = json.dumps(input_stream).encode('utf-8')
sc = CallLogic(input_stream)
result = sc.heartbeat_post()
self.assertTrue(
result, {
"result": "success",
"message": "Heartbeat registered successfully in Master"
})
def on_post(self, req, resp):
response_text = CallLogic(req.bounded_stream.read()).heartbeat_post()
response_json = json.dumps(response_text)
resp.body = response_json
resp.content_type = 'application/json'
if response_text["result"] == "success":
resp.status = falcon.HTTP_200
else:
resp.status = falcon.HTTP_401