refactor: ruff autofixes

gns3server/agent/gns3_copilot/utils/custom_netmiko/tests/test_huawei_ce.py

@@ -25,9 +25,7 @@ import os
 
 # Add project root to path using relative path
 test_dir = os.path.dirname(os.path.abspath(__file__))
-project_root = os.path.dirname(
-    os.path.dirname(os.path.dirname(
-        os.path.dirname(os.path.dirname(test_dir)))))
+project_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(test_dir)))))
 sys.path.insert(0, project_root)
 
 
@@ -103,13 +101,8 @@ class TestHuaweiTelnetCEDriver(unittest.TestCase):
     def test_initialization_parameters(self):
         """Test that initialization sets correct parameters."""
         # Create a mock instance (without actual connection)
-        with patch.object(
-            self.HuaweiTelnetCE, "__init__",
-            lambda self, *args, **kwargs: None
-        ):
-            instance = self.HuaweiTelnetCE.__new__(
-                self.HuaweiTelnetCE
-            )
+        with patch.object(self.HuaweiTelnetCE, "__init__", lambda self, *args, **kwargs: None):
+            instance = self.HuaweiTelnetCE.__new__(self.HuaweiTelnetCE)
 
             # Mock the necessary attributes
             instance.protocol = "telnet"
@@ -169,9 +162,7 @@ class TestHuaweiTelnetCEIntegration(unittest.TestCase):
         )
 
         # Create a mock instance
-        instance = GNS3HuaweiTelnetCE.__new__(
-            GNS3HuaweiTelnetCE
-        )
+        instance = GNS3HuaweiTelnetCE.__new__(GNS3HuaweiTelnetCE)
 
         # Mock the necessary attributes and methods
         instance.host = "127.0.0.1"
@@ -203,12 +194,8 @@ def run_tests():
     suite = unittest.TestSuite()
 
     # Add test cases
-    suite.addTests(
-        loader.loadTestsFromTestCase(TestHuaweiTelnetCEDriver)
-    )
-    suite.addTests(
-        loader.loadTestsFromTestCase(TestHuaweiTelnetCEIntegration)
-    )
+    suite.addTests(loader.loadTestsFromTestCase(TestHuaweiTelnetCEDriver))
+    suite.addTests(loader.loadTestsFromTestCase(TestHuaweiTelnetCEIntegration))
 
     # Run tests
     runner = unittest.TextTestRunner(verbosity=2)
@@ -218,9 +205,7 @@ def run_tests():
     print("\n" + "=" * 100)
     print("Test Summary:")
     print(f"  Run: {result.testsRun}")
-    success_count = (
-        result.testsRun - len(result.failures) - len(result.errors)
-    )
+    success_count = result.testsRun - len(result.failures) - len(result.errors)
     print(f"  Success: {success_count}")
     print(f"  Failed: {len(result.failures)}")
     print(f"  Errors: {len(result.errors)}")

gns3server/agent/gns3_copilot/utils/custom_netmiko/tests/test_ruijie_telnet.py

@@ -25,9 +25,7 @@ import os
 
 # Add project root to path using relative path
 test_dir = os.path.dirname(os.path.abspath(__file__))
-project_root = os.path.dirname(
-    os.path.dirname(os.path.dirname(
-        os.path.dirname(os.path.dirname(test_dir)))))
+project_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(test_dir)))))
 sys.path.insert(0, project_root)
 
 
@@ -92,9 +90,7 @@ class TestRuijieTelnetEnhancedDriver(unittest.TestCase):
         import re
 
         # Should have INTERACTIVE_PATTERNS as a class attribute
-        self.assertTrue(
-            hasattr(self.RuijieTelnetEnhanced, "INTERACTIVE_PATTERNS")
-        )
+        self.assertTrue(hasattr(self.RuijieTelnetEnhanced, "INTERACTIVE_PATTERNS"))
 
         # Check it's a list
         patterns = self.RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
@@ -108,18 +104,11 @@ class TestRuijieTelnetEnhancedDriver(unittest.TestCase):
     def test_preprocess_interactive_commands(self):
         """Test that interactive commands are preprocessed correctly."""
         # Create a mock instance (without actual connection)
