Move PrinterOutputDevice into cura.PrinterOutput module

2 files changed, 38 insertions, 1 deletions

diff --git a/tests/PrinterOutput/TestNetworkedPrinterOutputDevice.py b/tests/PrinterOutput/TestNetworkedPrinterOutputDevice.py
index b3f7277051..da3ce66ac4 100644
--- a/tests/PrinterOutput/TestNetworkedPrinterOutputDevice.py
+++ b/tests/PrinterOutput/TestNetworkedPrinterOutputDevice.py
@@ -4,7 +4,7 @@ from unittest.mock import MagicMock
 from PyQt5.QtNetwork import QNetworkAccessManager
 from PyQt5.QtCore import QUrl
 from cura.PrinterOutput.NetworkedPrinterOutputDevice import NetworkedPrinterOutputDevice, AuthState
-from cura.PrinterOutputDevice import ConnectionState
+from cura.PrinterOutput.PrinterOutputDevice import ConnectionState
 
 
 def test_properties():
diff --git a/tests/PrinterOutput/TestPrinterOutputDevice.py b/tests/PrinterOutput/TestPrinterOutputDevice.py
new file mode 100644
index 0000000000..4c12a34859
--- /dev/null
+++ b/tests/PrinterOutput/TestPrinterOutputDevice.py
@@ -0,0 +1,37 @@
+from unittest.mock import MagicMock
+
+import pytest
+from cura.PrinterOutput.PrinterOutputDevice import PrinterOutputDevice
+
+test_validate_data_get_set = [
+    {"attribute": "connectionText", "value": "yay"},
+    {"attribute": "connectionState", "value": 1},
+]
+
+
+@pytest.mark.parametrize("data", test_validate_data_get_set)
+def test_getAndSet(data):
+    model = PrinterOutputDevice("whatever")
+
+    # Convert the first letter into a capital
+    attribute = list(data["attribute"])
+    attribute[0] = attribute[0].capitalize()
+    attribute = "".join(attribute)
+
+    # mock the correct emit
+    setattr(model, data["attribute"] + "Changed", MagicMock())
+
+    # Attempt to set the value
+    getattr(model, "set" + attribute)(data["value"])
+
+    # Check if signal fired.
+    signal = getattr(model, data["attribute"] + "Changed")
+    assert signal.emit.call_count == 1
+
+    # Ensure that the value got set
+    assert getattr(model, data["attribute"]) == data["value"]
+
+    # Attempt to set the value again
+    getattr(model, "set" + attribute)(data["value"])
+    # The signal should not fire again
+    assert signal.emit.call_count == 1