Skip to content

Commit 0746277

Browse files
committed
Output only NodeType name into nodetypes-cache
1 parent abf7fa4 commit 0746277

File tree

2 files changed

+5
-6
lines changed

2 files changed

+5
-6
lines changed

plugwise_usb/network/cache.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,7 @@ async def save_cache(self) -> None:
2828
"""Save the node information to file."""
2929
cache_data_to_save: dict[str, str] = {}
3030
for mac, node_type in self._nodetypes.items():
31-
node_value = str(node_type)
32-
cache_data_to_save[mac] = node_value
31+
cache_data_to_save[mac] = node_type.name
3332
_LOGGER.debug("Save NodeTypes %s", str(len(cache_data_to_save)))
3433
await self.write_cache(cache_data_to_save, True) # rewrite set to True is required
3534

tests/test_usb.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1738,10 +1738,10 @@ async def makedirs(cache_dir: str, exist_ok: bool) -> None:
17381738
)
17391739
mock_file_stream.writelines.assert_called_with(
17401740
[
1741-
"0123456789ABCDEF;NodeType.CIRCLE_PLUS\n",
1742-
"FEDCBA9876543210;NodeType.CIRCLE\n",
1743-
"1298347650AFBECD;NodeType.SCAN\n",
1744-
"1234ABCD4321FEDC;NodeType.STEALTH\n",
1741+
"0123456789ABCDEF;CIRCLE_PLUS\n",
1742+
"FEDCBA9876543210;CIRCLE\n",
1743+
"1298347650AFBECD;SCAN\n",
1744+
"1234ABCD4321FEDC;STEALTH\n",
17451745
]
17461746
)
17471747
assert pw_nw_cache.nodetypes == {

0 commit comments

Comments
 (0)