3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-11-24 19:31:16 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fixed test 'selectIntent' for IntentManager-tests.

Browse source 
part of CURA-6091
This commit is contained in:
Remco Burema 2019-07-02 13:15:18 +02:00
parent fee6ec417f
commit 50cbf71f68
1 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/TestIntentManager.py
View file

@ -49,14 +49,24 @@ def mockFindMetadata(**kwargs) -> List[Dict[str, Any]]:
for data in mocked_intent_metadata: for data in mocked_intent_metadata:
should_add = True should_add = True
for key, value in kwargs.items(): for key, value in kwargs.items():
if key in data.keys():
should_add &= (data[key] == value) should_add &= (data[key] == value)
if should_add: if should_add:
result.append(data) result.append(data)
return result return result
def mockFindContainers(**kwargs) -> List[MockContainer]:
result = []
metadatas = mockFindMetadata(**kwargs)
for metadata in metadatas:
result.append(MockContainer(metadata))
return result
def doSetup(application, extruder_manager, quality_manager, container_registry, global_stack) -> None: def doSetup(application, extruder_manager, quality_manager, container_registry, global_stack) -> None:
container_registry.findContainersMetadata = MagicMock(side_effect=mockFindMetadata) container_registry.findContainersMetadata = MagicMock(side_effect=mockFindMetadata)
container_registry.findContainers = MagicMock(side_effect=mockFindContainers)
quality_manager.getQualityGroups = MagicMock(return_value=mocked_qualitygroup_metadata) quality_manager.getQualityGroups = MagicMock(return_value=mocked_qualitygroup_metadata)
for _, qualitygroup in mocked_qualitygroup_metadata.items(): for _, qualitygroup in mocked_qualitygroup_metadata.items():
@ -115,7 +125,7 @@ def test_currentAvailableIntentCategories(application, extruder_manager, quality
assert len(categories) == 3 assert len(categories) == 3
def test_currentAvailableIntentCategories(application, extruder_manager, quality_manager, intent_manager, container_registry, global_stack): def test_selectIntent(application, extruder_manager, quality_manager, intent_manager, container_registry, global_stack):
doSetup(application, extruder_manager, quality_manager, container_registry, global_stack) doSetup(application, extruder_manager, quality_manager, container_registry, global_stack)
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)): with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)):
@ -126,6 +136,5 @@ def test_currentAvailableIntentCategories(application, extruder_manager, quality
intent_manager.selectIntent(intent, quality) intent_manager.selectIntent(intent, quality)
extruder_stacks = extruder_manager.getUsedExtruderStacks() extruder_stacks = extruder_manager.getUsedExtruderStacks()
assert len(extruder_stacks) == 2 assert len(extruder_stacks) == 2
assert extruder_stacks[0].intent is not None assert extruder_stacks[0].intent.getMetaDataEntry("intent_category") == intent
assert extruder_stacks[1].intent is not None assert extruder_stacks[1].intent.getMetaDataEntry("intent_category") == intent
# ... need MachineManager for this, split up methods anyway -> make into class, see examples others