3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-07-23 14:44:13 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fix test for cloud output device ufp uploading

Browse source
This commit is contained in:
ChrisTerBeke 2019-02-15 17:03:38 +01:00
parent 119c868650
commit 897c932e80
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
plugins/UM3NetworkPrinting/tests/Cloud/TestCloudOutputDevice.py
View file

@ -120,7 +120,7 @@ class TestCloudOutputDevice(TestCase):
def test_print_to_cloud(self): def test_print_to_cloud(self):
active_machine_mock = self.app.getGlobalContainerStack.return_value active_machine_mock = self.app.getGlobalContainerStack.return_value
active_machine_mock.getMetaDataEntry.side_effect = {"file_formats": "application/gzip"}.get active_machine_mock.getMetaDataEntry.side_effect = {"file_formats": "application/x-ufp"}.get
request_upload_response = parseFixture("putJobUploadResponse") request_upload_response = parseFixture("putJobUploadResponse")
request_print_response = parseFixture("postJobPrintResponse") request_print_response = parseFixture("postJobPrintResponse")
@ -130,6 +130,10 @@ class TestCloudOutputDevice(TestCase):
file_handler = MagicMock() file_handler = MagicMock()
file_handler.getSupportedFileTypesWrite.return_value = [{ file_handler.getSupportedFileTypesWrite.return_value = [{
"extension": "ufp",
"mime_type": "application/x-ufp",
"mode": 2
}, {
"extension": "gcode.gz", "extension": "gcode.gz",
"mime_type": "application/gzip", "mime_type": "application/gzip",
"mode": 2, "mode": 2,
@ -138,15 +142,11 @@ class TestCloudOutputDevice(TestCase):
lambda stream, nodes: stream.write(str(nodes).encode()) lambda stream, nodes: stream.write(str(nodes).encode())
scene_nodes = [SceneNode()] scene_nodes = [SceneNode()]
expected_mesh = str(scene_nodes).encode()
self.device.requestWrite(scene_nodes, file_handler=file_handler, file_name="FileName") self.device.requestWrite(scene_nodes, file_handler=file_handler, file_name="FileName")
self.network.flushReplies() self.network.flushReplies()
self.assertEqual( self.assertEqual(
{"data": {"content_type": "application/gzip", "file_size": len(expected_mesh), "job_name": "FileName"}}, {"data": {"content_type": "application/x-ufp", "job_name": "FileName"}},
json.loads(self.network.getRequestBody("PUT", self.REQUEST_UPLOAD_URL).decode()) json.loads(self.network.getRequestBody("PUT", self.REQUEST_UPLOAD_URL).decode())
) )
self.assertEqual(expected_mesh,
self.network.getRequestBody("PUT", request_upload_response["data"]["upload_url"]))
self.assertIsNone(self.network.getRequestBody("POST", self.PRINT_URL)) self.assertIsNone(self.network.getRequestBody("POST", self.PRINT_URL))