3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-11-02 20:52:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Added tests for adding & removing discovered devices

Browse source 
CURA-6179
This commit is contained in:
Jaime van Kessel 2019-02-28 09:47:25 +01:00
parent bbe1b1590a
commit 1c7e047a38
2 changed files with 20 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/TestMachineManager.py
View file

@ -11,6 +11,7 @@ from cura.Settings.MachineManager import MachineManager
def container_registry() -> ContainerRegistry:
return MagicMock()
@pytest.fixture()
def extruder_manager(application, container_registry) -> ExtruderManager:
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)):
@ -21,6 +22,7 @@ def extruder_manager(application, container_registry) -> ExtruderManager:
return manager
@pytest.fixture()
def machine_manager(application, extruder_manager, container_registry) -> MachineManager:
application.getExtruderManager = MagicMock(return_value = extruder_manager)
@ -49,4 +51,14 @@ def test_discoveredMachine(machine_manager):
mocked_callback = MagicMock()
machine_manager.addDiscoveredPrinter("test", "zomg", mocked_callback, "derp")
machine_manager.addMachineFromDiscoveredPrinter("test")
mocked_callback.assert_called_with("test")
mocked_callback.assert_called_with("test")
assert len(machine_manager.discoveredPrinters) == 1
# Test if removing it works
machine_manager.removeDiscoveredPrinter("test")
assert len(machine_manager.discoveredPrinters) == 0
# Just in case, nothing should happen.
machine_manager.addMachineFromDiscoveredPrinter("test")
assert mocked_callback.call_count == 1