|
@@ -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)
|