-        with patch.object(
-            self.RuijieTelnetEnhanced, "__init__",
-            lambda self, *args, **kwargs: None
-        ):
-            instance = self.RuijieTelnetEnhanced.__new__(
-                self.RuijieTelnetEnhanced
-            )
+        with patch.object(self.RuijieTelnetEnhanced, "__init__", lambda self, *args, **kwargs: None):
+            instance = self.RuijieTelnetEnhanced.__new__(self.RuijieTelnetEnhanced)
 
             # Set INTERACTIVE_PATTERNS from class
-            instance.INTERACTIVE_PATTERNS = (
-                self.RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
-            )
+            instance.INTERACTIVE_PATTERNS = self.RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
 
             # Test commands with router-id (should trigger 'yes')
             commands = [
@@ -128,34 +117,23 @@ class TestRuijieTelnetEnhancedDriver(unittest.TestCase):
                 "network 10.0.0.0 0.0.0.255 area 0",
             ]
 
-            processed = instance._preprocess_interactive_commands(
-                commands
-            )
+            processed = instance._preprocess_interactive_commands(commands)
 
             # Should have inserted 'yes' after router-id command
             self.assertEqual(len(processed), 4)
             self.assertEqual(processed[0], "router ospf 1")
             self.assertEqual(processed[1], "router-id 1.1.1.1")
             self.assertEqual(processed[2], "yes")
-            self.assertEqual(
-                processed[3], "network 10.0.0.0 0.0.0.255 area 0"
-            )
+            self.assertEqual(processed[3], "network 10.0.0.0 0.0.0.255 area 0")
 
     def test_preprocess_non_interactive_commands(self):
         """Test that non-interactive commands are not modified."""
         # Create a mock instance
-        with patch.object(
-            self.RuijieTelnetEnhanced, "__init__",
-            lambda self, *args, **kwargs: None
-        ):
-            instance = self.RuijieTelnetEnhanced.__new__(
-                self.RuijieTelnetEnhanced
-            )
+        with patch.object(self.RuijieTelnetEnhanced, "__init__", lambda self, *args, **kwargs: None):
+            instance = self.RuijieTelnetEnhanced.__new__(self.RuijieTelnetEnhanced)
 
             # Set INTERACTIVE_PATTERNS from class
-            instance.INTERACTIVE_PATTERNS = (
-                self.RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
-            )
+            instance.INTERACTIVE_PATTERNS = self.RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
 
             # Test commands without interactive prompts
             commands = [
@@ -190,9 +168,7 @@ class TestRuijieTelnetEnhancedDriver(unittest.TestCase):
 
         # Check that default_enter is handled in __init__
         # (The actual logic is in the __init__ method body)
-        init_source = inspect.getsource(
-            self.RuijieTelnetEnhanced.__init__
-        )
+        init_source = inspect.getsource(self.RuijieTelnetEnhanced.__init__)
         self.assertIn("default_enter", init_source)
         self.assertIn("\\r\\n", init_source)
 
@@ -207,9 +183,7 @@ class TestRuijieTelnetEnhancedIntegration(unittest.TestCase):
         )
 
         # Create a mock instance
-        instance = RuijieTelnetEnhanced.__new__(
-            RuijieTelnetEnhanced
-        )
+        instance = RuijieTelnetEnhanced.__new__(RuijieTelnetEnhanced)
 
         # Mock the necessary attributes and methods
         instance.RETURN = "\r\n"
@@ -224,9 +198,7 @@ class TestRuijieTelnetEnhancedIntegration(unittest.TestCase):
         commands = ["interface GigabitEthernet 0/1", "description Test"]
 
         # Call send_config_set
-        instance.send_config_set(
-            commands, enter_config_mode=False, exit_config_mode=False
-        )
+        instance.send_config_set(commands, enter_config_mode=False, exit_config_mode=False)
 
         # Verify _send_config_batch was called
         instance._send_config_batch.assert_called_once()
@@ -238,17 +210,13 @@ class TestRuijieTelnetEnhancedIntegration(unittest.TestCase):
         )
 
         # Create a mock instance
