diff --git a/test/services/test_netbox.py b/test/services/test_netbox.py index f1932030f3a8211e92467fae8e1bdb8b0999ff5d..cdf3fae56a7565e59eedcf9e1bd4c3c9033e3d95 100644 --- a/test/services/test_netbox.py +++ b/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