From 82980cfb1307f607543e00f24d33b18921cd732a Mon Sep 17 00:00:00 2001
From: shtlrs <amrbellalouna@gmail.com>
Date: Mon, 18 Dec 2023 22:32:57 +0100
Subject: [PATCH] add tests for the new JSON notifier

---
 tests/test_json.py | 80 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 tests/test_json.py

diff --git a/tests/test_json.py b/tests/test_json.py
new file mode 100644
index 0000000..acef05a
--- /dev/null
+++ b/tests/test_json.py
@@ -0,0 +1,80 @@
+import pytest
+import requests_mock
+
+from blackbox.exceptions import MissingFields
+from blackbox.handlers.notifiers.json import Json
+from blackbox.utils import reports
+
+
+URL = "https://some-domain.com/api/blackbox-notifications"
+
+
+@pytest.fixture
+def mock_valid_json_config():
+    """Mock valid Json config."""
+    return {"url": URL}
+
+
+@pytest.fixture
+def mock_invalid_json_config():
+    """Mock invalid Json config."""
+    return {"key": "value"}
+
+
+def test_json_handler_can_be_instantiated_with_required_fields(mock_valid_json_config):
+    Json(**mock_valid_json_config)
+
+
+def test_json_handler_fails_without_required_fields(mock_invalid_json_config):
+    """Test if the json notifier handler cannot be instantiated with missing fields."""
+    with pytest.raises(MissingFields):
+        Json(**mock_invalid_json_config)
+
+
+def test_json_notifier(mock_valid_json_config, report):
+    """Test report parsing for raw JSON notifications."""
+    json_notifier = Json(**mock_valid_json_config)
+    json_notifier.report = report
+
+    expected_report = {
+        "backup-data": [
+            {
+                "source": "main_mongo",
+                "output": "salad",
+                "backup": [
+                    {
+                        "name": "main_s3",
+                        "success": True
+                    }
+                ],
+                "success": True
+            },
+            {
+                "source": "secondary_mongo",
+                "output": "ham-sandwich",
+                "backup": [
+                    {
+                        "name": "main_dropbox",
+                        "success": True
+                    },
+                    {
+                        "name": "secondary_s3",
+                        "success": False
+                    }
+                ],
+                "success": False
+            }
+        ]
+    }
+
+    database = reports.DatabaseReport(database_id="secondary_mongo", success=False, output='')
+    database.report_storage("main_dropbox", True, "ham-")
+    database.report_storage("secondary_s3", False, "sandwich")
+
+    report.databases.append(database)
+
+    with requests_mock.Mocker() as m:
+        adapter = m.post(URL)
+        json_notifier.notify()
+        assert adapter.call_count == 1
+        assert adapter.last_request.json() == expected_report