From 481df1fcf84fc9f8b016bcccd5ff9dcbd121cfab Mon Sep 17 00:00:00 2001 From: AlanCoding Date: Mon, 1 May 2017 10:17:49 -0400 Subject: [PATCH] mock message obect in callback receiver test --- .../commands/test_run_fact_cache_receiver.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/awx/main/tests/functional/commands/test_run_fact_cache_receiver.py b/awx/main/tests/functional/commands/test_run_fact_cache_receiver.py index 461024ee9c..0a7e4f97c6 100644 --- a/awx/main/tests/functional/commands/test_run_fact_cache_receiver.py +++ b/awx/main/tests/functional/commands/test_run_fact_cache_receiver.py @@ -6,6 +6,9 @@ import pytest from datetime import datetime import json +# Mock +import mock + # Django from django.utils import timezone @@ -41,21 +44,21 @@ def check_process_fact_message_module(fact_returned, data, module_name): @pytest.mark.django_db def test_process_fact_message_ansible(fact_msg_ansible, monkeypatch_jsonbfield_get_db_prep_save): receiver = FactBrokerWorker(None) - fact_returned = receiver.process_fact_message(fact_msg_ansible, None) + fact_returned = receiver.process_fact_message(fact_msg_ansible, mock.MagicMock()) check_process_fact_message_module(fact_returned, fact_msg_ansible, 'ansible') @pytest.mark.django_db def test_process_fact_message_packages(fact_msg_packages, monkeypatch_jsonbfield_get_db_prep_save): receiver = FactBrokerWorker(None) - fact_returned = receiver.process_fact_message(fact_msg_packages, None) + fact_returned = receiver.process_fact_message(fact_msg_packages, mock.MagicMock()) check_process_fact_message_module(fact_returned, fact_msg_packages, 'packages') @pytest.mark.django_db def test_process_fact_message_services(fact_msg_services, monkeypatch_jsonbfield_get_db_prep_save): receiver = FactBrokerWorker(None) - fact_returned = receiver.process_fact_message(fact_msg_services, None) + fact_returned = receiver.process_fact_message(fact_msg_services, mock.MagicMock()) check_process_fact_message_module(fact_returned, fact_msg_services, 'services') @@ -75,10 +78,10 @@ def test_process_facts_message_ansible_overwrite(fact_scans, fact_msg_ansible, m value = 'hello world' receiver = FactBrokerWorker(None) - receiver.process_fact_message(fact_msg_ansible, None) + receiver.process_fact_message(fact_msg_ansible, mock.MagicMock()) fact_msg_ansible['facts'][key] = value - fact_returned = receiver.process_fact_message(fact_msg_ansible, None) + fact_returned = receiver.process_fact_message(fact_msg_ansible, mock.MagicMock()) fact_obj = Fact.objects.get(id=fact_returned.id) assert key in fact_obj.facts