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

Add tests for _findProfileWriter

Browse source
This commit is contained in:
Jaime van Kessel 2019-05-31 13:18:14 +02:00
parent 43360aace0
commit 1d4fe56a21
1 changed files with 21 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/Settings/TestCuraContainerRegistry.py
View file

@ -188,7 +188,6 @@ class TestExportQualityProfile:
# With an invalid file_type, we should get a false for success.
assert not container_registry.exportQualityProfile([], "zomg", "invalid")
def test_exportQualityProfileFailedWriter(self, container_registry):
# Create a writer that always fails.
mocked_writer = unittest.mock.MagicMock(name = "mocked_writer")
@ -199,7 +198,6 @@ class TestExportQualityProfile:
with unittest.mock.patch("UM.Application.Application.getInstance"):
assert not container_registry.exportQualityProfile([], "zomg", "test files (*.tst)")
def test_exportQualityProfileExceptionWriter(self, container_registry):
# Create a writer that always fails.
mocked_writer = unittest.mock.MagicMock(name = "mocked_writer")
@ -210,7 +208,6 @@ class TestExportQualityProfile:
with unittest.mock.patch("UM.Application.Application.getInstance"):
assert not container_registry.exportQualityProfile([], "zomg", "test files (*.tst)")
def test_exportQualityProfileSuccessWriter(self, container_registry):
# Create a writer that always fails.
mocked_writer = unittest.mock.MagicMock(name="mocked_writer")
@ -219,4 +216,24 @@ class TestExportQualityProfile:
# Ensure that it actually fails if the writer did.
with unittest.mock.patch("UM.Application.Application.getInstance"):
assert container_registry.exportQualityProfile([], "zomg", "test files (*.tst)")
assert container_registry.exportQualityProfile([], "zomg", "test files (*.tst)")
def test__findProfileWriterNoPlugins(container_registry):
# Mock it so that no IO plugins are found.
container_registry._getIOPlugins = unittest.mock.MagicMock(return_value = [])
mocked_plugin_registry = unittest.mock.MagicMock(name = "plugin registry")
with unittest.mock.patch("UM.PluginRegistry.PluginRegistry.getInstance", mocked_plugin_registry):
# Since there are no writers, don't return any
assert container_registry._findProfileWriter(".zomg", "dunno") is None
def test__findProfileWriter(container_registry):
# Mock it so that no IO plugins are found.
container_registry._getIOPlugins = unittest.mock.MagicMock(return_value = [("writer_id", {"profile_writer": [{"extension": ".zomg", "description": "dunno"}]})])
mocked_plugin_registry = unittest.mock.MagicMock(name = "plugin registry")
with unittest.mock.patch("UM.PluginRegistry.PluginRegistry.getInstance", mocked_plugin_registry):
# In this case it's getting a mocked object (from the mocked_plugin_registry)
assert container_registry._findProfileWriter(".zomg", "dunno") is not None