Add tests for PrintJobOutputModel & PrinterOutputModel

2 files changed, 160 insertions, 0 deletions

diff --git a/tests/PrinterOutput/TestPrintJobOutputModel.py b/tests/PrinterOutput/TestPrintJobOutputModel.py
new file mode 100644
index 0000000000..658cff7a7e
--- /dev/null
+++ b/tests/PrinterOutput/TestPrintJobOutputModel.py
@@ -0,0 +1,78 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from cura.PrinterOutput.ConfigurationModel import ConfigurationModel
+from cura.PrinterOutput.PrintJobOutputModel import PrintJobOutputModel
+from cura.PrinterOutput.PrinterOutputModel import PrinterOutputModel
+
+test_validate_data_get_set = [
+    {"attribute": "compatibleMachineFamilies", "value": ["yay"]},
+]
+
+test_validate_data_get_update = [
+    {"attribute": "configuration", "value": ConfigurationModel()},
+    {"attribute": "owner", "value": "WHOO"},
+    {"attribute": "assignedPrinter", "value": PrinterOutputModel(MagicMock())},
+    {"attribute": "key", "value": "YAY"},
+    {"attribute": "name", "value": "Turtles"},
+    {"attribute": "timeTotal", "value": 10},
+    {"attribute": "timeElapsed", "value": 20},
+    {"attribute": "state", "value": "BANANNA!"},
+]
+
+
+@pytest.mark.parametrize("data", test_validate_data_get_set)
+def test_getAndSet(data):
+    model = PrintJobOutputModel(MagicMock())
+
+    # 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
+
+
+@pytest.mark.parametrize("data", test_validate_data_get_update)
+def test_getAndUpdate(data):
+    model = PrintJobOutputModel(MagicMock())
+
+    # 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, "update" + 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, "update" + attribute)(data["value"])
+    # The signal should not fire again
+    assert signal.emit.call_count == 1
diff --git a/tests/PrinterOutput/TestPrinterOutputModel.py b/tests/PrinterOutput/TestPrinterOutputModel.py
new file mode 100644
index 0000000000..f0ed3af633
--- /dev/null
+++ b/tests/PrinterOutput/TestPrinterOutputModel.py
@@ -0,0 +1,82 @@
+
+
+from unittest.mock import MagicMock
+
+import pytest
+
+from cura.PrinterOutput.ConfigurationModel import ConfigurationModel
+from cura.PrinterOutput.PrintJobOutputModel import PrintJobOutputModel
+from cura.PrinterOutput.PrinterOutputModel import PrinterOutputModel
+
+test_validate_data_get_set = [
+    {"attribute": "name", "value": "YAY"},
+    {"attribute": "targetBedTemperature", "value": 192},
+]
+
+test_validate_data_get_update = [
+    {"attribute": "isPreheating", "value": True},
+    {"attribute": "type", "value": "WHOO"},
+    {"attribute": "buildplate", "value": "NFHA"},
+    {"attribute": "key", "value": "YAY"},
+    {"attribute": "name", "value": "Turtles"},
+    {"attribute": "bedTemperature", "value": 200},
+    {"attribute": "targetBedTemperature", "value": 9001},
+    {"attribute": "activePrintJob", "value": PrintJobOutputModel(MagicMock())},
+    {"attribute": "state", "value": "BEEPBOOP"},
+]
+
+
+@pytest.mark.parametrize("data", test_validate_data_get_set)
+def test_getAndSet(data):
+    model = PrinterOutputModel(MagicMock())
+
+    # 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
+
+
+@pytest.mark.parametrize("data", test_validate_data_get_update)
+def test_getAndUpdate(data):
+    model = PrinterOutputModel(MagicMock())
+
+    # 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, "update" + 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, "update" + attribute)(data["value"])
+    # The signal should not fire again
+    assert signal.emit.call_count == 1