-        instance = RuijieTelnetEnhanced.__new__(
-            RuijieTelnetEnhanced
-        )
-        instance.INTERACTIVE_PATTERNS = (
-            RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
-        )
+        instance = RuijieTelnetEnhanced.__new__(RuijieTelnetEnhanced)
+        instance.INTERACTIVE_PATTERNS = RuijieTelnetEnhanced.INTERACTIVE_PATTERNS
 
         # Test commands with multiple interactive prompts
         commands = [
-            "router-id 1.1.1.1",      # Should trigger 'yes'
-            "erase startup-config",    # Should trigger 'yes'
+            "router-id 1.1.1.1",  # Should trigger 'yes'
+            "erase startup-config",  # Should trigger 'yes'
             "interface GigabitEthernet 0/1",  # Should NOT trigger
         ]
 
@@ -273,12 +241,8 @@ def run_tests():
     suite = unittest.TestSuite()
 
     # Add test cases
-    suite.addTests(
-        loader.loadTestsFromTestCase(TestRuijieTelnetEnhancedDriver)
-    )
-    suite.addTests(
-        loader.loadTestsFromTestCase(TestRuijieTelnetEnhancedIntegration)
-    )
+    suite.addTests(loader.loadTestsFromTestCase(TestRuijieTelnetEnhancedDriver))
+    suite.addTests(loader.loadTestsFromTestCase(TestRuijieTelnetEnhancedIntegration))
 
     # Run tests
     runner = unittest.TextTestRunner(verbosity=2)
@@ -288,9 +252,7 @@ def run_tests():
     print("\n" + "=" * 100)
     print("Test Summary:")
     print(f"  Run: {result.testsRun}")
-    success_count = (
-        result.testsRun - len(result.failures) - len(result.errors)
-    )
+    success_count = result.testsRun - len(result.failures) - len(result.errors)
     print(f"  Success: {success_count}")
     print(f"  Failed: {len(result.failures)}")
     print(f"  Errors: {len(result.errors)}")

gns3server/agent/gns3_copilot/utils/custom_netmiko/tests/test_vpcs_telnet.py

@@ -27,9 +27,7 @@ import os
 
 # Add project root to path using relative path
 test_dir = os.path.dirname(os.path.abspath(__file__))
-project_root = os.path.dirname(
-    os.path.dirname(os.path.dirname(
-        os.path.dirname(os.path.dirname(test_dir)))))
+project_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(test_dir)))))
 sys.path.insert(0, project_root)
 
 
@@ -121,6 +119,7 @@ class TestVPCSTelnetInit(unittest.TestCase):
         from gns3server.agent.gns3_copilot.utils.custom_netmiko.vpcs_telnet import (  # noqa: E501
             VPCSTelnet,
         )
+
         cls.VPCSTelnet = VPCSTelnet
 
     def test_default_enter_parameter(self):
@@ -134,6 +133,7 @@ class TestVPCSTelnetInit(unittest.TestCase):
 
     def test_default_parameters_when_none(self):
         """Test default parameters when not provided."""
+
         # Mock the BaseConnection.__init__ to avoid actual connection
         # But also set device_type since VPCSTelnet.__init__ sets it
         def mock_init(self, *args, **kwargs):
@@ -141,23 +141,20 @@ class TestVPCSTelnetInit(unittest.TestCase):
             kwargs.setdefault("device_type", "gns3_vpcs_telnet")
             self.device_type = kwargs.get("device_type")
 
-        with patch.object(
-            self.VPCSTelnet, "__init__", mock_init
-        ):
+        with patch.object(self.VPCSTelnet, "__init__", mock_init):
             instance = self.VPCSTelnet(host="127.0.0.1")
             self.assertEqual(instance.device_type, "gns3_vpcs_telnet")
 
     def test_device_type_set(self):
         """Test that device_type is set to gns3_vpcs_telnet."""
+
         # Mock the __init__ to set device_type
         def mock_init(self, *args, **kwargs):
             # Simulate what VPCSTelnet.__init__ does
             kwargs.setdefault("device_type", "gns3_vpcs_telnet")
             self.device_type = kwargs.get("device_type")
 
-        with patch.object(
-            self.VPCSTelnet, "__init__", mock_init
-        ):
+        with patch.object(self.VPCSTelnet, "__init__", mock_init):
             instance = self.VPCSTelnet(host="127.0.0.1")
             self.assertEqual(instance.device_type, "gns3_vpcs_telnet")
 
