0
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 03:31:50 +02:00

test: fixed client mock

This commit is contained in:
2023-07-19 10:06:21 +02:00
parent d37fbf5c4f
commit 9883caa1b3
2 changed files with 14 additions and 10 deletions

View File

@ -1,7 +1,9 @@
from pytestqt import qtbot
from unittest import mock
from bec_widgets import line_plot
import numpy as np
from pytestqt import qtbot
from bec_widgets import line_plot
def test_line_plot_emits_no_signal(qtbot):
@ -63,12 +65,14 @@ def test_line_plot_raise_warning_wrong_signal_request(qtbot):
metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
with mock.patch("bec_widgets.line_plot.BECClient") as mock_client:
# TODO fix mock_client
mock_client.device_manager.devices.keys.return_value = ["y1"]
mock_dict = {"y1": [1, 2]}
mock_client().device_manager.devices.__contains__.side_effect = mock_dict.__contains__
# = {"y1": [1, 2]}
with mock.patch.object(plot, "update_signal") as mock_update_signal:
mock_update_signal.emit()
plot(data=data, metadata=metadata)
assert all(plot.y_value_list == ["y1"])
assert plot.y_value_list == ["y1"]
def test_line_plot_update(qtbot):