mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 13:49:39 +08:00
81 lines
4.3 KiB
Python
81 lines
4.3 KiB
Python
from unittest import TestCase
|
|
from unittest.mock import MagicMock
|
|
|
|
from PyQt6.QtCore import QUrl
|
|
from unittest.mock import patch
|
|
from UM.MimeTypeDatabase import MimeTypeDatabase
|
|
from cura.Settings.ContainerManager import ContainerManager
|
|
import tempfile
|
|
import os
|
|
|
|
class TestContainerManager(TestCase):
|
|
def setUp(self):
|
|
|
|
self._application = MagicMock()
|
|
self._container_registry = MagicMock()
|
|
self._machine_manager = MagicMock()
|
|
self._machine_manager.activeMachine.extruderList = [MagicMock(name="Left Extruder Mock"), MagicMock(name="Right Extruder Mock")]
|
|
|
|
self._mocked_mime = MagicMock()
|
|
self._mocked_mime.preferredSuffix = "omg"
|
|
self._mocked_mime.suffixes = ["omg"]
|
|
self._mocked_mime.comment = "UnitTest!"
|
|
|
|
self._mocked_container = MagicMock()
|
|
self._mocked_container_data = "SOME DATA :D"
|
|
self._mocked_container.serialize = MagicMock(return_value = self._mocked_container_data)
|
|
|
|
self._containers_meta_data = [{"id": "test", "test_data": "omg"}]
|
|
self._container_registry.findContainersMetadata = MagicMock(return_value = self._containers_meta_data)
|
|
self._container_registry.getMimeTypeForContainer = MagicMock(return_value = self._mocked_mime)
|
|
self._container_registry.findContainers = MagicMock(return_value = [self._mocked_container])
|
|
self._application.getContainerRegistry = MagicMock(return_value = self._container_registry)
|
|
self._application.getMachineManager = MagicMock(return_value = self._machine_manager)
|
|
|
|
# Destroy the previous instance of the container manager
|
|
if ContainerManager.getInstance() is not None:
|
|
ContainerManager._ContainerManager__instance = None
|
|
|
|
self._container_manager = ContainerManager(self._application)
|
|
MimeTypeDatabase.addMimeType(self._mocked_mime)
|
|
|
|
def tearDown(self):
|
|
MimeTypeDatabase.removeMimeType(self._mocked_mime)
|
|
|
|
def test_getContainerMetaDataEntry(self):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=self._application)):
|
|
assert self._container_manager.getContainerMetaDataEntry("test", "test_data") == "omg"
|
|
assert self._container_manager.getContainerMetaDataEntry("test", "entry_that_is_not_defined") == ""
|
|
|
|
def test_clearUserContainer(self):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=self._application)):
|
|
self._container_manager.clearUserContainers()
|
|
assert self._machine_manager.activeMachine.userChanges.clear.call_count == 1
|
|
|
|
def test_getContainerNameFilters(self):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=self._application)):
|
|
# If nothing is added, we still expect to get the all files filter
|
|
assert self._container_manager.getContainerNameFilters("") == ['All Files (*)']
|
|
|
|
# Pretend that a new type was added.
|
|
self._container_registry.getContainerTypes = MagicMock(return_value=[("None", None)])
|
|
assert self._container_manager.getContainerNameFilters("") == ['UnitTest! (*.omg)', 'All Files (*)']
|
|
|
|
def test_exportContainerUnknownFileType(self):
|
|
# The filetype is not known, so this should cause an error!
|
|
assert self._container_manager.exportContainer("test", "zomg", "whatever")["status"] == "error"
|
|
|
|
def test_exportContainerInvalidPath(self):
|
|
assert self._container_manager.exportContainer("test", "zomg", "")["status"] == "error"
|
|
assert self._container_manager.exportContainer("test", "zomg", QUrl())["status"] == "error"
|
|
|
|
def test_exportContainerInvalidId(self):
|
|
assert self._container_manager.exportContainer("", "whatever", "whatever")["status"] == "error"
|
|
|
|
def test_exportContainer(self):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=self._application)):
|
|
with tempfile.TemporaryDirectory() as tmpdirname:
|
|
result = self._container_manager.exportContainer("test", "whatever", os.path.join(tmpdirname, "whatever.omg"))
|
|
assert(os.path.exists(result["path"]))
|
|
with open(result["path"], "r", encoding="utf-8") as f:
|
|
assert f.read() == self._mocked_container_data |