File: test_doorlock.py

package info (click to toggle)
python-uiprotect 6.1.0-2.1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,848 kB
  • sloc: python: 16,451; makefile: 21
file content (160 lines) | stat: -rw-r--r-- 4,648 bytes parent folder | download
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
154
155
156
157
158
159
160
# mypy: disable-error-code="attr-defined, dict-item, assignment, union-attr"

from __future__ import annotations

from datetime import timedelta
from typing import TYPE_CHECKING

import pytest

from tests.conftest import TEST_CAMERA_EXISTS, TEST_DOORLOCK_EXISTS
from uiprotect.data.types import LockStatusType
from uiprotect.exceptions import BadRequest
from uiprotect.utils import to_ms

if TYPE_CHECKING:
    from uiprotect.data import Camera, Doorlock, Light


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.asyncio()
async def test_doorlock_set_paired_camera_none(doorlock_obj: Doorlock):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.camera_id = "bad_id"

    await doorlock_obj.set_paired_camera(None)

    doorlock_obj.api.api_request.assert_called_with(
        f"doorlocks/{doorlock_obj.id}",
        method="patch",
        json={"camera": None},
    )


@pytest.mark.skipif(
    not TEST_DOORLOCK_EXISTS or not TEST_CAMERA_EXISTS,
    reason="Missing testdata",
)
@pytest.mark.asyncio()
async def test_doorlock_set_paired_camera(doorlock_obj: Light, camera_obj: Camera):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.camera_id = None

    await doorlock_obj.set_paired_camera(camera_obj)

    doorlock_obj.api.api_request.assert_called_with(
        f"doorlocks/{doorlock_obj.id}",
        method="patch",
        json={"camera": camera_obj.id},
    )


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.parametrize("status", [True, False])
@pytest.mark.asyncio()
async def test_doorlock_set_status_light(doorlock_obj: Doorlock, status: bool):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.led_settings.is_enabled = not status

    await doorlock_obj.set_status_light(status)

    doorlock_obj.api.api_request.assert_called_with(
        f"doorlocks/{doorlock_obj.id}",
        method="patch",
        json={"ledSettings": {"isEnabled": status}},
    )


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.parametrize(
    "duration",
    [
        timedelta(seconds=0),
        timedelta(seconds=15),
        timedelta(seconds=3600),
        timedelta(seconds=3601),
    ],
)
@pytest.mark.asyncio()
async def test_doorlock_set_auto_close_time(
    doorlock_obj: Doorlock,
    duration: timedelta,
):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.auto_close_time = timedelta(seconds=30)

    duration_invalid = duration is not None and int(duration.total_seconds()) == 3601
    if duration_invalid:
        with pytest.raises(BadRequest):
            await doorlock_obj.set_auto_close_time(duration)
        assert not doorlock_obj.api.api_request.called
    else:
        await doorlock_obj.set_auto_close_time(duration)

        expected = {"autoCloseTimeMs": to_ms(duration)}

        doorlock_obj.api.api_request.assert_called_with(
            f"doorlocks/{doorlock_obj.id}",
            method="patch",
            json=expected,
        )


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.asyncio()
async def test_doorlock_close(doorlock_obj: Doorlock):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.lock_status = LockStatusType.OPEN

    await doorlock_obj.close_lock()

    doorlock_obj.api.api_request.assert_called_with(
        f"doorlocks/{doorlock_obj.id}/close",
        method="post",
    )


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.asyncio()
async def test_doorlock_close_invalid(doorlock_obj: Doorlock):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.lock_status = LockStatusType.CLOSED

    with pytest.raises(BadRequest):
        await doorlock_obj.close_lock()

    assert not doorlock_obj.api.api_request.called


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.asyncio()
async def test_doorlock_open(doorlock_obj: Doorlock):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.lock_status = LockStatusType.CLOSED

    await doorlock_obj.open_lock()

    doorlock_obj.api.api_request.assert_called_with(
        f"doorlocks/{doorlock_obj.id}/open",
        method="post",
    )


@pytest.mark.skipif(not TEST_DOORLOCK_EXISTS, reason="Missing testdata")
@pytest.mark.asyncio()
async def test_doorlock_open_invalid(doorlock_obj: Doorlock):
    doorlock_obj.api.api_request.reset_mock()

    doorlock_obj.lock_status = LockStatusType.OPEN

    with pytest.raises(BadRequest):
        await doorlock_obj.open_lock()

    assert not doorlock_obj.api.api_request.called