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
|
"""Unit tests for dlna."""
import ipaddress
import socket
from datetime import date, datetime, time, timedelta, timezone
import pytest
from async_upnp_client.utils import (
CaseInsensitiveDict,
async_get_local_ip,
get_local_ip,
parse_date_time,
str_to_time,
)
from .common import ADVERTISEMENT_HEADERS_DEFAULT
def test_case_insensitive_dict() -> None:
"""Test CaseInsensitiveDict."""
ci_dict = CaseInsensitiveDict()
ci_dict["Key"] = "value"
assert ci_dict["Key"] == "value"
assert ci_dict["key"] == "value"
assert ci_dict["KEY"] == "value"
assert CaseInsensitiveDict(key="value") == {"key": "value"}
assert CaseInsensitiveDict({"key": "value"}, key="override_value") == {
"key": "override_value"
}
def test_case_insensitive_dict_dict_equality() -> None:
"""Test CaseInsensitiveDict against dict equality."""
ci_dict = CaseInsensitiveDict()
ci_dict["Key"] = "value"
assert ci_dict == {"Key": "value"}
assert ci_dict == {"key": "value"}
assert ci_dict == {"KEY": "value"}
def test_case_insensitive_dict_profile() -> None:
"""Test CaseInsensitiveDict under load, for profiling."""
for _ in range(0, 10000):
assert (
CaseInsensitiveDict(ADVERTISEMENT_HEADERS_DEFAULT)
== ADVERTISEMENT_HEADERS_DEFAULT
)
def test_case_insensitive_dict_equality() -> None:
"""Test CaseInsensitiveDict equality."""
assert CaseInsensitiveDict(key="value") == CaseInsensitiveDict(KEY="value")
def test_str_to_time() -> None:
"""Test string to time parsing."""
assert str_to_time("0:0:10") == timedelta(hours=0, minutes=0, seconds=10)
assert str_to_time("0:10:0") == timedelta(hours=0, minutes=10, seconds=0)
assert str_to_time("10:0:0") == timedelta(hours=10, minutes=0, seconds=0)
assert str_to_time("0:0:10.10") == timedelta(
hours=0, minutes=0, seconds=10, milliseconds=10
)
assert str_to_time("+0:0:10") == timedelta(hours=0, minutes=0, seconds=10)
assert str_to_time("-0:0:10") == timedelta(hours=0, minutes=0, seconds=-10)
assert str_to_time("") is None
assert str_to_time(" ") is None
def test_parse_date_time() -> None:
"""Test string to datetime parsing."""
tz0 = timezone(timedelta(hours=0))
tz1 = timezone(timedelta(hours=1))
assert parse_date_time("2012-07-19") == date(2012, 7, 19)
assert parse_date_time("12:28:14") == time(12, 28, 14)
assert parse_date_time("2012-07-19 12:28:14") == datetime(2012, 7, 19, 12, 28, 14)
assert parse_date_time("2012-07-19T12:28:14") == datetime(2012, 7, 19, 12, 28, 14)
assert parse_date_time("12:28:14+01:00") == time(12, 28, 14, tzinfo=tz1)
assert parse_date_time("12:28:14 +01:00") == time(12, 28, 14, tzinfo=tz1)
assert parse_date_time("2012-07-19T12:28:14z") == datetime(
2012, 7, 19, 12, 28, 14, tzinfo=tz0
)
assert parse_date_time("2012-07-19T12:28:14Z") == datetime(
2012, 7, 19, 12, 28, 14, tzinfo=tz0
)
assert parse_date_time("2012-07-19T12:28:14+01:00") == datetime(
2012, 7, 19, 12, 28, 14, tzinfo=tz1
)
assert parse_date_time("2012-07-19T12:28:14 +01:00") == datetime(
2012, 7, 19, 12, 28, 14, tzinfo=tz1
)
TEST_ADDRESSES = [
None,
"8.8.8.8",
"8.8.8.8:80",
"http://8.8.8.8",
"google.com",
"http://google.com",
"http://google.com:443",
]
@pytest.mark.parametrize("target_url", TEST_ADDRESSES)
def test_get_local_ip(target_url: str) -> None:
"""Test getting of a local IP that is not loopback."""
local_ip_str = get_local_ip(target_url)
local_ip = ipaddress.ip_address(local_ip_str)
assert not local_ip.is_loopback
@pytest.mark.asyncio
@pytest.mark.parametrize("target_url", TEST_ADDRESSES)
async def test_async_get_local_ip(target_url: str) -> None:
"""Test getting of a local IP that is not loopback."""
addr_family, local_ip_str = await async_get_local_ip(target_url)
local_ip = ipaddress.ip_address(local_ip_str)
assert not local_ip.is_loopback
if local_ip.version == 4:
assert addr_family == socket.AddressFamily.AF_INET # pylint: disable=no-member
else:
assert addr_family == socket.AddressFamily.AF_INET6 # pylint: disable=no-member
|
