STAR-322: Creating a network manager mock

plugins/UM3NetworkPrinting/tests/Cloud/NetworkManagerMock.py

@@ -0,0 +1,42 @@
+# Copyright (c) 2018 Ultimaker B.V.
+# Cura is released under the terms of the LGPLv3 or higher.
+import json
+from unittest.mock import MagicMock
+
+from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkReply, QNetworkRequest
+
+from UM.Signal import Signal
+
+
+class NetworkManagerMock:
+    finished = Signal()
+    authenticationRequired = Signal()
+
+    _OPERATIONS = {
+        "GET": QNetworkAccessManager.GetOperation,
+        "POST": QNetworkAccessManager.PostOperation,
+        "PUT": QNetworkAccessManager.PutOperation,
+        "DELETE": QNetworkAccessManager.DeleteOperation,
+        "HEAD": QNetworkAccessManager.HeadOperation,
+    }
+
+    def __getattr__(self, item):
+        operation = self._OPERATIONS.get(item.upper())
+        if operation:
+            return lambda request, *_: self._fakeReply(operation, request)
+        return super().__getattribute__(item)
+
+    def _fakeReply(self, operation: QNetworkAccessManager.Operation, request: QNetworkRequest) -> QNetworkReply:
+        reply_mock = MagicMock()
+        reply_mock.url = request.url
+        reply_mock.operation.return_value = operation
+        return reply_mock
+
+    def respond(self, method: str, url: str, status_code: int, response: dict) -> QNetworkReply:
+        reply_mock = MagicMock()
+        reply_mock.url().toString.return_value = url
+        reply_mock.operation.return_value = self._OPERATIONS[method]
+        reply_mock.attribute.return_value = status_code
+        reply_mock.readAll.return_value = json.dumps(response).encode()
+        self.finished.emit(reply_mock)
+        return reply_mock