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
|
"""Tests for the port forwarding module."""
from unittest.mock import AsyncMock
import pytest
from asusrouter.modules.port_forwarding import (
KEY_PORT_FORWARDING_STATE,
AsusPortForwarding,
set_state,
)
async_callback = AsyncMock()
@pytest.mark.asyncio
@pytest.mark.parametrize(
("state", "expect_modify", "expect_call", "expected_args"),
[
# Correct states
(AsusPortForwarding.ON, True, True, {KEY_PORT_FORWARDING_STATE: 1}),
(AsusPortForwarding.OFF, False, True, {KEY_PORT_FORWARDING_STATE: 0}),
# Wrong states
(AsusPortForwarding.UNKNOWN, False, False, {}),
(None, False, False, {}),
],
)
async def test_set_state(
state: AsusPortForwarding | None,
expect_modify: bool,
expect_call: bool,
expected_args: dict[str, int],
) -> None:
"""Test set_state."""
# Call the set_state function
await set_state(
callback=async_callback, state=state, expect_modify=expect_modify
)
# Check if the callback function was called
if expect_call:
async_callback.assert_called_once_with(
service="restart_firewall",
arguments=expected_args,
apply=True,
expect_modify=expect_modify,
)
else:
async_callback.assert_not_called()
# Reset the mock callback function
async_callback.reset_mock()
|
