Return stub records in testing.

awx/main/managers.py

@@ -1,6 +1,8 @@
 # Copyright (c) 2014 Ansible, Inc.
 # All Rights Reserved.
 
+import sys
+
 from django.conf import settings
 from django.db import models
 from django.utils.functional import cached_property
@@ -14,16 +16,34 @@ class InstanceManager(models.Manager):
     """
     def me(self):
         """Return the currently active instance."""
+        # If we are running unit tests, return a stub record.
+        if len(sys.argv) >= 2 and sys.argv[1] == 'test':
+            return self.model(id=1, primary=True,
+                              uuid='00000000-0000-0000-0000-000000000000')
+
+        # Return the appropriate record from the database.
         return self.get(uuid=settings.SYSTEM_UUID)
 
     def my_role(self):
         """Return the role of the currently active instance, as a string
         ('primary' or 'secondary').
         """
+        # If we are running unit tests, we are primary, because reasons.
+        if len(sys.argv) >= 2 and sys.argv[1] == 'test':
+            return 'primary'
+
+        # Check if this instance is primary; if so, return "primary", otherwise
+        # "secondary".
         if self.me().primary:
             return 'primary'
         return 'secondary'
 
     def primary(self):
         """Return the primary instance."""
+        # If we are running unit tests, return a stub record.
+        if len(sys.argv) >= 2 and sys.argv[1] == 'test':
+            return self.model(id=1, primary=True,
+                              uuid='00000000-0000-0000-0000-000000000000')
+
+        # Return the appropriate record from the database.
         return self.get(primary=True)