mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-06-04 11:14:21 +08:00
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
# Copyright (c) 2018 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import pytest
|
|
import json
|
|
|
|
import unittest.mock
|
|
from cura.CuraApplication import CuraApplication
|
|
|
|
from UM.Version import Version
|
|
|
|
from plugins.FirmwareUpdateChecker.FirmwareUpdateCheckerJob import FirmwareUpdateCheckerJob
|
|
from plugins.FirmwareUpdateChecker.FirmwareUpdateCheckerLookup import FirmwareUpdateCheckerLookup
|
|
|
|
json_data = \
|
|
{
|
|
"ned":
|
|
{
|
|
"id": 1,
|
|
"name": "ned",
|
|
"check_urls": ["http://urlecho.appspot.com/echo?status=200&body=5.1.2.3"],
|
|
"update_url": "https://ultimaker.com/en/resources/20500-upgrade-firmware",
|
|
"version_parser": "default"
|
|
},
|
|
"olivia":
|
|
{
|
|
"id": 3,
|
|
"name": "olivia",
|
|
"check_urls": ["http://urlecho.appspot.com/echo?status=200&body=4.3.2.1", "https://this-url-clearly-doesnt-exist.net"],
|
|
"update_url": "https://ultimaker.com/en/resources/20500-upgrade-firmware",
|
|
"version_parser": "default"
|
|
},
|
|
"emmerson":
|
|
{
|
|
"id": 5,
|
|
"name": "emmerson",
|
|
"check_urls": ["http://urlecho.appspot.com/echo?status=200&body=0.2.2.2", "http://urlecho.appspot.com/echo?status=200&body=6.7.8.1"],
|
|
"update_url": "https://ultimaker.com/en/resources/20500-upgrade-firmware",
|
|
"version_parser": "default"
|
|
}
|
|
}
|
|
|
|
def dummyCallback():
|
|
pass
|
|
|
|
@pytest.mark.parametrize("name, id", [
|
|
("ned" , 1),
|
|
("olivia" , 3),
|
|
("emmerson", 5),
|
|
])
|
|
def test_FirmwareUpdateCheckerLookup(id, name):
|
|
lookup = FirmwareUpdateCheckerLookup(name, json_data.get(name))
|
|
|
|
assert lookup.getMachineName() == name
|
|
assert lookup.getMachineId() == id
|
|
assert len(lookup.getCheckUrls()) >= 1
|
|
assert lookup.getRedirectUserUrl() is not None
|
|
|
|
@pytest.mark.parametrize("name, version", [
|
|
("ned" , Version("5.1.2.3")),
|
|
("olivia" , Version("4.3.2.1")),
|
|
("emmerson", Version("6.7.8.1")),
|
|
])
|
|
def test_FirmwareUpdateCheckerJob_getCurrentVersion(name, version):
|
|
machine_data = json_data.get(name)
|
|
job = FirmwareUpdateCheckerJob(False, name, machine_data, dummyCallback)
|
|
job._headers = {"User-Agent": "Cura-UnitTests 0"}
|
|
|
|
assert job.getCurrentVersion() == version
|