fix flake problem

Author: shokakucarrier

tests/test_radas_send_handler.py

@@ -3,6 +3,7 @@
 from unittest import mock
 from charon.pkgs.radas_signature_handler import sign_in_radas
 
+
 def test_sign_in_radas_normal_flow(self):
     with tempfile.TemporaryDirectory() as tmpdir:
         # Mock configuration
@@ -12,14 +13,17 @@ def test_sign_in_radas_normal_flow(self):
         mock_config.get_radas_config.return_value = mock_radas_config
 
         # Mock Container run to avoid real AMQP connection
-        with mock.patch("charon.pkgs.radas_signature_handler.Container") as mock_container, \
-             mock.patch("charon.pkgs.radas_signature_handler.get_config", return_value=mock_config), \
-             mock.patch("charon.pkgs.radas_signature_handler.uuid.uuid4", return_value="mocked-uuid"):
-            
+        with mock.patch(
+                "charon.pkgs.radas_signature_handler.Container") as mock_container, \
+             mock.patch(
+                "charon.pkgs.radas_signature_handler.get_config", return_value=mock_config), \
+             mock.patch(
+                "charon.pkgs.radas_signature_handler.uuid.uuid4", return_value="mocked-uuid"):
+
             # Test parameters
             test_result_path = os.path.join(tmpdir, "results")
             os.makedirs(test_result_path)
-            
+
             # Call the function
             sign_in_radas(
                 repo_url="quay.io/test/repo",
@@ -29,19 +33,20 @@ def test_sign_in_radas_normal_flow(self):
                 ignore_patterns=[],
                 radas_config=mock_radas_config
             )
-            
+
             # Verify Container.run() was called twice (sender and receiver)
             self.assertEqual(mock_container.call_count, 2)
-            
+
             # Verify request ID propagation
             receiver_call = mock_container.call_args_list[1]
             self.assertEqual(receiver_call.args[0].request_id, "mocked-uuid")
 
+
 def test_sign_in_radas_with_disabled_config(self):
     # Mock disabled configuration
     mock_config = mock.MagicMock()
     mock_config.is_radas_enabled.return_value = False
-    
+
     with mock.patch("charon.pkgs.radas_signature_handler.get_config", return_value=mock_config), \
          self.assertRaises(SystemExit):
         sign_in_radas(
@@ -53,29 +58,30 @@ def test_sign_in_radas_with_disabled_config(self):
             radas_config=mock.MagicMock()
         )
 
+
 def test_sign_in_radas_connection_cleanup(self):
     # Mock configuration and connection objects
     mock_config = mock.MagicMock()
     mock_config.is_radas_enabled.return_value = True
     mock_radas_config = mock.MagicMock()
-    
+
     with mock.patch("charon.pkgs.radas_signature_handler.Container") as mock_container, \
          mock.patch("charon.pkgs.radas_signature_handler.get_config", return_value=mock_config):
-        
+
         # Mock connection objects
         mock_sender_conn = mock.MagicMock()
         mock_listener_conn = mock.MagicMock()
-        
+
         # Create test call
         def container_side_effect(*args, **kwargs):
             if args[0].__class__.__name__ == "RadasReceiver":
                 args[0].conn = mock_listener_conn
             elif args[0].__class__.__name__ == "RadasSender":
                 args[0].conn = mock_sender_conn
             return mock.MagicMock()
-        
+
         mock_container.side_effect = container_side_effect
-        
+
         sign_in_radas(
             repo_url="quay.io/test/repo",
             requester="test-user",
@@ -84,7 +90,7 @@ def container_side_effect(*args, **kwargs):
             ignore_patterns=[],
             radas_config=mock_radas_config
         )
-        
+
         # Verify connections are closed
         mock_sender_conn.close.assert_called_once()
         mock_listener_conn.close.assert_called_once()