Commit 4b20818a authored by Carlos Natalino's avatar Carlos Natalino
Browse files

Adding one more mock and removing unneeded dependencies.

parent c25a617c

src/webui/requirements.in

+0 −1
Original line number Diff line number Diff line
@@ -8,4 +8,3 @@ prometheus-client 
pytest

pytest-benchmark

lorem-text
 
redis

src/webui/tests/test_unitary.py

+6 −1
Original line number Diff line number Diff line
@@ -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: