mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-12 09:17:50 -06:00
Rewrite tests
This commit is contained in:
ChrisTerBeke
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
|
||||
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.
|
||||
#
|
||||
|
|
|
|||
|
|
@ -1,6 +1,5 @@
|
|||
# Copyright (c) 2018 Ultimaker B.V.
|
||||
# Cura is released under the terms of the LGPLv3 or higher.
|
||||
|
||||
import io
|
||||
import json
|
||||
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.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):
|
||||
|
||||
def __init__(self):
|
||||
self.containersMetaData = None
|
||||
|
||||
|
|
@ -59,14 +51,16 @@ class FakeDevice(ClusterUM3OutputDevice):
|
|||
|
||||
|
||||
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',
|
||||
'brand': 'Generic', 'material': 'PLA', 'color_name': 'White',
|
||||
'GUID': 'badb0ee7-87c8-4f3f-9398-938587b67dce', 'version': '1', 'color_code': '#ffffff',
|
||||
'description': 'Test PLA White', 'adhesion_info': 'Use glue.', 'approximate_diameter': '3',
|
||||
'properties': {'density': '1.00', 'diameter': '2.85', 'weight': '750'},
|
||||
'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',
|
||||
'brand': 'Ultimaker', 'material': 'CPE', 'color_name': 'Black',
|
||||
'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'},
|
||||
'definition': 'fdmprinter', 'compatible': True}
|
||||
|
||||
_REMOTEMATERIAL_WHITE = {
|
||||
_REMOTE_MATERIAL_WHITE = {
|
||||
"guid": "badb0ee7-87c8-4f3f-9398-938587b67dce",
|
||||
"material": "PLA",
|
||||
"brand": "Generic",
|
||||
|
|
@ -82,7 +76,8 @@ class TestSendMaterialJob(TestCase):
|
|||
"color": "White",
|
||||
"density": 1.00
|
||||
}
|
||||
_REMOTEMATERIAL_BLACK = {
|
||||
|
||||
_REMOTE_MATERIAL_BLACK = {
|
||||
"guid": "5fbb362a-41f9-4818-bb43-15ea6df34aa4",
|
||||
"material": "PLA",
|
||||
"brand": "Generic",
|
||||
|
|
@ -91,224 +86,206 @@ class TestSendMaterialJob(TestCase):
|
|||
"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")
|
||||
def test_run(self, device_mock):
|
||||
with mock.patch.object(Logger, 'log', new=new_log):
|
||||
job = SendMaterialJob(device_mock)
|
||||
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")
|
||||
def test_sendMissingMaterials_withFailedRequest(self, reply_mock):
|
||||
def test_sendMissingMaterials_withFailedRequest(self, reply_mock, device_mock):
|
||||
reply_mock.attribute.return_value = 404
|
||||
|
||||
with mock.patch.object(Logger, 'log', new=new_log):
|
||||
SendMaterialJob(None).sendMissingMaterials(reply_mock)
|
||||
|
||||
SendMaterialJob(device_mock).run()
|
||||
reply_mock.attribute.assert_called_with(0)
|
||||
self.assertEqual(reply_mock.method_calls, [call.attribute(0)])
|
||||
self._assertLogEntries([('e', "Couldn't request current material storage on printer. Not syncing materials.")],
|
||||
_logentries)
|
||||
self.assertEqual(device_mock._onGetRemoteMaterials.method_calls, [])
|
||||
|
||||
@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")
|
||||
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:
|
||||
def test_sendMissingMaterials_withBadJsonAnswer(self, reply_mock, device_mock):
|
||||
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)
|
||||
|
||||
SendMaterialJob(device_mock).run()
|
||||
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([
|
||||
("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)
|
||||
# @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")
|
||||
# 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