Fix tests

This commit is contained in:
Lipu Fei 2019-04-18 07:38:00 +02:00
parent 36cefcf0a3
commit 56cec02d44

View file

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