GitHub-Proxy/Cura
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura synced 2025-08-17 16:25:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace DefinitionContainer with MagicMock in test_addExtruder

Browse Source 
Because the code checks for the type of DefinitionContainer.

Contributes to issue CURA-3497.
This commit is contained in:
Ghostkeeper 2017-04-06 16:30:04 +02:00
parent 608e62e081
commit 8506c21099
No known key found for this signature in database
GPG Key ID: C5F96EE2BC0F7E75
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/Settings/TestGlobalStack.py
View File

@ -132,19 +132,23 @@ def test_addExtruder(global_stack):
mock_definition = unittest.mock.MagicMock() mock_definition = unittest.mock.MagicMock()
mock_definition.getProperty = lambda key, property: 2 if key == "machine_extruder_count" and property == "value" else None mock_definition.getProperty = lambda key, property: 2 if key == "machine_extruder_count" and property == "value" else None
global_stack.definition = mock_definition with unittest.mock.patch("cura.Settings.CuraContainerStack.DefinitionContainer", unittest.mock.MagicMock):
global_stack.definition = mock_definition
assert len(global_stack.extruders) == 0 assert len(global_stack.extruders) == 0
first_extruder = unittest.mock.MagicMock() first_extruder = unittest.mock.MagicMock()
global_stack.addExtruder(first_extruder) with unittest.mock.patch("cura.Settings.CuraContainerStack.DefinitionContainer", unittest.mock.MagicMock):
global_stack.addExtruder(first_extruder)
assert len(global_stack.extruders) == 1 assert len(global_stack.extruders) == 1
assert global_stack.extruders[0] == first_extruder assert global_stack.extruders[0] == first_extruder
second_extruder = unittest.mock.MagicMock() second_extruder = unittest.mock.MagicMock()
global_stack.addExtruder(second_extruder) with unittest.mock.patch("cura.Settings.CuraContainerStack.DefinitionContainer", unittest.mock.MagicMock):
global_stack.addExtruder(second_extruder)
assert len(global_stack.extruders) == 2 assert len(global_stack.extruders) == 2
assert global_stack.extruders[1] == second_extruder assert global_stack.extruders[1] == second_extruder
with pytest.raises(TooManyExtrudersError): #Should be limited to 2 extruders because of machine_extruder_count. with unittest.mock.patch("cura.Settings.CuraContainerStack.DefinitionContainer", unittest.mock.MagicMock):
global_stack.addExtruder(unittest.mock.MagicMock()) with pytest.raises(TooManyExtrudersError): #Should be limited to 2 extruders because of machine_extruder_count.
global_stack.addExtruder(unittest.mock.MagicMock())
assert len(global_stack.extruders) == 2 #Didn't add the faulty extruder. assert len(global_stack.extruders) == 2 #Didn't add the faulty extruder.
## Tests whether the container types are properly enforced on the stack. ## Tests whether the container types are properly enforced on the stack.