add test_refresh_materialized_view.py

Author: qcdyx

functions-python/tasks_executor/tests/tasks/test_refresh_materialized_view.py

@@ -0,0 +1,28 @@
+from unittest.mock import patch
+from tasks.refresh_feedsearch_view import refresh_materialized_view
+
+
+# Patch shared.database.database functions
+@patch("tasks.refresh_feedsearch_view.with_db_session", lambda f: f)
+def test_refresh_materialized_view_handler_dry_run():
+    payload = {"dry_run": True}
+    with patch("tasks.refresh_feedsearch_view.get_parameters", return_value=True):
+        resp, status = refresh_materialized_view.refresh_materialized_view_handler(
+            payload
+        )
+        assert status == 200
+        assert resp["dry_run"] is True
+        assert "Dry run" in resp["message"]
+
+
+@patch("tasks.refresh_feedsearch_view.refresh_materialized_view")
+@patch("tasks.refresh_feedsearch_view.with_db_session", lambda f: f)
+def test_refresh_materialized_view_handler_success(mock_refresh):
+    payload = {"dry_run": False}
+    mock_refresh.return_value = True
+    with patch("tasks.refresh_feedsearch_view.get_parameters", return_value=False):
+        resp, status = refresh_materialized_view.refresh_materialized_view_handler(
+            payload
+        )
+        assert status == 200
+        assert "Successfully refreshed" in resp["message"]