HAP-python/tests/test_state.py at 4.9.2 · ikalchev/HAP-python · GitHub

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
"""Test for pyhap.state."""
from unittest.mock import patch
from uuid import UUID

from cryptography.hazmat.primitives.asymmetric import ed25519
import pytest

from pyhap.const import CLIENT_PROP_PERMS, HAP_PERMISSIONS
from pyhap.state import State

CLIENT_UUID = UUID("7d0d1ee9-46fe-4a56-a115-69df3f6860c1")
CLIENT_UUID_BYTES = str(CLIENT_UUID).upper().encode("utf-8")
CLIENT2_UUID = UUID("7d0d1ee9-46fe-4a56-a115-69df3f6860c2")
CLIENT2_UUID_BYTES = str(CLIENT2_UUID).upper().encode("utf-8")


def test_setup():
    """Test if State class is setup correctly."""
    with pytest.raises(TypeError):
        State("invalid_argument")

    addr = "172.0.0.1"
    mac = "00:00:00:00:00:00"
    pin = b"123-45-678"
    port = 11111

    private_key = ed25519.Ed25519PrivateKey.generate()

    with patch("pyhap.util.get_local_address") as mock_local_addr, patch(
        "pyhap.util.generate_mac"
    ) as mock_gen_mac, patch("pyhap.util.generate_pincode") as mock_gen_pincode, patch(
        "pyhap.util.generate_setup_id"
    ) as mock_gen_setup_id, patch(
        "cryptography.hazmat.primitives.asymmetric.ed25519.Ed25519PrivateKey.generate",
        return_value=private_key,
    ) as mock_create_keypair:
        state = State(address=addr, mac=mac, pincode=pin, port=port)
        assert not mock_local_addr.called
        assert not mock_gen_mac.called
        assert not mock_gen_pincode.called
        assert mock_gen_setup_id.called
        assert mock_create_keypair.called

        assert state.address == addr
        assert state.addresses == [addr]
        assert state.mac == mac
        assert state.pincode == pin
        assert state.port == port

        state = State()
        assert mock_local_addr.called
        assert mock_gen_mac.called
        assert mock_gen_pincode.called
        assert state.port == 51827
        assert state.config_version == 1


def test_pairing_remove_last_admin():
    """Test if pairing methods work."""
    with patch("pyhap.util.get_local_address"), patch("pyhap.util.generate_mac"), patch(
        "pyhap.util.generate_pincode"
    ), patch("pyhap.util.generate_setup_id"):
        state = State()

    assert not state.paired
    assert not state.paired_clients

    state.add_paired_client(CLIENT_UUID_BYTES, "public", HAP_PERMISSIONS.ADMIN)
    assert state.paired
    assert state.paired_clients == {CLIENT_UUID: "public"}
    assert state.client_properties == {CLIENT_UUID: {CLIENT_PROP_PERMS: 1}}

    state.add_paired_client(CLIENT2_UUID_BYTES, "public", HAP_PERMISSIONS.USER)
    assert state.paired
    assert state.paired_clients == {CLIENT_UUID: "public", CLIENT2_UUID: "public"}
    assert state.client_properties == {
        CLIENT_UUID: {CLIENT_PROP_PERMS: 1},
        CLIENT2_UUID: {CLIENT_PROP_PERMS: 0},
    }
    assert state.uuid_to_bytes == {
        CLIENT_UUID: CLIENT_UUID_BYTES,
        CLIENT2_UUID: CLIENT2_UUID_BYTES,
    }

    # Removing the last admin should remove all non-admins
    state.remove_paired_client(CLIENT_UUID)
    assert not state.paired
    assert not state.paired_clients


def test_pairing_two_admins():
    """Test if pairing methods work."""
    with patch("pyhap.util.get_local_address"), patch("pyhap.util.generate_mac"), patch(
        "pyhap.util.generate_pincode"
    ), patch("pyhap.util.generate_setup_id"):
        state = State()

    assert not state.paired
    assert not state.paired_clients

    state.add_paired_client(CLIENT_UUID_BYTES, "public", HAP_PERMISSIONS.ADMIN)
    assert state.paired
    assert state.paired_clients == {CLIENT_UUID: "public"}
    assert state.client_properties == {CLIENT_UUID: {CLIENT_PROP_PERMS: 1}}

    state.add_paired_client(CLIENT2_UUID_BYTES, "public", HAP_PERMISSIONS.ADMIN)
    assert state.paired
    assert state.paired_clients == {CLIENT_UUID: "public", CLIENT2_UUID: "public"}
    assert state.client_properties == {
        CLIENT_UUID: {CLIENT_PROP_PERMS: 1},
        CLIENT2_UUID: {CLIENT_PROP_PERMS: 1},
    }

    # Removing the admin should leave the other admin
    state.remove_paired_client(CLIENT2_UUID)
    assert state.paired
    assert state.paired_clients == {CLIENT_UUID: "public"}
    assert state.client_properties == {CLIENT_UUID: {CLIENT_PROP_PERMS: 1}}
    assert not state.is_admin(CLIENT2_UUID)