GitHub-Proxy/Cura
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura synced 2025-08-14 12:15:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats 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. # With an invalid file_type, we should get a false for success.
assert not container_registry.exportQualityProfile([], "zomg", "invalid") assert not container_registry.exportQualityProfile([], "zomg", "invalid")
def test_exportQualityProfileFailedWriter(self, container_registry): def test_exportQualityProfileFailedWriter(self, container_registry):
# Create a writer that always fails. # Create a writer that always fails.
mocked_writer = unittest.mock.MagicMock(name = "mocked_writer") mocked_writer = unittest.mock.MagicMock(name = "mocked_writer")
@ -199,7 +198,6 @@ class TestExportQualityProfile:
with unittest.mock.patch("UM.Application.Application.getInstance"): with unittest.mock.patch("UM.Application.Application.getInstance"):
assert not container_registry.exportQualityProfile([], "zomg", "test files (*.tst)") assert not container_registry.exportQualityProfile([], "zomg", "test files (*.tst)")
def test_exportQualityProfileExceptionWriter(self, container_registry): def test_exportQualityProfileExceptionWriter(self, container_registry):
# Create a writer that always fails. # Create a writer that always fails.
mocked_writer = unittest.mock.MagicMock(name = "mocked_writer") mocked_writer = unittest.mock.MagicMock(name = "mocked_writer")
@ -210,7 +208,6 @@ class TestExportQualityProfile:
with unittest.mock.patch("UM.Application.Application.getInstance"): with unittest.mock.patch("UM.Application.Application.getInstance"):
assert not container_registry.exportQualityProfile([], "zomg", "test files (*.tst)") assert not container_registry.exportQualityProfile([], "zomg", "test files (*.tst)")
def test_exportQualityProfileSuccessWriter(self, container_registry): def test_exportQualityProfileSuccessWriter(self, container_registry):
# Create a writer that always fails. # Create a writer that always fails.
mocked_writer = unittest.mock.MagicMock(name="mocked_writer") mocked_writer = unittest.mock.MagicMock(name="mocked_writer")
@ -219,4 +216,24 @@ class TestExportQualityProfile:
# Ensure that it actually fails if the writer did. # Ensure that it actually fails if the writer did.
with unittest.mock.patch("UM.Application.Application.getInstance"): 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