3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-07-07 23:17:32 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Fix DiscoveredMachineModel test

Browse source
This commit is contained in:
Lipu Fei 2019-03-20 15:58:04 +01:00
parent c6bd31e5f6
commit 5d45dba8f3
3 changed files with 33 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tests/UI/MachineModels/TestDiscoveredPrintersModel.py Normal file
View file

@ -0,0 +1,27 @@
from unittest.mock import MagicMock
import pytest
from cura.UI.MachineModels.DiscoveredPrintersModel import DiscoveredPrintersModel
@pytest.fixture()
def discovered_printer_model(application) -> DiscoveredPrintersModel:
return DiscoveredPrintersModel()
def test_discoveredPrinters(discovered_printer_model):
mocked_device = MagicMock()
mocked_callback = MagicMock()
discovered_printer_model.addDiscoveredPrinter("ip", "key", "name", mocked_callback, "machine_type", mocked_device)
device = discovered_printer_model.discovered_printers[0]
discovered_printer_model.createMachineFromDiscoveredPrinter(device)
mocked_callback.assert_called_with("key")
assert len(discovered_printer_model.discovered_printers) == 1
# Test if removing it works
discovered_printer_model.removeDiscoveredPrinter("ip")
assert len(discovered_printer_model.discovered_printers) == 0