1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
"""Test the firmware update helper."""
from unittest.mock import patch
from zwave_js_server.firmware import driver_firmware_update_otw, update_firmware
from zwave_js_server.model.driver.firmware import (
DriverFirmwareUpdateData,
DriverFirmwareUpdateStatus,
)
from zwave_js_server.model.node.firmware import (
NodeFirmwareUpdateData,
NodeFirmwareUpdateStatus,
)
async def test_update_firmware_guess_format(url, client_session, multisensor_6):
"""Test update_firmware with guessed format."""
with (
patch("zwave_js_server.firmware.Client.connect") as connect_mock,
patch("zwave_js_server.firmware.Client.initialize") as initialize_mock,
patch("zwave_js_server.firmware.Client.async_send_command") as cmd_mock,
patch("zwave_js_server.firmware.Client.disconnect") as disconnect_mock,
):
node = multisensor_6
cmd_mock.return_value = {
"result": {"status": 255, "success": True, "reInterview": False}
}
result = await update_firmware(
url, node, [NodeFirmwareUpdateData("test", bytes(10))], client_session
)
assert result.status == NodeFirmwareUpdateStatus.OK_RESTART_PENDING
assert result.success
assert not result.reinterview
connect_mock.assert_called_once()
initialize_mock.assert_called_once()
cmd_mock.assert_called_once_with(
{
"command": "node.update_firmware",
"nodeId": node.node_id,
"updates": [
{
"filename": "test",
"file": "AAAAAAAAAAAAAA==",
}
],
},
require_schema=29,
)
disconnect_mock.assert_called_once()
async def test_update_firmware_known_format_and_target(
url, client_session, multisensor_6
):
"""Test update_firmware with known format."""
with (
patch("zwave_js_server.firmware.Client.connect") as connect_mock,
patch("zwave_js_server.firmware.Client.initialize") as initialize_mock,
patch("zwave_js_server.firmware.Client.async_send_command") as cmd_mock,
patch("zwave_js_server.firmware.Client.disconnect") as disconnect_mock,
):
node = multisensor_6
cmd_mock.return_value = {
"result": {"status": 255, "success": True, "reInterview": False}
}
result = await update_firmware(
url=url,
node=node,
updates=[NodeFirmwareUpdateData("test", bytes(10), "test", 1)],
session=client_session,
)
assert result.status == NodeFirmwareUpdateStatus.OK_RESTART_PENDING
assert result.success
assert not result.reinterview
connect_mock.assert_called_once()
initialize_mock.assert_called_once()
cmd_mock.assert_called_once_with(
{
"command": "node.update_firmware",
"nodeId": node.node_id,
"updates": [
{
"filename": "test",
"file": "AAAAAAAAAAAAAA==",
"fileFormat": "test",
"firmwareTarget": 1,
}
],
},
require_schema=29,
)
disconnect_mock.assert_called_once()
async def test_driver_firmware_update_otw_guess_format(url, client_session):
"""Test driver_firmware_update_otw with guessed format."""
with (
patch("zwave_js_server.firmware.Client.connect") as connect_mock,
patch("zwave_js_server.firmware.Client.initialize") as initialize_mock,
patch("zwave_js_server.firmware.Client.async_send_command") as cmd_mock,
patch("zwave_js_server.firmware.Client.disconnect") as disconnect_mock,
):
cmd_mock.return_value = {"result": {"status": 255, "success": True}}
result = await driver_firmware_update_otw(
url, DriverFirmwareUpdateData("test", bytes(10)), client_session
)
assert result.status == DriverFirmwareUpdateStatus.OK
assert result.success
connect_mock.assert_called_once()
initialize_mock.assert_called_once()
cmd_mock.assert_called_once_with(
{
"command": "driver.firmware_update_otw",
"filename": "test",
"file": "AAAAAAAAAAAAAA==",
},
require_schema=29,
)
disconnect_mock.assert_called_once()
async def test_driver_firmware_update_otw_known_format_and_target(url, client_session):
"""Test driver_firmware_update_otw with known format."""
with (
patch("zwave_js_server.firmware.Client.connect") as connect_mock,
patch("zwave_js_server.firmware.Client.initialize") as initialize_mock,
patch("zwave_js_server.firmware.Client.async_send_command") as cmd_mock,
patch("zwave_js_server.firmware.Client.disconnect") as disconnect_mock,
):
cmd_mock.return_value = {"result": {"status": 255, "success": True}}
result = await driver_firmware_update_otw(
url=url,
firmware_file=DriverFirmwareUpdateData("test", bytes(10), "test"),
session=client_session,
)
assert result.status == DriverFirmwareUpdateStatus.OK
assert result.success
connect_mock.assert_called_once()
initialize_mock.assert_called_once()
cmd_mock.assert_called_once_with(
{
"command": "driver.firmware_update_otw",
"filename": "test",
"file": "AAAAAAAAAAAAAA==",
"fileFormat": "test",
},
require_schema=29,
)
disconnect_mock.assert_called_once()
|
