External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-02-03 18:48:12 -03:30
Code Issues Packages Projects Releases Wiki Activity

mock message obect in callback receiver test

Browse Source
This commit is contained in:
AlanCoding
2017-05-01 10:17:49 -04:00
parent 3c0d4bf134
commit 481df1fcf8
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
awx/main/tests/functional/commands/test_run_fact_cache_receiver.py
View File

@@ -6,6 +6,9 @@ import pytest
from datetime import datetime from datetime import datetime
import json import json
# Mock
import mock
# Django # Django
from django.utils import timezone from django.utils import timezone
@@ -41,21 +44,21 @@ def check_process_fact_message_module(fact_returned, data, module_name):
@pytest.mark.django_db @pytest.mark.django_db
def test_process_fact_message_ansible(fact_msg_ansible, monkeypatch_jsonbfield_get_db_prep_save): def test_process_fact_message_ansible(fact_msg_ansible, monkeypatch_jsonbfield_get_db_prep_save):
receiver = FactBrokerWorker(None) 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') check_process_fact_message_module(fact_returned, fact_msg_ansible, 'ansible')
@pytest.mark.django_db @pytest.mark.django_db
def test_process_fact_message_packages(fact_msg_packages, monkeypatch_jsonbfield_get_db_prep_save): def test_process_fact_message_packages(fact_msg_packages, monkeypatch_jsonbfield_get_db_prep_save):
receiver = FactBrokerWorker(None) 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') check_process_fact_message_module(fact_returned, fact_msg_packages, 'packages')
@pytest.mark.django_db @pytest.mark.django_db
def test_process_fact_message_services(fact_msg_services, monkeypatch_jsonbfield_get_db_prep_save): def test_process_fact_message_services(fact_msg_services, monkeypatch_jsonbfield_get_db_prep_save):
receiver = FactBrokerWorker(None) 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') 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' value = 'hello world'
receiver = FactBrokerWorker(None) 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_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) fact_obj = Fact.objects.get(id=fact_returned.id)
assert key in fact_obj.facts assert key in fact_obj.facts