diff --git a/awx/main/tests/functional/models/test_ha.py b/awx/main/tests/functional/models/test_ha.py index c8ee9dc0a7..bcc627b898 100644 --- a/awx/main/tests/functional/models/test_ha.py +++ b/awx/main/tests/functional/models/test_ha.py @@ -10,12 +10,26 @@ from django.test.utils import override_settings @pytest.mark.django_db -def test_multiple_instances(): - for i in range(2): +def test_multiple_hybrid_instances(): + for i in range(3): Instance.objects.create(hostname=f'foo{i}', node_type='hybrid') assert is_ha_environment() +@pytest.mark.django_db +def test_double_control_instances(): + for i in range(2): + Instance.objects.create(hostname=f'foo{i}', node_type='control') + assert is_ha_environment() + + +@pytest.mark.django_db +def test_mix_hybrid_control_instances(): + Instance.objects.create(hostname='control_node', node_type='control') + Instance.objects.create(hostname='hybrid_node', node_type='hybrid') + assert is_ha_environment() + + @pytest.mark.django_db def test_db_localhost(): Instance.objects.create(hostname='foo', node_type='hybrid')