switched advertisement data to python

Author: patman15

tests/advertisement_data.json

@@ -0,0 +1,101 @@
+"""Test data for BLE Battery Management System integration config flow."""
+
+from typing import Final
+
+from .bluetooth import generate_advertisement_data
+
+ADVERTISEMENTS: Final[list] = [
+    # ( # conflicting integrated component: https://github.com/patman15/BMS_BLE-HA/issues/123
+    #     generate_advertisement_data(
+    #         local_name="NWJ20221223010330",#\x11",
+    #         manufacturer_data={65535: b"0UD7\xa2\xd2"},
+    #         service_uuids=["0000ffe0-0000-1000-8000-00805f9b34fb"],
+    #         rssi=-56,
+    #     ),
+    #     "ective_bms",
+    # ),
+    # (
+    #     generate_advertisement_data(
+    #         local_name="NWJ20221223010388",#\x11",
+    #         manufacturer_data={65535: b"0UD7b\xec"},
+    #         service_uuids=["0000ffe0-0000-1000-8000-00805f9b34fb"],
+    #         rssi=-47,
+    #     ),
+    #     "ective_bms",
+    # ),
+    (
+        generate_advertisement_data(
+            local_name="BatteryOben-00",
+            manufacturer_data={2917: b"\x88\xa0\xc8G\x80\x0f\xd5\xc5"},
+            service_uuids=["0000ffe0-0000-1000-8000-00805f9b34fb"],
+            tx_power=-127,
+            rssi=-83,
+        ),
+        "jikong_bms",
+    ),
+    (
+        generate_advertisement_data(
+            local_name="BatterieUnten-01",
+            manufacturer_data={2917: b"\x88\xa0\xc8G\x80\r\x08k"},
+            service_uuids=["0000ffe0-0000-1000-8000-00805f9b34fb"],
+            tx_power=-127,
+            rssi=-68,
+        ),
+        "jikong_bms",
+    ),
+    (
+        generate_advertisement_data(
+            local_name="JK_B2A8S20P",
+            manufacturer_data={2917: b"\x88\xa0\xc8G\x80\x14\x88\xb7"},
+            service_uuids=[
+                "00001800-0000-1000-8000-00805f9b34fb",
+                "00001801-0000-1000-8000-00805f9b34fb",
+                "0000180a-0000-1000-8000-00805f9b34fb",
+                "0000180f-0000-1000-8000-00805f9b34fb",
+                "0000fee7-0000-1000-8000-00805f9b34fb",
+                "0000ffe0-0000-1000-8000-00805f9b34fb",
+                "f000ffc0-0451-4000-b000-000000000000",
+            ],
+            rssi=-67,
+        ),
+        "jikong_bms",
+    ),
+    (
+        generate_advertisement_data(
+            local_name="SP05B2312190075       ",
+            service_uuids=["0000fff0-0000-1000-8000-00805f9b34fb"],
+            tx_power=-127,
+            rssi=-76,
+        ),
+        "seplos_bms",
+    ),
+    (
+        generate_advertisement_data(
+            local_name="BP02",
+            service_uuids=["0000ff00-0000-1000-8000-00805f9b34fb"],
+            rssi=-81,
+        ),
+        "seplos_v2_bms",
+    ),
+    (
+        generate_advertisement_data(
+            local_name="BP02",
+            service_uuids=[
+                "00001800-0000-1000-8000-00805f9b34fb",
+                "00001801-0000-1000-8000-00805f9b34fb",
+                "0000ff00-0000-1000-8000-00805f9b34fb",
+            ],
+            rssi=-90,
+        ),
+        "seplos_v2_bms",
+    ),
+    (
+        generate_advertisement_data(
+            local_name="LT-12V-1544",
+            manufacturer_data={33384: b"\x01\x02\x00\x07\x81\xb5N"},
+            tx_power=-127,
+            rssi=-71,
+        ),
+        "ej_bms",
+    ),
+]

tests/advertisement_data.py

@@ -1,7 +1,5 @@
 """Test the BLE Battery Management System integration config flow."""
 
-import json
-from pathlib import Path
 from typing import Final
 
 from bleak.backends.scanner import AdvertisementData
@@ -22,6 +20,7 @@
 from homeassistant.data_entry_flow import FlowResultType
 from homeassistant.helpers import entity_registry as er
 
+from .advertisement_data import ADVERTISEMENTS
 from .bluetooth import (
     generate_advertisement_data,
     generate_ble_device,
@@ -30,30 +29,9 @@
 from .conftest import mock_config, mock_update_min
 
 
-def load_advertisement_data():
-    """Load advertisement data from file."""
-    with Path("tests/advertisement_data.json").open(encoding="utf-8") as file:
-        return json.load(file)
-
-
 @pytest.fixture(
     name="advertisement",
-    params=[
-        (
-            generate_advertisement_data(
-                local_name=item["local_name"],
-                service_uuids=item.get("service_uuids", []),
-                manufacturer_data={
-                    int(k): bytes.fromhex(v)
-                    for k, v in item.get("manufacturer_data", {}).items()
-                },
-                tx_power=item.get("tx_power"),
-                rssi=item["rssi"],
-            ),
-            item["label"],
-        )
-        for item in load_advertisement_data()
-    ],
+    params=ADVERTISEMENTS,
     ids=lambda param: param[1],
 )
 def bms_advertisement(request) -> BluetoothServiceInfoBleak: