trim test script

Author: ChenyangLi4288

Lib/test/test_email/test_infinite_loop_fix.py

@@ -6,8 +6,6 @@
 
 import unittest
 import email.message
-import email.policy
-from email import policy
 
 
 class TestInfiniteLoopFix(unittest.TestCase):
@@ -16,21 +14,21 @@ class TestInfiniteLoopFix(unittest.TestCase):
     def test_long_parameter_key_no_infinite_loop(self):
         """Test that very long parameter keys don't cause infinite loops."""
         msg = email.message.EmailMessage()
-        
+
         # Create a parameter key that's 64 characters long (the problematic length)
         long_key = "a" * 64
-        
+
         # Add attachment first
         msg.add_attachment(
             b"test content",
             maintype="text",
             subtype="plain",
             filename="test.txt"
         )
-        
+
         # Set the long parameter using set_param
         msg.set_param(long_key, "test_value")
-        
+
         # This should not hang - it should complete in reasonable time
         try:
             result = msg.as_string()
@@ -43,21 +41,21 @@ def test_long_parameter_key_no_infinite_loop(self):
     def test_extremely_long_parameter_key(self):
         """Test with extremely long parameter keys that could cause maxchars < 0."""
         msg = email.message.EmailMessage()
-        
+
         # Create an extremely long parameter key (100+ characters)
         extremely_long_key = "b" * 100
-        
+
         # Add attachment first
         msg.add_attachment(
             b"test content",
             maintype="text",
             subtype="plain",
             filename="test.txt"
         )
-        
+
         # Set the extremely long parameter
         msg.set_param(extremely_long_key, "test_value")
-        
+
         # This should not hang even with extremely long keys
         try:
             result = msg.as_string()
@@ -69,25 +67,25 @@ def test_extremely_long_parameter_key(self):
     def test_multiple_long_parameters(self):
         """Test with multiple long parameters to ensure robust handling."""
         msg = email.message.EmailMessage()
-        
+
         # Add attachment first
         msg.add_attachment(
             b"test content",
             maintype="text",
             subtype="plain",
             filename="test.txt"
         )
-        
+
         # Add multiple long parameters
         long_params = {
             "a" * 64: "value1",
             "b" * 80: "value2",
             "c" * 90: "value3"
         }
-        
+
         for key, value in long_params.items():
             msg.set_param(key, value)
-        
+
         try:
             result = msg.as_string()
             self.assertIsInstance(result, str)
@@ -101,22 +99,22 @@ def test_edge_case_parameter_lengths(self):
         """Test edge cases around the problematic parameter lengths."""
         # Test parameter keys of various lengths around the problematic 64-char mark
         test_lengths = [60, 61, 62, 63, 64, 65, 66, 67, 68]
-        
+
         for length in test_lengths:
             key = "x" * length
             msg = email.message.EmailMessage()
-            
+
             # Add attachment first
             msg.add_attachment(
                 b"test content",
                 maintype="text",
                 subtype="plain",
                 filename="test.txt"
             )
-            
+
             # Set the parameter
             msg.set_param(key, "test_value")
-            
+
             try:
                 result = msg.as_string()
                 self.assertIsInstance(result, str)
@@ -127,18 +125,18 @@ def test_edge_case_parameter_lengths(self):
     def test_rfc_2231_compliance(self):
         """Test that the fix maintains RFC 2231 compliance."""
         msg = email.message.EmailMessage()
-        
+
         # Add attachment first
         msg.add_attachment(
             b"test content",
             maintype="text",
             subtype="plain",
             filename="test.txt"
         )
-        
+
         # Set a parameter with special characters that should trigger RFC 2231 encoding
         msg.set_param("long_param", "value_with_special_chars_ñáéíóú")
-        
+
         try:
             result = msg.as_string()
             self.assertIsInstance(result, str)