3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-07-06 22:47:29 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Cleanup test_enterprise_signed_user_script_allowed

Browse source 
CURA-7319
This commit is contained in:
Nino van Hooff 2020-03-24 16:28:43 +01:00
parent 5b045f89b1
commit c0d2977f4d
1 changed files with 7 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

24
plugins/PostProcessingPlugin/tests/TestPostProcessingPlugin.py
View file

@ -3,8 +3,6 @@ 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
@ -30,25 +28,17 @@ def test_community_bundled_script_allowed():
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"
def test_enterprise_signed_user_script_allowed():
mocked_trust = MagicMock()
mocked_trust.signedFileCheck = MagicMock(return_value=True)
realSignatureFileExistsFor = Trust.signatureFileExistsFor
Trust.signatureFileExistsFor = MagicMock(return_value=True)
assert PostProcessingPlugin._isScriptAllowed(file_path)
with patch("UM.Trust.Trust.getInstanceOrNone", return_value=mocked_trust):
assert PostProcessingPlugin._isScriptAllowed("blaat.py")
# cleanup
Trust.signatureFileExistsFor = realSignatureFileExistsFor