Lipu Fei 5 лет назад
Родитель
Сommit
56cec02d44
1 измененных файлов с 3 добавлено и 1 удалено
  1. 3 1
      tests/Machines/Models/TestDiscoveredPrintersModel.py

+ 3 - 1
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)