def test_check_threshold(self):
clients = self.get_client_nodes()
data = Main().parse_client(clients[0])
stats = Stats.get_or_create(**data)
with patch("app.tasks.send_email.delay"):
data = Main().parse_client(clients[0])
stats.cpu_usage = 18
stats.mem_usage = 55
cpu_alert, mem_alert = check_threshold(stats)
self.assertEqual(cpu_alert, False)
self.assertEqual(mem_alert, True)
def test_get_or_create(self):
clients = self.get_client_nodes()
data = Main().parse_client(clients[0])
stats = Stats.get_or_create(**data)
self.assertEqual(stats.ip, "11.11.11.11")