NAT-286: Fixed get_all_lags test

test/services/test_netbox.py

@@ -53,24 +53,21 @@ def card_type():
 
 @pytest.fixture(scope="module")
 def interface():
-    values = {"id": 1,
-              "name": "et-0/0/1",
-              "speed": 1000,
-              "type": "1000BaseT",
-              "enabled": False,
-              "mark_connected": False}
+    values = {
+        "id": 1,
+        "name": "et-0/0/1",
+        "speed": 1000,
+        "type": "1000BaseT",
+        "enabled": False,
+        "mark_connected": False,
+    }
     return Record(values, None, None)
 
 
 @patch("gso.services.netbox_client.pynetbox.api")
-def test_create_device(mock_api,
-                       device,
-                       device_type,
-                       device_role,
-                       site,
-                       device_bay,
-                       card_type,
-                       data_config_filename: PathLike):
+def test_create_device(
+    mock_api, device, device_type, device_role, site, device_bay, card_type, data_config_filename: PathLike
+):
     device_name = "mx1.lab.geant.net"
     device.name = device_name
     site_tier = SiteTier.TIER1
@@ -93,7 +90,7 @@ def test_create_device(mock_api,
 @patch("gso.services.netbox_client.pynetbox.api")
 def test_get_available_lags(mock_api, mock_from_subscription, data_config_filename: PathLike):
     router_id = uuid.uuid4()
-    feasible_lags = [f"LAG-{i}" for i in range(1, 10)]
+    feasible_lags = [f"LAG-{i}" for i in range(1, 11)]
 
     # Mock the pynetbox API instance
     mock_netbox = mock_api.return_value
@@ -119,10 +116,7 @@ def test_create_interface(mock_api, device, interface, data_config_filename: Pat
     mock_api.return_value.dcim.interfaces.create.return_value = interface
 
     # Create new interface
-    new_interface = NetboxClient().create_interface(interface.name,
-                                                    interface.type,
-                                                    interface.speed,
-                                                    device.name)
+    new_interface = NetboxClient().create_interface(interface.name, interface.type, interface.speed, device.name)
 
     # Check result
     assert new_interface is not None
@@ -145,7 +139,7 @@ def test_reserve_interface_exception(mock_api, device, interface, data_config_fi
     mock_api.return_value.dcim.devices.get.return_value = device
     mock_api.return_value.dcim.interfaces.get.return_value = interface
 
-# Check exception
+    # Check exception
     with pytest.raises(WorkflowStateError) as test_exception:
         NetboxClient().reserve_interface(device.name, interface.name)
         assert str(test_exception.value) == exception_message