From 454d6d28e72d8399b3150dd5c7e4000a0eae1794 Mon Sep 17 00:00:00 2001 From: Jessica Mack Date: Thu, 19 Jan 2023 13:53:27 -0500 Subject: [PATCH] mock additional pg_notify use in test Signed-off-by: Jessica Mack --- awx/main/tests/functional/api/test_settings.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/awx/main/tests/functional/api/test_settings.py b/awx/main/tests/functional/api/test_settings.py index bcfa499af3..eb57f27e14 100644 --- a/awx/main/tests/functional/api/test_settings.py +++ b/awx/main/tests/functional/api/test_settings.py @@ -294,6 +294,7 @@ def test_logging_aggregator_valid_settings(put, post, admin, type, host, port, u mock_settings['LOG_AGGREGATOR_PASSWORD'] = password # mock testing pg_notify mocker.patch("awx.conf.views.send_pg_notify", return_value=None) + mocker.patch("awx.main.tasks.system.send_pg_notify", return_value=None) url = reverse('api:setting_singleton_detail', kwargs={'category_slug': 'logging'}) response = put(url, data=mock_settings, user=admin, expect=200) assert type in response.data.get('LOG_AGGREGATOR_TYPE') @@ -316,6 +317,7 @@ def test_logging_aggregator_connection_test_valid(put, post, admin, mocker): mock_settings['LOG_AGGREGATOR_HOST'] = host # mock testing pg_notify mocker.patch("awx.conf.views.send_pg_notify", return_value=None) + mocker.patch("awx.main.tasks.system.send_pg_notify", return_value=None) # POST to save these mock settings url = reverse('api:setting_singleton_detail', kwargs={'category_slug': 'logging'}) put(url, data=mock_settings, user=admin, expect=200)