import unittest
from unittest.mock import MagicMock
from hawkeye_monitor.monitor import DataIntegrityMonitor
from hawkeye_monitor.checklist import get_checklist


class TestMonitor(unittest.TestCase):
    def setUp(self):
        self.monitor = DataIntegrityMonitor(
            host="localhost",
            password="$ome5ecr3t",
            cloud_id="deployment-name:dXMtZWFzdDQuZ2Nw",
        )
        self.monitor.client = MagicMock()

    def test_run_query(self):
        test_query = {"query": {"match_all": {}}}
        test_index = "test_index"
        self.monitor.client.search.return_value = {"hits": {"hits": []}}
        checklist = get_checklist()
        response = self.monitor.run_checks(test_index, checklist)

        self.monitor.client.search.assert_called_with(index=test_index, body=test_query)
        print(response)
        self.assertEqual(response, {"hits": {"hits": []}})


if __name__ == "__main__":
    unittest.main()