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

Add test for gettingFallbackMaterialByMaterialType

Browse source
This commit is contained in:
Jaime van Kessel 2019-07-19 16:19:24 +02:00
parent 18f89529a7
commit d59e74c867
1 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/TestMaterialManager.py
View file

@ -4,8 +4,8 @@ from cura.Machines.MaterialManager import MaterialManager
mocked_registry = MagicMock() mocked_registry = MagicMock()
material_1 = {"id": "test", "GUID":"TEST!", "base_file": "base_material", "definition": "fdmmachine", "approximate_diameter": 3, "brand": "generic"} material_1 = {"id": "test", "GUID":"TEST!", "base_file": "base_material", "definition": "fdmmachine", "approximate_diameter": "3", "brand": "generic", "material": "pla"}
material_2 = {"id": "base_material", "GUID": "TEST2!", "base_file": "test", "definition": "fdmmachine", "approximate_diameter": 3} material_2 = {"id": "base_material", "GUID": "TEST2!", "base_file": "test", "definition": "fdmmachine", "approximate_diameter": "3", "material": "pla"}
mocked_registry.findContainersMetadata = MagicMock(return_value = [material_1, material_2]) mocked_registry.findContainersMetadata = MagicMock(return_value = [material_1, material_2])
@ -115,6 +115,21 @@ class TestAvailableMaterials:
assert "test" not in available_materials assert "test" not in available_materials
class Test_getFallbackMaterialIdByMaterialType:
def test_happyFlow(self, application):
with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)):
manager = MaterialManager(mocked_registry)
manager.initialize()
assert manager.getFallbackMaterialIdByMaterialType("pla") == "test"
def test_unknownMaterial(self, application):
with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)):
manager = MaterialManager(mocked_registry)
manager.initialize()
assert manager.getFallbackMaterialIdByMaterialType("OMGZOMG") is None
def test_getMaterialNode(application): def test_getMaterialNode(application):
with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)): with patch("UM.Application.Application.getInstance", MagicMock(return_value=application)):
manager = MaterialManager(mocked_registry) manager = MaterialManager(mocked_registry)