From e5da7cd591c65892e851df3edfdac22bcd908fc3 Mon Sep 17 00:00:00 2001 From: Libor Pichler Date: Wed, 13 Nov 2024 06:58:44 +0100 Subject: [PATCH] Test that replication method was not called when there are no principals in group to migrate --- tests/migration_tool/tests_migrate.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/tests/migration_tool/tests_migrate.py b/tests/migration_tool/tests_migrate.py index 09bb1b73d..58dc147f8 100644 --- a/tests/migration_tool/tests_migrate.py +++ b/tests/migration_tool/tests_migrate.py @@ -23,12 +23,13 @@ from uuid import uuid4 from django.test import TestCase, override_settings + from django.utils import timezone from api.models import CrossAccountRequest, Tenant from management.models import * -from management.role.definer import seed_roles + from management.tenant_service.tenant_service import BootstrappedTenant from management.tenant_service.v2 import V2TenantBootstrapService from migration_tool.in_memory_tuples import ( @@ -43,7 +44,9 @@ subject, subject_type, ) -from migration_tool.migrate import migrate_data + +from migration_tool.migrate import migrate_data, migrate_groups_for_tenant + from management.group.definer import seed_group, clone_default_group_in_public_schema from tests.management.role.test_dual_write import RbacFixture @@ -161,6 +164,14 @@ def setUp(self): ) self.cross_account_request.roles.add(self.system_role_2) + @override_settings(REPLICATION_TO_RELATION_ENABLED=True, PRINCIPAL_USER_DOMAIN="redhat", READ_ONLY_API_MODE=True) + @patch("migration_tool.migrate.RelationApiDualWriteGroupHandler.replicate") + def test_migration_of_data_no_replication_event_to_migrate_groups(self, replicate_method): + """Test that we get the correct access for a principal.""" + kwargs = {"exclude_apps": ["app1"], "orgs": ["7654321"]} + migrate_data(**kwargs) + replicate_method.assert_not_called() + @override_settings(REPLICATION_TO_RELATION_ENABLED=True, PRINCIPAL_USER_DOMAIN="redhat", READ_ONLY_API_MODE=True) @patch("management.relation_replicator.logging_replicator.logger") def test_migration_of_data(self, logger_mock):