1
0
mirror of https://github.com/ansible/awx.git synced 2024-11-01 08:21:15 +03:00
This commit is contained in:
Chris Meyers 2015-05-22 14:30:10 -04:00
parent b4c0d2b3b3
commit d9d9d589b0

View File

@ -5,15 +5,25 @@ import mock
# Django # Django
from django.test import SimpleTestCase from django.test import SimpleTestCase
from django.conf import settings
# AWX # AWX
from awx.main.models import * # noqa from awx.main.models import * # noqa
from awx.main.ha import * # noqa from awx.main.ha import * # noqa
from awx.main.tests.base import BaseTest
__all__ = ['HAUnitTest',] __all__ = ['HAUnitTest',]
TEST_LOCALHOST = {
'HOST': 'localhost'
}
TEST_127_0_0_1 = {
'HOST': '127.0.0.1'
}
TEST_FILE = {
'HOST': '/i/might/be/a/file',
}
class HAUnitTest(SimpleTestCase): class HAUnitTest(SimpleTestCase):
@mock.patch('awx.main.models.Instance.objects.count', return_value=2) @mock.patch('awx.main.models.Instance.objects.count', return_value=2)
@ -21,16 +31,16 @@ class HAUnitTest(SimpleTestCase):
self.assertTrue(is_ha_environment()) self.assertTrue(is_ha_environment())
@mock.patch('awx.main.models.Instance.objects.count', return_value=1) @mock.patch('awx.main.models.Instance.objects.count', return_value=1)
@mock.patch.dict('django.conf.settings.DATABASES', { 'HOST': 'localhost' }) @mock.patch.dict('django.conf.settings.DATABASES', TEST_LOCALHOST)
def test_db_localhost(self, ignore): def test_db_localhost(self, ignore):
self.assertFalse(is_ha_environment()) self.assertFalse(is_ha_environment())
@mock.patch('awx.main.models.Instance.objects.count', return_value=1) @mock.patch('awx.main.models.Instance.objects.count', return_value=1)
@mock.patch.dict('django.conf.settings.DATABASES', { 'HOST': '127.0.0.1' }) @mock.patch.dict('django.conf.settings.DATABASES', TEST_127_0_0_1)
def test_db_127_0_0_1(self, ignore): def test_db_127_0_0_1(self, ignore):
self.assertFalse(is_ha_environment()) self.assertFalse(is_ha_environment())
@mock.patch('awx.main.models.Instance.objects.count', return_value=1) @mock.patch('awx.main.models.Instance.objects.count', return_value=1)
@mock.patch.dict('django.conf.settings.DATABASES', { 'HOST': '/i/might/be/a/file' }) @mock.patch.dict('django.conf.settings.DATABASES', TEST_FILE)
def test_db_file_socket(self, ignore): def test_db_file_socket(self, ignore):
self.assertFalse(is_ha_environment()) self.assertFalse(is_ha_environment())