mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-06 22:47:29 -06:00
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
|
|
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"
|
|
)
|