mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-12 01:07:52 -06:00
Rewrite tests
This commit is contained in:
parent
23e957e1c5
commit
ee9210d8d1
2 changed files with 198 additions and 221 deletions
|
@ -32,7 +32,7 @@ class SendMaterialJob(Job):
|
||||||
|
|
||||||
## Send the request to the printer and register a callback
|
## Send the request to the printer and register a callback
|
||||||
def run(self) -> None:
|
def run(self) -> None:
|
||||||
self.device.get("materials/", on_finished = self.sendMissingMaterials)
|
self.device.get("materials/", on_finished = self._onGetRemoteMaterials)
|
||||||
|
|
||||||
## Process the materials reply from the printer.
|
## Process the materials reply from the printer.
|
||||||
#
|
#
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
# Copyright (c) 2018 Ultimaker B.V.
|
# Copyright (c) 2018 Ultimaker B.V.
|
||||||
# Cura is released under the terms of the LGPLv3 or higher.
|
# Cura is released under the terms of the LGPLv3 or higher.
|
||||||
|
|
||||||
import io
|
import io
|
||||||
import json
|
import json
|
||||||
from typing import Any, List
|
from typing import Any, List
|
||||||
|
@ -17,16 +16,9 @@ from plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice import ClusterUM3Outp
|
||||||
from plugins.UM3NetworkPrinting.src.Models import ClusterMaterial
|
from plugins.UM3NetworkPrinting.src.Models import ClusterMaterial
|
||||||
from plugins.UM3NetworkPrinting.src.SendMaterialJob import SendMaterialJob
|
from plugins.UM3NetworkPrinting.src.SendMaterialJob import SendMaterialJob
|
||||||
|
|
||||||
# All log entries written to Log.log by the class-under-test are written to this list. It is cleared before each test
|
|
||||||
# run and check afterwards
|
|
||||||
_logentries = []
|
|
||||||
|
|
||||||
|
|
||||||
def new_log(*args):
|
|
||||||
_logentries.append(args)
|
|
||||||
|
|
||||||
|
|
||||||
class ContainerRegistryMock(ContainerRegistryInterface):
|
class ContainerRegistryMock(ContainerRegistryInterface):
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.containersMetaData = None
|
self.containersMetaData = None
|
||||||
|
|
||||||
|
@ -59,14 +51,16 @@ class FakeDevice(ClusterUM3OutputDevice):
|
||||||
|
|
||||||
|
|
||||||
class TestSendMaterialJob(TestCase):
|
class TestSendMaterialJob(TestCase):
|
||||||
_LOCALMATERIAL_WHITE = {'type': 'material', 'status': 'unknown', 'id': 'generic_pla_white',
|
|
||||||
|
_LOCAL_MATERIAL_WHITE = {'type': 'material', 'status': 'unknown', 'id': 'generic_pla_white',
|
||||||
'base_file': 'generic_pla_white', 'setting_version': 5, 'name': 'White PLA',
|
'base_file': 'generic_pla_white', 'setting_version': 5, 'name': 'White PLA',
|
||||||
'brand': 'Generic', 'material': 'PLA', 'color_name': 'White',
|
'brand': 'Generic', 'material': 'PLA', 'color_name': 'White',
|
||||||
'GUID': 'badb0ee7-87c8-4f3f-9398-938587b67dce', 'version': '1', 'color_code': '#ffffff',
|
'GUID': 'badb0ee7-87c8-4f3f-9398-938587b67dce', 'version': '1', 'color_code': '#ffffff',
|
||||||
'description': 'Test PLA White', 'adhesion_info': 'Use glue.', 'approximate_diameter': '3',
|
'description': 'Test PLA White', 'adhesion_info': 'Use glue.', 'approximate_diameter': '3',
|
||||||
'properties': {'density': '1.00', 'diameter': '2.85', 'weight': '750'},
|
'properties': {'density': '1.00', 'diameter': '2.85', 'weight': '750'},
|
||||||
'definition': 'fdmprinter', 'compatible': True}
|
'definition': 'fdmprinter', 'compatible': True}
|
||||||
_LOCALMATERIAL_BLACK = {'type': 'material', 'status': 'unknown', 'id': 'generic_pla_black',
|
|
||||||
|
_LOCAL_MATERIAL_BLACK = {'type': 'material', 'status': 'unknown', 'id': 'generic_pla_black',
|
||||||
'base_file': 'generic_pla_black', 'setting_version': 5, 'name': 'Yellow CPE',
|
'base_file': 'generic_pla_black', 'setting_version': 5, 'name': 'Yellow CPE',
|
||||||
'brand': 'Ultimaker', 'material': 'CPE', 'color_name': 'Black',
|
'brand': 'Ultimaker', 'material': 'CPE', 'color_name': 'Black',
|
||||||
'GUID': '5fbb362a-41f9-4818-bb43-15ea6df34aa4', 'version': '1', 'color_code': '#000000',
|
'GUID': '5fbb362a-41f9-4818-bb43-15ea6df34aa4', 'version': '1', 'color_code': '#000000',
|
||||||
|
@ -74,7 +68,7 @@ class TestSendMaterialJob(TestCase):
|
||||||
'properties': {'density': '1.01', 'diameter': '2.85', 'weight': '750'},
|
'properties': {'density': '1.01', 'diameter': '2.85', 'weight': '750'},
|
||||||
'definition': 'fdmprinter', 'compatible': True}
|
'definition': 'fdmprinter', 'compatible': True}
|
||||||
|
|
||||||
_REMOTEMATERIAL_WHITE = {
|
_REMOTE_MATERIAL_WHITE = {
|
||||||
"guid": "badb0ee7-87c8-4f3f-9398-938587b67dce",
|
"guid": "badb0ee7-87c8-4f3f-9398-938587b67dce",
|
||||||
"material": "PLA",
|
"material": "PLA",
|
||||||
"brand": "Generic",
|
"brand": "Generic",
|
||||||
|
@ -82,7 +76,8 @@ class TestSendMaterialJob(TestCase):
|
||||||
"color": "White",
|
"color": "White",
|
||||||
"density": 1.00
|
"density": 1.00
|
||||||
}
|
}
|
||||||
_REMOTEMATERIAL_BLACK = {
|
|
||||||
|
_REMOTE_MATERIAL_BLACK = {
|
||||||
"guid": "5fbb362a-41f9-4818-bb43-15ea6df34aa4",
|
"guid": "5fbb362a-41f9-4818-bb43-15ea6df34aa4",
|
||||||
"material": "PLA",
|
"material": "PLA",
|
||||||
"brand": "Generic",
|
"brand": "Generic",
|
||||||
|
@ -91,224 +86,206 @@ class TestSendMaterialJob(TestCase):
|
||||||
"density": 1.00
|
"density": 1.00
|
||||||
}
|
}
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
# Make sure the we start with clean (log) slate
|
|
||||||
_logentries.clear()
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
# If there are still log entries that were not checked something is wrong or we must add checks for them
|
|
||||||
self.assertEqual(len(_logentries), 0)
|
|
||||||
|
|
||||||
@patch("plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice")
|
@patch("plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice")
|
||||||
def test_run(self, device_mock):
|
def test_run(self, device_mock):
|
||||||
with mock.patch.object(Logger, 'log', new=new_log):
|
job = SendMaterialJob(device_mock)
|
||||||
job = SendMaterialJob(device_mock)
|
job.run()
|
||||||
job.run()
|
device_mock.get.assert_called_with("materials/", on_finished=job._onGetRemoteMaterials)
|
||||||
|
|
||||||
device_mock.get.assert_called_with("materials/", on_finished=job.sendMissingMaterials)
|
|
||||||
self.assertEqual(0, len(_logentries))
|
|
||||||
|
|
||||||
|
@patch("plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice")
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
@patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
def test_sendMissingMaterials_withFailedRequest(self, reply_mock):
|
def test_sendMissingMaterials_withFailedRequest(self, reply_mock, device_mock):
|
||||||
reply_mock.attribute.return_value = 404
|
reply_mock.attribute.return_value = 404
|
||||||
|
SendMaterialJob(device_mock).run()
|
||||||
with mock.patch.object(Logger, 'log', new=new_log):
|
|
||||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
reply_mock.attribute.assert_called_with(0)
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0)])
|
self.assertEqual(reply_mock.method_calls, [call.attribute(0)])
|
||||||
self._assertLogEntries([('e', "Couldn't request current material storage on printer. Not syncing materials.")],
|
self.assertEqual(device_mock._onGetRemoteMaterials.method_calls, [])
|
||||||
_logentries)
|
|
||||||
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
|
||||||
def test_sendMissingMaterials_withBadJsonAnswer(self, reply_mock):
|
|
||||||
reply_mock.attribute.return_value = 200
|
|
||||||
reply_mock.readAll.return_value = QByteArray(b'Six sick hicks nick six slick bricks with picks and sticks.')
|
|
||||||
|
|
||||||
with mock.patch.object(Logger, 'log', new=new_log):
|
|
||||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
|
||||||
self._assertLogEntries(
|
|
||||||
[('e', "Request material storage on printer: I didn't understand the printer's answer.")],
|
|
||||||
_logentries)
|
|
||||||
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
|
||||||
def test_sendMissingMaterials_withMissingGuid(self, reply_mock):
|
|
||||||
reply_mock.attribute.return_value = 200
|
|
||||||
remoteMaterialWithoutGuid = self._REMOTEMATERIAL_WHITE.copy()
|
|
||||||
del remoteMaterialWithoutGuid["guid"]
|
|
||||||
reply_mock.readAll.return_value = QByteArray(json.dumps([remoteMaterialWithoutGuid]).encode("ascii"))
|
|
||||||
|
|
||||||
with mock.patch.object(Logger, 'log', new=new_log):
|
|
||||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
|
||||||
self._assertLogEntries(
|
|
||||||
[('e', "Request material storage on printer: Printer's answer was missing GUIDs.")],
|
|
||||||
_logentries)
|
|
||||||
|
|
||||||
@patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: [])
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
|
||||||
def test_sendMissingMaterials_WithInvalidVersionInLocalMaterial(self, reply_mock):
|
|
||||||
reply_mock.attribute.return_value = 200
|
|
||||||
reply_mock.readAll.return_value = QByteArray(json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
|
||||||
|
|
||||||
containerRegistry = ContainerRegistryMock()
|
|
||||||
localMaterialWhiteWithInvalidVersion = self._LOCALMATERIAL_WHITE.copy()
|
|
||||||
localMaterialWhiteWithInvalidVersion["version"] = "one"
|
|
||||||
containerRegistry.setContainersMetadata([localMaterialWhiteWithInvalidVersion])
|
|
||||||
|
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
|
||||||
with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
|
||||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
|
||||||
self._assertLogEntries([('e', "Material generic_pla_white has invalid version number one.")], _logentries)
|
|
||||||
|
|
||||||
@patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: [])
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
|
||||||
def test_sendMissingMaterials_WithMultipleLocalVersionsLowFirst(self, reply_mock):
|
|
||||||
reply_mock.attribute.return_value = 200
|
|
||||||
reply_mock.readAll.return_value = QByteArray(json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
|
||||||
|
|
||||||
containerRegistry = ContainerRegistryMock()
|
|
||||||
localMaterialWhiteWithHigherVersion = self._LOCALMATERIAL_WHITE.copy()
|
|
||||||
localMaterialWhiteWithHigherVersion["version"] = "2"
|
|
||||||
containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, localMaterialWhiteWithHigherVersion])
|
|
||||||
|
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
|
||||||
with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
|
||||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
|
||||||
self._assertLogEntries([], _logentries)
|
|
||||||
|
|
||||||
@patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: [])
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
|
||||||
def test_sendMissingMaterials_MaterialMissingOnPrinter(self, reply_mock):
|
|
||||||
reply_mock.attribute.return_value = 200
|
|
||||||
reply_mock.readAll.return_value = QByteArray(
|
|
||||||
json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
|
||||||
|
|
||||||
containerRegistry = ContainerRegistryMock()
|
|
||||||
containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, self._LOCALMATERIAL_BLACK])
|
|
||||||
|
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
|
||||||
with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
|
||||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
|
||||||
self._assertLogEntries([], _logentries)
|
|
||||||
|
|
||||||
@patch("builtins.open", lambda a, b: io.StringIO("<xml></xml>"))
|
|
||||||
@patch("UM.MimeTypeDatabase.MimeTypeDatabase.getMimeTypeForFile",
|
|
||||||
lambda _: MimeType(name="application/x-ultimaker-material-profile", comment="Ultimaker Material Profile",
|
|
||||||
suffixes=["xml.fdm_material"]))
|
|
||||||
@patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: ["/materials/generic_pla_white.xml.fdm_material"])
|
|
||||||
@patch("plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice")
|
@patch("plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice")
|
||||||
def test_sendMaterialsToPrinter(self, device_mock):
|
|
||||||
device_mock._createFormPart.return_value = "_xXx_"
|
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
|
||||||
job = SendMaterialJob(device_mock)
|
|
||||||
job.sendMaterialsToPrinter({'generic_pla_white'})
|
|
||||||
|
|
||||||
self._assertLogEntries([("d", "Syncing material generic_pla_white with cluster.")], _logentries)
|
|
||||||
self.assertEqual([call._createFormPart('name="file"; filename="generic_pla_white.xml.fdm_material"', '<xml></xml>'),
|
|
||||||
call.postFormWithParts(on_finished=job.sendingFinished, parts = ["_xXx_"], target = "materials/")], device_mock.method_calls)
|
|
||||||
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
@patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
def test_sendingFinished_success(self, reply_mock) -> None:
|
def test_sendMissingMaterials_withBadJsonAnswer(self, reply_mock, device_mock):
|
||||||
reply_mock.attribute.return_value = 200
|
reply_mock.attribute.return_value = 200
|
||||||
with mock.patch.object(Logger, 'log', new=new_log):
|
|
||||||
SendMaterialJob(None).sendingFinished(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_once_with(0)
|
|
||||||
self.assertEqual(0, len(_logentries))
|
|
||||||
|
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
|
||||||
def test_sendingFinished_failed(self, reply_mock) -> None:
|
|
||||||
reply_mock.attribute.return_value = 404
|
|
||||||
reply_mock.readAll.return_value = QByteArray(b'Six sick hicks nick six slick bricks with picks and sticks.')
|
reply_mock.readAll.return_value = QByteArray(b'Six sick hicks nick six slick bricks with picks and sticks.')
|
||||||
|
SendMaterialJob(device_mock).run()
|
||||||
with mock.patch.object(Logger, 'log', new=new_log):
|
|
||||||
SendMaterialJob(None).sendingFinished(reply_mock)
|
|
||||||
|
|
||||||
reply_mock.attribute.assert_called_with(0)
|
reply_mock.attribute.assert_called_with(0)
|
||||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.attribute(0), call.readAll()])
|
self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
||||||
|
self.assertEqual(device_mock._onGetRemoteMaterials.method_calls, [])
|
||||||
|
|
||||||
self._assertLogEntries([
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
("e", "Received error code from printer when syncing material: 404"),
|
# def test_sendMissingMaterials_withMissingGuid(self, reply_mock):
|
||||||
("e", "Six sick hicks nick six slick bricks with picks and sticks.")
|
# reply_mock.attribute.return_value = 200
|
||||||
], _logentries)
|
# remoteMaterialWithoutGuid = self._REMOTEMATERIAL_WHITE.copy()
|
||||||
|
# del remoteMaterialWithoutGuid["guid"]
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
# reply_mock.readAll.return_value = QByteArray(json.dumps([remoteMaterialWithoutGuid]).encode("ascii"))
|
||||||
def test_parseReply(self, reply_mock):
|
#
|
||||||
reply_mock.readAll.return_value = QByteArray(json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
# with mock.patch.object(Logger, 'log', new=new_log):
|
||||||
|
# SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
||||||
response = SendMaterialJob._parseReply(reply_mock)
|
#
|
||||||
|
# reply_mock.attribute.assert_called_with(0)
|
||||||
self.assertTrue(len(response) == 1)
|
# self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
||||||
self.assertEqual(next(iter(response.values())), ClusterMaterial(**self._REMOTEMATERIAL_WHITE))
|
# self._assertLogEntries(
|
||||||
|
# [('e', "Request material storage on printer: Printer's answer was missing GUIDs.")],
|
||||||
@patch("PyQt5.QtNetwork.QNetworkReply")
|
# _logentries)
|
||||||
def test_parseReplyWithInvalidMaterial(self, reply_mock):
|
#
|
||||||
remoteMaterialWithInvalidVersion = self._REMOTEMATERIAL_WHITE.copy()
|
# @patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: [])
|
||||||
remoteMaterialWithInvalidVersion["version"] = "one"
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
reply_mock.readAll.return_value = QByteArray(json.dumps([remoteMaterialWithInvalidVersion]).encode("ascii"))
|
# def test_sendMissingMaterials_WithInvalidVersionInLocalMaterial(self, reply_mock):
|
||||||
|
# reply_mock.attribute.return_value = 200
|
||||||
with self.assertRaises(ValueError):
|
# reply_mock.readAll.return_value = QByteArray(json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
||||||
SendMaterialJob._parseReply(reply_mock)
|
#
|
||||||
|
# containerRegistry = ContainerRegistryMock()
|
||||||
def test__getLocalMaterials(self):
|
# localMaterialWhiteWithInvalidVersion = self._LOCALMATERIAL_WHITE.copy()
|
||||||
containerRegistry = ContainerRegistryMock()
|
# localMaterialWhiteWithInvalidVersion["version"] = "one"
|
||||||
containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, self._LOCALMATERIAL_BLACK])
|
# containerRegistry.setContainersMetadata([localMaterialWhiteWithInvalidVersion])
|
||||||
|
#
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
# with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
||||||
local_materials = SendMaterialJob(None)._getLocalMaterials()
|
# SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
||||||
|
#
|
||||||
self.assertTrue(len(local_materials) == 2)
|
# reply_mock.attribute.assert_called_with(0)
|
||||||
|
# self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
||||||
def test__getLocalMaterialsWithMultipleVersions(self):
|
# self._assertLogEntries([('e', "Material generic_pla_white has invalid version number one.")], _logentries)
|
||||||
containerRegistry = ContainerRegistryMock()
|
#
|
||||||
localMaterialWithNewerVersion = self._LOCALMATERIAL_WHITE.copy()
|
# @patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: [])
|
||||||
localMaterialWithNewerVersion["version"] = 2
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, localMaterialWithNewerVersion])
|
# def test_sendMissingMaterials_WithMultipleLocalVersionsLowFirst(self, reply_mock):
|
||||||
|
# reply_mock.attribute.return_value = 200
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
# reply_mock.readAll.return_value = QByteArray(json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
||||||
with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
#
|
||||||
local_materials = SendMaterialJob(None)._getLocalMaterials()
|
# containerRegistry = ContainerRegistryMock()
|
||||||
|
# localMaterialWhiteWithHigherVersion = self._LOCALMATERIAL_WHITE.copy()
|
||||||
self.assertTrue(len(local_materials) == 1)
|
# localMaterialWhiteWithHigherVersion["version"] = "2"
|
||||||
self.assertTrue(list(local_materials.values())[0].version == 2)
|
# containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, localMaterialWhiteWithHigherVersion])
|
||||||
|
#
|
||||||
containerRegistry.setContainersMetadata([localMaterialWithNewerVersion, self._LOCALMATERIAL_WHITE])
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
|
# with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
||||||
with mock.patch.object(Logger, "log", new=new_log):
|
# SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
||||||
with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
#
|
||||||
local_materials = SendMaterialJob(None)._getLocalMaterials()
|
# reply_mock.attribute.assert_called_with(0)
|
||||||
|
# self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
||||||
self.assertTrue(len(local_materials) == 1)
|
# self._assertLogEntries([], _logentries)
|
||||||
self.assertTrue(list(local_materials.values())[0].version == 2)
|
#
|
||||||
|
# @patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: [])
|
||||||
def _assertLogEntries(self, first, second):
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
"""
|
# def test_sendMissingMaterials_MaterialMissingOnPrinter(self, reply_mock):
|
||||||
Inspects the two sets of log entry tuples and fails when they are not the same
|
# reply_mock.attribute.return_value = 200
|
||||||
:param first: The first set of tuples
|
# reply_mock.readAll.return_value = QByteArray(
|
||||||
:param second: The second set of tuples
|
# json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
||||||
"""
|
#
|
||||||
self.assertEqual(len(first), len(second))
|
# containerRegistry = ContainerRegistryMock()
|
||||||
|
# containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, self._LOCALMATERIAL_BLACK])
|
||||||
while len(first) > 0:
|
#
|
||||||
e1, m1 = first[0]
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
e2, m2 = second[0]
|
# with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
||||||
self.assertEqual(e1, e2)
|
# SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
||||||
self.assertEqual(m1, m2)
|
#
|
||||||
first.pop(0)
|
# reply_mock.attribute.assert_called_with(0)
|
||||||
second.pop(0)
|
# self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.readAll()])
|
||||||
|
# self._assertLogEntries([], _logentries)
|
||||||
|
#
|
||||||
|
# @patch("builtins.open", lambda a, b: io.StringIO("<xml></xml>"))
|
||||||
|
# @patch("UM.MimeTypeDatabase.MimeTypeDatabase.getMimeTypeForFile",
|
||||||
|
# lambda _: MimeType(name="application/x-ultimaker-material-profile", comment="Ultimaker Material Profile",
|
||||||
|
# suffixes=["xml.fdm_material"]))
|
||||||
|
# @patch("UM.Resources.Resources.getAllResourcesOfType", lambda _: ["/materials/generic_pla_white.xml.fdm_material"])
|
||||||
|
# @patch("plugins.UM3NetworkPrinting.src.ClusterUM3OutputDevice")
|
||||||
|
# def test_sendMaterialsToPrinter(self, device_mock):
|
||||||
|
# device_mock._createFormPart.return_value = "_xXx_"
|
||||||
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
|
# job = SendMaterialJob(device_mock)
|
||||||
|
# job.sendMaterialsToPrinter({'generic_pla_white'})
|
||||||
|
#
|
||||||
|
# self._assertLogEntries([("d", "Syncing material generic_pla_white with cluster.")], _logentries)
|
||||||
|
# self.assertEqual([call._createFormPart('name="file"; filename="generic_pla_white.xml.fdm_material"', '<xml></xml>'),
|
||||||
|
# call.postFormWithParts(on_finished=job.sendingFinished, parts = ["_xXx_"], target = "materials/")], device_mock.method_calls)
|
||||||
|
#
|
||||||
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
|
# def test_sendingFinished_success(self, reply_mock) -> None:
|
||||||
|
# reply_mock.attribute.return_value = 200
|
||||||
|
# with mock.patch.object(Logger, 'log', new=new_log):
|
||||||
|
# SendMaterialJob(None).sendingFinished(reply_mock)
|
||||||
|
#
|
||||||
|
# reply_mock.attribute.assert_called_once_with(0)
|
||||||
|
# self.assertEqual(0, len(_logentries))
|
||||||
|
#
|
||||||
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
|
# def test_sendingFinished_failed(self, reply_mock) -> None:
|
||||||
|
# reply_mock.attribute.return_value = 404
|
||||||
|
# reply_mock.readAll.return_value = QByteArray(b'Six sick hicks nick six slick bricks with picks and sticks.')
|
||||||
|
#
|
||||||
|
# with mock.patch.object(Logger, 'log', new=new_log):
|
||||||
|
# SendMaterialJob(None).sendingFinished(reply_mock)
|
||||||
|
#
|
||||||
|
# reply_mock.attribute.assert_called_with(0)
|
||||||
|
# self.assertEqual(reply_mock.method_calls, [call.attribute(0), call.attribute(0), call.readAll()])
|
||||||
|
#
|
||||||
|
# self._assertLogEntries([
|
||||||
|
# ("e", "Received error code from printer when syncing material: 404"),
|
||||||
|
# ("e", "Six sick hicks nick six slick bricks with picks and sticks.")
|
||||||
|
# ], _logentries)
|
||||||
|
#
|
||||||
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
|
# def test_parseReply(self, reply_mock):
|
||||||
|
# reply_mock.readAll.return_value = QByteArray(json.dumps([self._REMOTEMATERIAL_WHITE]).encode("ascii"))
|
||||||
|
#
|
||||||
|
# response = SendMaterialJob._parseReply(reply_mock)
|
||||||
|
#
|
||||||
|
# self.assertTrue(len(response) == 1)
|
||||||
|
# self.assertEqual(next(iter(response.values())), ClusterMaterial(**self._REMOTEMATERIAL_WHITE))
|
||||||
|
#
|
||||||
|
# @patch("PyQt5.QtNetwork.QNetworkReply")
|
||||||
|
# def test_parseReplyWithInvalidMaterial(self, reply_mock):
|
||||||
|
# remoteMaterialWithInvalidVersion = self._REMOTEMATERIAL_WHITE.copy()
|
||||||
|
# remoteMaterialWithInvalidVersion["version"] = "one"
|
||||||
|
# reply_mock.readAll.return_value = QByteArray(json.dumps([remoteMaterialWithInvalidVersion]).encode("ascii"))
|
||||||
|
#
|
||||||
|
# with self.assertRaises(ValueError):
|
||||||
|
# SendMaterialJob._parseReply(reply_mock)
|
||||||
|
#
|
||||||
|
# def test__getLocalMaterials(self):
|
||||||
|
# containerRegistry = ContainerRegistryMock()
|
||||||
|
# containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, self._LOCALMATERIAL_BLACK])
|
||||||
|
#
|
||||||
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
|
# with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
||||||
|
# local_materials = SendMaterialJob(None)._getLocalMaterials()
|
||||||
|
#
|
||||||
|
# self.assertTrue(len(local_materials) == 2)
|
||||||
|
#
|
||||||
|
# def test__getLocalMaterialsWithMultipleVersions(self):
|
||||||
|
# containerRegistry = ContainerRegistryMock()
|
||||||
|
# localMaterialWithNewerVersion = self._LOCALMATERIAL_WHITE.copy()
|
||||||
|
# localMaterialWithNewerVersion["version"] = 2
|
||||||
|
# containerRegistry.setContainersMetadata([self._LOCALMATERIAL_WHITE, localMaterialWithNewerVersion])
|
||||||
|
#
|
||||||
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
|
# with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
||||||
|
# local_materials = SendMaterialJob(None)._getLocalMaterials()
|
||||||
|
#
|
||||||
|
# self.assertTrue(len(local_materials) == 1)
|
||||||
|
# self.assertTrue(list(local_materials.values())[0].version == 2)
|
||||||
|
#
|
||||||
|
# containerRegistry.setContainersMetadata([localMaterialWithNewerVersion, self._LOCALMATERIAL_WHITE])
|
||||||
|
#
|
||||||
|
# with mock.patch.object(Logger, "log", new=new_log):
|
||||||
|
# with mock.patch.object(ContainerRegistry, "getInstance", lambda: containerRegistry):
|
||||||
|
# local_materials = SendMaterialJob(None)._getLocalMaterials()
|
||||||
|
#
|
||||||
|
# self.assertTrue(len(local_materials) == 1)
|
||||||
|
# self.assertTrue(list(local_materials.values())[0].version == 2)
|
||||||
|
#
|
||||||
|
# def _assertLogEntries(self, first, second):
|
||||||
|
# """
|
||||||
|
# Inspects the two sets of log entry tuples and fails when they are not the same
|
||||||
|
# :param first: The first set of tuples
|
||||||
|
# :param second: The second set of tuples
|
||||||
|
# """
|
||||||
|
# self.assertEqual(len(first), len(second))
|
||||||
|
#
|
||||||
|
# while len(first) > 0:
|
||||||
|
# e1, m1 = first[0]
|
||||||
|
# e2, m2 = second[0]
|
||||||
|
# self.assertEqual(e1, e2)
|
||||||
|
# self.assertEqual(m1, m2)
|
||||||
|
# first.pop(0)
|
||||||
|
# second.pop(0)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue