From 7e29f9e3f27e4a4cdf7b875c0e6a47fb1dc3906c Mon Sep 17 00:00:00 2001 From: Tong He <68936428+unnecessary-username@users.noreply.github.com> Date: Tue, 10 Mar 2026 16:01:11 +0800 Subject: [PATCH] Enrich tests against is_ha_environment() --- awx/main/tests/functional/models/test_ha.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) 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')