From 56cec02d4494627696ccb87ce9fcd24df9ae489b Mon Sep 17 00:00:00 2001 From: Lipu Fei Date: Thu, 18 Apr 2019 07:38:00 +0200 Subject: [PATCH] Fix tests --- tests/Machines/Models/TestDiscoveredPrintersModel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/Machines/Models/TestDiscoveredPrintersModel.py b/tests/Machines/Models/TestDiscoveredPrintersModel.py index 4ccc3ba523..28be2536a0 100644 --- a/tests/Machines/Models/TestDiscoveredPrintersModel.py +++ b/tests/Machines/Models/TestDiscoveredPrintersModel.py @@ -1,4 +1,4 @@ -from unittest.mock import MagicMock +from unittest.mock import MagicMock, PropertyMock import pytest @@ -12,6 +12,8 @@ def discovered_printer_model(application) -> DiscoveredPrintersModel: def test_discoveredPrinters(discovered_printer_model): mocked_device = MagicMock() + cluster_size = PropertyMock(return_value = 1) + type(mocked_device).clusterSize = cluster_size mocked_callback = MagicMock() discovered_printer_model.addDiscoveredPrinter("ip", "key", "name", mocked_callback, "machine_type", mocked_device)