diff --git a/src/webui/requirements.in b/src/webui/requirements.in index 2bb282e6f6e5d7c570baf9bd3ccbd0ffa86cef31..b913056357776460e11afe6ed04d0479e10793e6 100644 --- a/src/webui/requirements.in +++ b/src/webui/requirements.in @@ -8,4 +8,3 @@ prometheus-client pytest pytest-benchmark lorem-text -redis diff --git a/src/webui/tests/test_unitary.py b/src/webui/tests/test_unitary.py index dd50beeb82e0f3e589587db85ac0a379c8945a94..6983049b5f49ed9a08ca0e2b4fb2c10b7bb4db84 100644 --- a/src/webui/tests/test_unitary.py +++ b/src/webui/tests/test_unitary.py @@ -6,7 +6,7 @@ from flask.app import Flask from flask.helpers import url_for # from device.client.DeviceClient import DeviceClient from webui.service import create_app -from webui.proto.context_pb2 import Empty, DeviceId, DeviceList +from webui.proto.context_pb2 import Empty, DeviceId, DeviceList, TopologyIdList class TestWebUI(ClientTestCase): app = create_app(use_config={'TESTING': True, @@ -31,6 +31,11 @@ class TestWebUI(ClientTestCase): self.mocker_add_device.start() self.addCleanup(self.mocker_add_device.stop) + self.mocker_list_topology_ids = mock.patch('webui.service.device.routes.context_client.ListTopologyIds') + self.mocker_list_topology_ids.return_value = TopologyIdList() + self.mocker_list_topology_ids.start() + self.addCleanup(self.mocker_list_topology_ids.stop) + return super().setUp(client) def tearDown(self, client: FlaskClient) -> None: