Skip to content

Commit 0ce3abe

Browse files
committed
Adapt testing
1 parent df02262 commit 0ce3abe

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/test_usb.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1704,29 +1704,29 @@ async def makedirs(cache_dir: str, exist_ok: bool) -> None:
17041704
await pw_nw_cache.initialize_cache()
17051705
# test with invalid data
17061706
mock_read_data = [
1707-
"0123456789ABCDEF;NodeType.CIRCLE_PLUS",
1707+
"0123456789ABCDEF;NodeType.CIRCLE",
17081708
"FEDCBA9876543210;None",
17091709
]
17101710
file_chunks_iter = iter(mock_read_data)
17111711
mock_file_stream = MagicMock(readlines=lambda *args, **kwargs: file_chunks_iter)
17121712
with patch("aiofiles.threadpool.sync_open", return_value=mock_file_stream):
17131713
await pw_nw_cache.restore_cache()
17141714
assert pw_nw_cache.nodetypes == {
1715-
"0123456789ABCDEF": pw_api.NodeType.CIRCLE_PLUS,
1715+
"0123456789ABCDEF": pw_api.NodeType.CIRCLE,
17161716
}
17171717

17181718
# test with valid data
17191719
mock_read_data = [
1720-
"0123456789ABCDEF;CIRCLE_PLUS",
1721-
"FEDCBA9876543210;NodeType.CIRCLE",
1720+
"0123456789ABCDEF;NodeType.CIRCLE",
1721+
"FEDCBA9876543210;CIRCLE",
17221722
"1298347650AFBECD;6",
17231723
]
17241724
file_chunks_iter = iter(mock_read_data)
17251725
mock_file_stream = MagicMock(readlines=lambda *args, **kwargs: file_chunks_iter)
17261726
with patch("aiofiles.threadpool.sync_open", return_value=mock_file_stream):
17271727
await pw_nw_cache.restore_cache()
17281728
assert pw_nw_cache.nodetypes == {
1729-
"0123456789ABCDEF": pw_api.NodeType.CIRCLE_PLUS,
1729+
"0123456789ABCDEF": pw_api.NodeType.CIRCLE,
17301730
"FEDCBA9876543210": pw_api.NodeType.CIRCLE,
17311731
"1298347650AFBECD": pw_api.NodeType.SCAN,
17321732
}
@@ -1738,14 +1738,14 @@ async def makedirs(cache_dir: str, exist_ok: bool) -> None:
17381738
)
17391739
mock_file_stream.writelines.assert_called_with(
17401740
[
1741-
"0123456789ABCDEF;CIRCLE_PLUS\n",
1741+
"0123456789ABCDEF;CIRCLE\n",
17421742
"FEDCBA9876543210;CIRCLE\n",
17431743
"1298347650AFBECD;SCAN\n",
17441744
"1234ABCD4321FEDC;STEALTH\n",
17451745
]
17461746
)
17471747
assert pw_nw_cache.nodetypes == {
1748-
"0123456789ABCDEF": pw_api.NodeType.CIRCLE_PLUS,
1748+
"0123456789ABCDEF": pw_api.NodeType.CIRCLE,
17491749
"FEDCBA9876543210": pw_api.NodeType.CIRCLE,
17501750
"1298347650AFBECD": pw_api.NodeType.SCAN,
17511751
"1234ABCD4321FEDC": pw_api.NodeType.STEALTH,

0 commit comments

Comments
 (0)