Skip to content
Snippets Groups Projects
Commit 354e314d authored by Hakan Calim's avatar Hakan Calim
Browse files

NAT-286: add test for get interface by device

parent 3add3838
No related branches found
No related tags found
1 merge request!89Feature/nat 286 create unit tests for netbox client
...@@ -236,3 +236,22 @@ def test_delete_device(mock_api, device, data_config_filename: PathLike): ...@@ -236,3 +236,22 @@ def test_delete_device(mock_api, device, data_config_filename: PathLike):
NetboxClient().delete_device(device.name) NetboxClient().delete_device(device.name)
mock_delete.assert_called_once() mock_delete.assert_called_once()
@patch("gso.services.netbox_client.pynetbox.api")
def test_get_interfaces_by_device(mock_api, device, interface, data_config_filename: PathLike):
"""
Test if a interface is returned for a device
"""
# Setup interface speed
speed = 1000
# Mock netbox api
mock_api.return_value.dcim.devices.get.return_value = device
mock_api.return_value.dcim.interfaces.filter.return_value = [interface, interface]
# Call get interfaces by device
interfaces = NetboxClient().get_interfaces_by_device(device.name, speed)
assert interfaces is not None
assert len(interfaces) == 2
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment