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

NAT-286 adjusted tests for netbox client to merge from develop

parent 3fe41c67
Branches
Tags
1 merge request!89Feature/nat 286 create unit tests for netbox client
Pipeline #84242 failed
...@@ -8,7 +8,7 @@ from unittest.mock import patch, Mock ...@@ -8,7 +8,7 @@ from unittest.mock import patch, Mock
import pytest import pytest
from pynetbox.core.response import Record from pynetbox.core.response import Record
from gso.services.netbox_client import NetBoxClient from gso.services.netbox_client import NetboxClient
from gso.products.product_blocks.site import SiteTier from gso.products.product_blocks.site import SiteTier
from gso.utils.exceptions import WorkflowStateError from gso.utils.exceptions import WorkflowStateError
...@@ -41,7 +41,7 @@ def site(): ...@@ -41,7 +41,7 @@ def site():
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def device_bay(): def device_bay():
values = {"id": 1, "name": "test123"} values = {"id": 1, "name": "bay_test", "position": 1}
return Record(values, None, None) return Record(values, None, None)
...@@ -62,21 +62,6 @@ def interface(): ...@@ -62,21 +62,6 @@ def interface():
return Record(values, None, None) return Record(values, None, None)
@pytest.fixture(scope="module")
def app():
app = Mock()
app.token = "abc123"
app.base_url = "http://localhost:8080/testing/api"
return app
@pytest.fixture(scope="module")
def endpoint():
endpoint = Mock()
endpoint.name = "test-endpoint"
return endpoint
@patch("gso.services.netbox_client.pynetbox.api") @patch("gso.services.netbox_client.pynetbox.api")
def test_create_device(mock_api, def test_create_device(mock_api,
device, device,
...@@ -99,7 +84,7 @@ def test_create_device(mock_api, ...@@ -99,7 +84,7 @@ def test_create_device(mock_api,
mock_api.return_value.dcim.module_types.get.return_value = card_type mock_api.return_value.dcim.module_types.get.return_value = card_type
mock_api.return_value.dcim.module_types.create.return_value = card_type mock_api.return_value.dcim.module_types.create.return_value = card_type
new_device = NetBoxClient().create_device(device_name, site_tier) new_device = NetboxClient().create_device(device_name, site_tier)
assert new_device is not None assert new_device is not None
assert new_device.name == device_name assert new_device.name == device_name
...@@ -120,7 +105,7 @@ def test_get_available_lags(mock_api, mock_from_subscription, data_config_filena ...@@ -120,7 +105,7 @@ def test_get_available_lags(mock_api, mock_from_subscription, data_config_filena
mock_router = mock_subscription.router mock_router = mock_subscription.router
mock_router.router_fqdn = "test_router" mock_router.router_fqdn = "test_router"
netbox_client = NetBoxClient() netbox_client = NetboxClient()
result = netbox_client.get_available_lags(router_id) result = netbox_client.get_available_lags(router_id)
# Check the result of the function # Check the result of the function
...@@ -134,7 +119,7 @@ def test_create_interface(mock_api, device, interface, data_config_filename: Pat ...@@ -134,7 +119,7 @@ def test_create_interface(mock_api, device, interface, data_config_filename: Pat
mock_api.return_value.dcim.interfaces.create.return_value = interface mock_api.return_value.dcim.interfaces.create.return_value = interface
# Create new interface # Create new interface
new_interface = NetBoxClient().create_interface(interface.name, new_interface = NetboxClient().create_interface(interface.name,
interface.type, interface.type,
interface.speed, interface.speed,
device.name) device.name)
...@@ -162,7 +147,7 @@ def test_reserve_interface_exception(mock_api, device, interface, data_config_fi ...@@ -162,7 +147,7 @@ def test_reserve_interface_exception(mock_api, device, interface, data_config_fi
# Check exception # Check exception
with pytest.raises(WorkflowStateError) as test_exception: with pytest.raises(WorkflowStateError) as test_exception:
NetBoxClient().reserve_interface(device.name, interface.name) NetboxClient().reserve_interface(device.name, interface.name)
assert str(test_exception.value) == exception_message assert str(test_exception.value) == exception_message
...@@ -184,7 +169,7 @@ def test_reserve_interface(mock_api, device, interface, data_config_filename: Pa ...@@ -184,7 +169,7 @@ def test_reserve_interface(mock_api, device, interface, data_config_filename: Pa
interface.save = mock_save interface.save = mock_save
# Check reservation of interface # Check reservation of interface
updated_interface = NetBoxClient().reserve_interface(device.name, interface.name) updated_interface = NetboxClient().reserve_interface(device.name, interface.name)
assert updated_interface is not None assert updated_interface is not None
assert updated_interface.enabled is True assert updated_interface.enabled is True
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment