Fix mocker

Author: AlanCoding

test_app/tests/rbac/remote/test_service_api.py

@@ -504,8 +504,9 @@ class TestRestClientSyncAssignment:
     Generated by Claude Sonnet 4.5
     """
 
-    def test_sync_assignment_sends_only_object_ansible_id_for_registered_objects(self, rando, organization, org_admin_rd, mocker):
+    def test_sync_assignment_sends_only_object_ansible_id_for_registered_objects(self, rando, organization, org_admin_rd):
         """Test that sync_assignment removes object_id when object_ansible_id is present"""
+        from unittest.mock import MagicMock, patch
         from ansible_base.resource_registry.rest_client import ResourceAPIClient
 
         # Create an assignment to an organization (which has a resource)
@@ -515,24 +516,24 @@ def test_sync_assignment_sends_only_object_ansible_id_for_registered_objects(sel
         client = ResourceAPIClient(service_url='http://example.com', service_path='/api/v1/service-index/')
 
         # Mock the _sync_assignment method to capture what data is sent
-        mock_sync = mocker.patch.object(client, '_sync_assignment')
+        with patch.object(client, '_sync_assignment', return_value=MagicMock()) as mock_sync:
+            # Call sync_assignment
+            client.sync_assignment(assignment)
 
-        # Call sync_assignment
-        client.sync_assignment(assignment)
+            # Verify _sync_assignment was called
+            assert mock_sync.called
+            sent_data = mock_sync.call_args[0][0]
 
-        # Verify _sync_assignment was called
-        assert mock_sync.called
-        sent_data = mock_sync.call_args[0][0]
+            # Should have object_ansible_id
+            assert 'object_ansible_id' in sent_data
+            assert sent_data['object_ansible_id'] == str(organization.resource.ansible_id)
 
-        # Should have object_ansible_id
-        assert 'object_ansible_id' in sent_data
-        assert sent_data['object_ansible_id'] == str(organization.resource.ansible_id)
+            # Should NOT have object_id (removed by sync_assignment)
+            assert 'object_id' not in sent_data, "object_id should not be sent for registered objects"
 
-        # Should NOT have object_id (removed by sync_assignment)
-        assert 'object_id' not in sent_data, "object_id should not be sent for registered objects"
-
-    def test_sync_assignment_sends_only_object_id_for_non_registered_objects(self, rando, inventory, inv_rd, mocker):
+    def test_sync_assignment_sends_only_object_id_for_non_registered_objects(self, rando, inventory, inv_rd):
         """Test that sync_assignment keeps object_id when object_ansible_id is None"""
+        from unittest.mock import MagicMock, patch
         from ansible_base.resource_registry.rest_client import ResourceAPIClient
 
         # Create an assignment to an inventory (which doesn't have a resource)
@@ -542,21 +543,20 @@ def test_sync_assignment_sends_only_object_id_for_non_registered_objects(self, r
         client = ResourceAPIClient(service_url='http://example.com', service_path='/api/v1/service-index/')
 
         # Mock the _sync_assignment method to capture what data is sent
-        mock_sync = mocker.patch.object(client, '_sync_assignment')
-
-        # Call sync_assignment
-        client.sync_assignment(assignment)
+        with patch.object(client, '_sync_assignment', return_value=MagicMock()) as mock_sync:
+            # Call sync_assignment
+            client.sync_assignment(assignment)
 
-        # Verify _sync_assignment was called
-        assert mock_sync.called
-        sent_data = mock_sync.call_args[0][0]
+            # Verify _sync_assignment was called
+            assert mock_sync.called
+            sent_data = mock_sync.call_args[0][0]
 
-        # Should have object_id
-        assert 'object_id' in sent_data
-        assert sent_data['object_id'] == str(inventory.id)
+            # Should have object_id
+            assert 'object_id' in sent_data
+            assert sent_data['object_id'] == str(inventory.id)
 
-        # object_ansible_id should either be absent or None
-        assert sent_data.get('object_ansible_id') is None
+            # object_ansible_id should either be absent or None
+            assert sent_data.get('object_ansible_id') is None
 
 
 @pytest.mark.django_db