test: add test case for miot spec

2025-06-21 15:20:00 +08:00 · 2024-12-13 21:07:43 +08:00
parent bac1c7027b
commit ad7f9eb610
3 changed files with 113 additions and 27 deletions
--- a/test/test_spec.py
+++ b/test/test_spec.py
@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+"""Unit test for miot_spec.py."""
+import json
+import random
+import time
+from urllib.request import Request, urlopen
+import pytest
+
+# pylint: disable=import-outside-toplevel, unused-argument
+
+
+@pytest.mark.parametrize('urn', [
+    'urn:miot-spec-v2:device:gateway:0000A019:xiaomi-hub1:3',
+    'urn:miot-spec-v2:device:light:0000A001:mijia-group3:3:0000C802',
+    'urn:miot-spec-v2:device:air-conditioner:0000A004:xiaomi-ar03r1:1',
+    'urn:miot-spec-v2:device:air-purifier:0000A007:xiaomi-va5:1:0000D050',
+    'urn:miot-spec-v2:device:humidifier:0000A00E:xiaomi-p800:1',
+    'urn:miot-spec-v2:device:curtain:0000A00C:xiaomi-acn010:1:0000D031',
+    'urn:miot-spec-v2:device:motion-sensor:0000A014:xiaomi-pir1:2',
+    'urn:miot-spec-v2:device:light:0000A001:philips-strip3:2'])
+@pytest.mark.asyncio
+@pytest.mark.dependency()
+async def test_spec_parse_async(test_cache_path, test_lang, urn):
+    from miot.miot_spec import MIoTSpecParser
+    from miot.miot_storage import MIoTStorage
+
+    storage = MIoTStorage(test_cache_path)
+    spec_parser = MIoTSpecParser(lang=test_lang, storage=storage)
+    await spec_parser.init_async()
+    assert await spec_parser.parse(urn=urn)
+
+
+@pytest.mark.parametrize('urn_list', [[
+    'urn:miot-spec-v2:device:gateway:0000A019:xiaomi-hub1:3',
+    'urn:miot-spec-v2:device:light:0000A001:mijia-group3:3:0000C802',
+    'urn:miot-spec-v2:device:air-conditioner:0000A004:xiaomi-ar03r1:1',
+    'urn:miot-spec-v2:device:air-purifier:0000A007:xiaomi-va5:1:0000D050',
+    'urn:miot-spec-v2:device:humidifier:0000A00E:xiaomi-p800:1',
+    'urn:miot-spec-v2:device:curtain:0000A00C:xiaomi-acn010:1:0000D031',
+    'urn:miot-spec-v2:device:motion-sensor:0000A014:xiaomi-pir1:2',
+    'urn:miot-spec-v2:device:light:0000A001:philips-strip3:2']])
+@pytest.mark.asyncio
+@pytest.mark.dependency()
+async def test_spec_refresh_async(test_cache_path, test_lang, urn_list):
+    from miot.miot_spec import MIoTSpecParser
+    from miot.miot_storage import MIoTStorage
+
+    storage = MIoTStorage(test_cache_path)
+    spec_parser = MIoTSpecParser(lang=test_lang, storage=storage)
+    await spec_parser.init_async()
+    assert await spec_parser.refresh_async(urn_list=urn_list) == len(urn_list)
+
+
+@pytest.mark.asyncio
+@pytest.mark.dependency()
+async def test_spec_random_parse_async(test_cache_path, test_lang):
+    from miot.miot_spec import MIoTSpecParser
+    from miot.miot_storage import MIoTStorage
+
+    test_count = 10
+    # get test data
+
+    def get_release_instance() -> list[str]:
+        request = Request(
+            'https://miot-spec.org/miot-spec-v2/instances?status=released',
+            method='GET')
+        with urlopen(request) as response:
+            content = response.read()
+            res_obj = json.loads(str(content, 'utf-8'))
+            result: list[str] = []
+            for item in res_obj['instances']:
+                result.append(item['type'])
+            return result
+    test_urns: list[str] = get_release_instance()
+    test_urn_index: list[int] = random.sample(
+        list(range(len(test_urns))), test_count)
+
+    # get local cache
+    storage = MIoTStorage(test_cache_path)
+    # local_urns = storage.get_names(domain='miot_specs', type_=dict)
+    # print('local urns, %s', local_urns)
+    spec_parser = MIoTSpecParser(lang=test_lang, storage=storage)
+    await spec_parser.init_async()
+    start: int = time.time()*1000
+    for index in test_urn_index:
+        urn: str = test_urns[int(index)]
+        result = await spec_parser.parse(urn=urn, skip_cache=True)
+        assert result is not None
+        # print('parse, %s, %s\n', urn, json.dumps(result.dump()))
+    end: int = time.time()*1000
+    print(f'takes time, {test_count}, {end-start}')