mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-19 06:57:29 +08:00
4a171eebf6
- Move machine related models to cura.Machines.Models - Move printer device related models to cura.PrinterOutput.Models - Other UI/GUI related modules in cura.UI
27 lines
907 B
Python
27 lines
907 B
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from cura.Machines.Models.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
|