From cc7edc770e0cc44d4c663e9a11f643f237ab2eb9 Mon Sep 17 00:00:00 2001
From: Matthew Jones <mat@matburt.net>
Date: Mon, 15 Dec 2014 15:13:27 -0500
Subject: [PATCH] Small fix for inventory source unit tests if the source is
 custom during check_inventory_source

---
 awx/main/tests/inventory.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/awx/main/tests/inventory.py b/awx/main/tests/inventory.py
index d8785b23d9..657ef0b713 100644
--- a/awx/main/tests/inventory.py
+++ b/awx/main/tests/inventory.py
@@ -1229,6 +1229,8 @@ class InventoryUpdatesTest(BaseTransactionTest):
                     })
                 else:
                     data.update({'source': inventory_source.source})
+                if inventory_source.source == 'custom':
+                    data['source_script'] = inventory_source.source_script.pk
                 response = self.put(inv_src_url2, data, expect=400)
                 self.assertTrue('source' in response, response)
         # Make sure we can delete the inventory update.