# Copyright (c) 2018 Ultimaker B.V.
# Copyright (c) 2018 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.
import json
import os
from unittest import TestCase
from unittest.mock import patch, MagicMock

from cura.CuraApplication import CuraApplication
from src.Cloud.CloudApiClient import CloudApiClient
from src.Cloud.CloudOutputDeviceManager import CloudOutputDeviceManager
from src.Cloud.Models.CloudJobResponse import CloudJobResponse
from src.Cloud.Models.CloudJobUploadRequest import CloudJobUploadRequest
from .NetworkManagerMock import NetworkManagerMock


@patch("cura.NetworkClient.QNetworkAccessManager")
class TestCloudApiClient(TestCase):
    def _errorHandler(self):
        pass

    def setUp(self):
        super().setUp()
        self.account = MagicMock()
        self.account.isLoggedIn.return_value = True

        self.app = CuraApplication.getInstance()
        self.network = NetworkManagerMock()
        self.manager = CloudOutputDeviceManager()
        self.api = CloudApiClient(self.account, self._errorHandler)

    def test_GetClusters(self, network_mock):
        network_mock.return_value = self.network

        result = []

        with open("{}/Fixtures/getClusters.json".format(os.path.dirname(__file__)), "rb") as f:
            response = f.read()

        self.network.prepareReply("GET", "https://api-staging.ultimaker.com/connect/v1/clusters", 200, response)
        # the callback is a function that adds the result of the call to getClusters to the result list
        self.api.getClusters(lambda clusters: result.extend(clusters))

        self.network.flushReplies()

        self.assertEqual(2, len(result))

    def test_getClusterStatus(self, network_mock):
        network_mock.return_value = self.network

        result = []

        with open("{}/Fixtures/getClusterStatusResponse.json".format(os.path.dirname(__file__)), "rb") as f:
            response = f.read()

        self.network.prepareReply("GET",
                                  "https://api-staging.ultimaker.com/connect/v1/clusters/R0YcLJwar1ugh0ikEZsZs8NWKV6vJP_LdYsXgXqAcaNC/status",
                                  200, response
                                  )
        self.api.getClusterStatus("R0YcLJwar1ugh0ikEZsZs8NWKV6vJP_LdYsXgXqAcaNC", lambda status: result.append(status))

        self.network.flushReplies()

        self.assertEqual(len(result), 1)
        status = result[0]

        self.assertEqual(len(status.printers), 2)
        self.assertEqual(len(status.print_jobs), 1)

    def test_requestUpload(self, network_mock):
        network_mock.return_value = self.network
        results = []

        with open("{}/Fixtures/requestUploadResponse.json".format(os.path.dirname(__file__)), "rb") as f:
            response = f.read()

        self.network.prepareReply("PUT", "https://api-staging.ultimaker.com/cura/v1/jobs/upload", 200, response)
        self.api.requestUpload(CloudJobUploadRequest(job_name = "job name", file_size = 143234, content_type = "text/plain"),
                               lambda r: results.append(r))
        self.network.flushReplies()

        self.assertEqual(results[0].content_type, "text/plain")
        self.assertEqual(results[0].status, "uploading")

    def test_uploadMesh(self, network_mock):
        network_mock.return_value = self.network
        results = []
        progress = MagicMock()

        with open("{}/Fixtures/requestUploadResponse.json".format(os.path.dirname(__file__)), "rb") as f:
            thedata = json.loads(f.read().decode("ascii"))
            data = thedata["data"]
            upload_response = CloudJobResponse(**data)

        self.network.prepareReply("PUT", upload_response.upload_url, 200,
                                  '{ data : "" }')  # Network client doesn't look into the reply

        self.api.uploadMesh(upload_response, b'', lambda job_id: results.append(job_id),
                            progress.advance, progress.error)

        self.network.flushReplies()

        self.assertEqual(len(results), 1)
        self.assertEqual(results[0], upload_response.job_id)

    def test_requestPrint(self, network_mock):
        network_mock.return_value = self.network
        results = []

        cluster_id = "NWKV6vJP_LdYsXgXqAcaNCR0YcLJwar1ugh0ikEZsZs8"
        job_id = "db34b096-c4d5-46f3-bea7-da6a19905e6c"

        with open("{}/Fixtures/requestPrintResponse.json".format(os.path.dirname(__file__)), "rb") as f:
            response = f.read()

        self.network.prepareReply("POST",
                                  "https://api-staging.ultimaker.com/connect/v1/clusters/{}/print/{}"
                                  .format(cluster_id, job_id),
                                  200, response)

        self.api.requestPrint(cluster_id, job_id, lambda r: results.append(r))

        self.network.flushReplies()

        self.assertEqual(len(results), 1)
        self.assertEqual(results[0].job_id, "db34b096-c4d5-46f3-bea7-da6a19905e6c")
        self.assertEqual(results[0].status, "queued")