@@ -171,6 +168,7 @@ class TestVPCSTelnetMethods(unittest.TestCase):
         from gns3server.agent.gns3_copilot.utils.custom_netmiko.vpcs_telnet import (  # noqa: E501
             VPCSTelnet,
         )
+
         cls.VPCSTelnet = VPCSTelnet
 
     def test_check_config_mode_always_false(self):
@@ -213,6 +211,7 @@ class TestVPCSTelnetSendCommand(unittest.TestCase):
         from gns3server.agent.gns3_copilot.utils.custom_netmiko.vpcs_telnet import (  # noqa: E501
             VPCSTelnet,
         )
+
         cls.VPCSTelnet = VPCSTelnet
 
     def test_send_command_writes_bytes(self):
@@ -243,7 +242,7 @@ class TestVPCSTelnetSendCommand(unittest.TestCase):
 
         # Check that expect_string defaults to r"PC\d+>"
         source = inspect.getsource(self.VPCSTelnet.send_command)
-        self.assertIn("expect_string = r\"PC\\d+>\"", source)
+        self.assertIn('expect_string = r"PC\\d+>"', source)
 
     def test_send_command_timing_calls_send_command(self):
         """Test that send_command_timing delegates to send_command."""
@@ -277,6 +276,7 @@ class TestVPCSTelnetTelnetLogin(unittest.TestCase):
         from gns3server.agent.gns3_copilot.utils.custom_netmiko.vpcs_telnet import (  # noqa: E501
             VPCSTelnet,
         )
+
         cls.VPCSTelnet = VPCSTelnet
 
     def test_telnet_login_default_prompt_pattern(self):
@@ -335,19 +335,16 @@ class TestVPCSTelnetRegistration(unittest.TestCase):
         from gns3server.agent.gns3_copilot.utils.custom_netmiko import (
             vpcs_telnet,
         )
+
         cls.vpcs_telnet = vpcs_telnet
 
     def test_register_function_exists(self):
         """Test that register_custom_device_type function exists."""
-        self.assertTrue(
-            hasattr(self.vpcs_telnet, "register_custom_device_type")
-        )
+        self.assertTrue(hasattr(self.vpcs_telnet, "register_custom_device_type"))
 
     def test_register_function_is_callable(self):
         """Test that register_custom_device_type is callable."""
-        self.assertTrue(
-            callable(self.vpcs_telnet.register_custom_device_type)
-        )
+        self.assertTrue(callable(self.vpcs_telnet.register_custom_device_type))
 
 
 class TestVPCSTelnetAnsiStripping(unittest.TestCase):
@@ -359,6 +356,7 @@ class TestVPCSTelnetAnsiStripping(unittest.TestCase):
         from gns3server.agent.gns3_copilot.utils.custom_netmiko.vpcs_telnet import (  # noqa: E501
             VPCSTelnet,
         )
+
         cls.VPCSTelnet = VPCSTelnet
 
     def test_strip_ansi_codes_method_exists(self):
@@ -435,14 +433,10 @@ IP/MASK     : 192.168.1.20/24"""
         # Mock necessary attributes
         instance.remote_conn = MagicMock()
         instance.remote_conn.write = MagicMock()
-        instance.read_until_pattern = MagicMock(
-            return_value="\x1b[1mNAME\x1b[0m        : PC2\nPC2>"
-        )
+        instance.read_until_pattern = MagicMock(return_value="\x1b[1mNAME\x1b[0m        : PC2\nPC2>")
 
         # Mock _strip_ansi_codes to verify it's called
-        instance._strip_ansi_codes = Mock(
-            return_value="NAME        : PC2\nPC2>"
-        )
+        instance._strip_ansi_codes = Mock(return_value="NAME        : PC2\nPC2>")
 
         # Call send_command
         instance.send_command("show ip")
@@ -476,9 +470,7 @@ def run_tests():
     print("\n" + "=" * 100)
     print("Test Summary:")
     print(f"  Run: {result.testsRun}")
-    success_count = (
-        result.testsRun - len(result.failures) - len(result.errors)
-    )
+    success_count = result.testsRun - len(result.failures) - len(result.errors)
     print(f"  Success: {success_count}")
     print(f"  Failed: {len(result.failures)}")
     print(f"  Errors: {len(result.errors)}")