Finish postprocessing script signature checking

CURA-7319

plugins/PostProcessingPlugin/tests/TestPostProcessingPlugin.py

@@ -0,0 +1,65 @@
+
+import os
+import sys
+from unittest.mock import patch, MagicMock
+
+from pytest import fixture
+
+from UM.Resources import Resources
+from UM.Trust import Trust
+from ..PostProcessingPlugin import PostProcessingPlugin
+
+# not sure if needed
+sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), ".."))
+
+""" In this file, commnunity refers to regular Cura for makers."""
+
+
+# noinspection PyProtectedMember
+@patch("cura.ApplicationMetadata.IsEnterpriseVersion", False)
+def test_community_user_script_allowed():
+    assert PostProcessingPlugin._isScriptAllowed("blaat.py")
+
+# noinspection PyProtectedMember
+@patch("cura.ApplicationMetadata.IsEnterpriseVersion", False)
+def test_community_bundled_script_allowed():
+    assert PostProcessingPlugin._isScriptAllowed(_bundled_file_path())
+
+# noinspection PyProtectedMember
+@patch("cura.ApplicationMetadata.IsEnterpriseVersion", True)
+def test_enterprise_unsigned_user_script_not_allowed():
+    assert not PostProcessingPlugin._isScriptAllowed("blaat.py")
+
+@fixture
+def mocked_get_instance_or_none():
+    mocked_trust = MagicMock()
+    mocked_trust.signedFileCheck = MagicMock(return_value=True)
+    return mocked_trust
+
+@fixture
+def mocked_get_signature_file_exists_for():
+    return MagicMock(return_value=True)
+
+# noinspection PyProtectedMember
+@patch("cura.ApplicationMetadata.IsEnterpriseVersion", True)
+@patch("UM.Trust", "signatureFileExistsFor")
+@patch("UM.Trust.Trust.getInstanceOrNone")
+def test_enterprise_signed_user_script_allowed(mocked_instance_or_none, mocked_get_instance_or_none):
+    file_path = "blaat.py"
+    realSignatureFileExistsFor = Trust.signatureFileExistsFor
+    Trust.signatureFileExistsFor = MagicMock(return_value=True)
+    assert PostProcessingPlugin._isScriptAllowed(file_path)
+
+    # cleanup
+    Trust.signatureFileExistsFor = realSignatureFileExistsFor
+
+# noinspection PyProtectedMember
+@patch("cura.ApplicationMetadata.IsEnterpriseVersion", False)
+def test_enterprise_bundled_script_allowed():
+    assert PostProcessingPlugin._isScriptAllowed(_bundled_file_path())
+
+
+def _bundled_file_path():
+    return os.path.join(
+        Resources.getStoragePath(Resources.Resources) + "scripts/blaat.py"
+    )