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

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 import sys
from unittest.mock import patch, MagicMock from unittest.mock import patch, MagicMock
from pytest import fixture
from UM.Resources import Resources from UM.Resources import Resources
from UM.Trust import Trust from UM.Trust import Trust
from ..PostProcessingPlugin import PostProcessingPlugin from ..PostProcessingPlugin import PostProcessingPlugin
@ -30,25 +28,17 @@ def test_community_bundled_script_allowed():
def test_enterprise_unsigned_user_script_not_allowed(): def test_enterprise_unsigned_user_script_not_allowed():
assert not PostProcessingPlugin._isScriptAllowed("blaat.py") 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 # noinspection PyProtectedMember
@patch("cura.ApplicationMetadata.IsEnterpriseVersion", True) @patch("cura.ApplicationMetadata.IsEnterpriseVersion", True)
@patch("UM.Trust", "signatureFileExistsFor") def test_enterprise_signed_user_script_allowed():
@patch("UM.Trust.Trust.getInstanceOrNone") mocked_trust = MagicMock()
def test_enterprise_signed_user_script_allowed(mocked_instance_or_none, mocked_get_instance_or_none): mocked_trust.signedFileCheck = MagicMock(return_value=True)
file_path = "blaat.py"
realSignatureFileExistsFor = Trust.signatureFileExistsFor realSignatureFileExistsFor = Trust.signatureFileExistsFor
Trust.signatureFileExistsFor = MagicMock(return_value=True) 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 # cleanup
Trust.signatureFileExistsFor = realSignatureFileExistsFor Trust.signatureFileExistsFor = realSignatureFileExistsFor