From be56912dc365834da7277dec1a42e34b4131edb3 Mon Sep 17 00:00:00 2001
From: Jake McDermott <jmcdermott@ansible.com>
Date: Tue, 9 May 2017 12:07:39 -0400
Subject: [PATCH] unit test inventory source serializer get_related (#6214)

---
 .../serializers/test_inventory_serializers.py | 38 +++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/awx/main/tests/unit/api/serializers/test_inventory_serializers.py b/awx/main/tests/unit/api/serializers/test_inventory_serializers.py
index e79ea7c10b..8df41f8b45 100644
--- a/awx/main/tests/unit/api/serializers/test_inventory_serializers.py
+++ b/awx/main/tests/unit/api/serializers/test_inventory_serializers.py
@@ -6,9 +6,11 @@ from mock import PropertyMock
 # AWX
 from awx.api.serializers import (
     CustomInventoryScriptSerializer,
+    InventorySourceSerializer,
 )
 from awx.main.models import (
     CustomInventoryScript,
+    InventorySource,
     User,
 )
 
@@ -20,6 +22,20 @@ from rest_framework.test import (
 )
 
 
+@pytest.fixture
+def inventory_source(mocker):
+    obj = mocker.MagicMock(
+        pk=22,
+        inventory=mocker.MagicMock(pk=23),
+        update=mocker.MagicMock(),
+        source_project_id=None,
+        current_update=None,
+        last_update=None,
+        spec=InventorySource
+    )
+    return obj
+
+
 class TestCustomInventoryScriptSerializer(object):
     @pytest.mark.parametrize("superuser,sysaudit,admin_role,value",
                              ((True, False, False, '#!/python'),
@@ -45,3 +61,25 @@ class TestCustomInventoryScriptSerializer(object):
 
                     representation = serializer.to_representation(cis)
                     assert representation['script'] == value
+
+
+@mock.patch('awx.api.serializers.UnifiedJobTemplateSerializer.get_related', lambda x,y: {})
+@mock.patch('awx.api.serializers.InventorySourceOptionsSerializer.get_related', lambda x,y: {})
+class TestInventorySourceSerializerGetRelated(object):
+    @pytest.mark.parametrize('related_resource_name', [
+        'activity_stream',
+        'notification_templates_error',
+        'notification_templates_success',
+        'notification_templates_any',
+        'inventory_updates',
+        'update',
+        'hosts',
+        'groups',
+    ])
+    def test_get_related(self, test_get_related, inventory_source, related_resource_name):
+        test_get_related(
+            InventorySourceSerializer,
+            inventory_source,
+            'inventory_sources',
+            related_resource_name
+        )