From 96c2b7f419168ea53f749cb04836e76ca1d6e49b Mon Sep 17 00:00:00 2001 From: Kenneth Kehl <@kkehl@flexion.us> Date: Thu, 20 Feb 2025 15:03:48 -0800 Subject: [PATCH] fix --- .../notify_client/test_notification_client.py | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/tests/app/notify_client/test_notification_client.py b/tests/app/notify_client/test_notification_client.py index 0e5f8a1e1..574a5c1c1 100644 --- a/tests/app/notify_client/test_notification_client.py +++ b/tests/app/notify_client/test_notification_client.py @@ -58,14 +58,13 @@ def test_client_gets_notifications_for_service_and_job_by_page( mocker.patch( "app.extensions.RedisClient.get", - return_value=None, + return_value={}, ) - mocker.patch( - "app.extensions.RedisClient.set", - ) + mocker.patch("app.extensions.RedisClient.set", return_value={}) mock_get = mocker.patch( - "app.notify_client.notification_api_client.NotificationApiClient.get" + "app.notify_client.notification_api_client.NotificationApiClient.get", + return_value={}, ) NotificationApiClient().get_notifications_for_service("abcd1234", **arguments) mock_get.assert_called_once_with(**expected_call) @@ -111,8 +110,16 @@ def test_client_gets_notifications_for_service_and_job_by_page( def test_client_gets_notifications_for_service_and_job_by_page_posts_for_to( mocker, arguments, expected_call ): + + mocker.patch( + "app.extensions.RedisClient.get", + return_value={}, + ) + + mocker.patch("app.extensions.RedisClient.set", return_value={}) mock_post = mocker.patch( - "app.notify_client.notification_api_client.NotificationApiClient.post" + "app.notify_client.notification_api_client.NotificationApiClient.post", + return_value={}, ) NotificationApiClient().get_notifications_for_service("abcd1234", **arguments) mock_post.assert_called_once_with(**expected_call)