From 4b20818a7a07e6c8d10b12c42fcf7f567ace0d1d Mon Sep 17 00:00:00 2001 From: Carlos Natalino <carlos.natalino@chalmers.se> Date: Wed, 17 Nov 2021 09:20:48 +0000 Subject: [PATCH] Adding one more mock and removing unneeded dependencies. --- src/webui/requirements.in | 1 - src/webui/tests/test_unitary.py | 7 ++++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/webui/requirements.in b/src/webui/requirements.in index 2bb282e6f..b91305635 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 dd50beeb8..6983049b5 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: -- GitLab