Add script to convert zwave_js device diagnostics to fixture (#102799)

tests/components/zwave_js/scripts/__init__.py

@@ -0,0 +1 @@
+"""Tests for zwave_js scripts."""

tests/components/zwave_js/scripts/test_convert_device_diagnostics_to_fixture.py

@@ -0,0 +1,80 @@
+"""Test convert_device_diagnostics_to_fixture script."""
+import copy
+import json
+from pathlib import Path
+import sys
+from unittest.mock import patch
+
+import pytest
+
+from homeassistant.components.zwave_js.scripts.convert_device_diagnostics_to_fixture import (
+    extract_fixture_data,
+    get_fixtures_dir_path,
+    load_file,
+    main,
+)
+
+from tests.common import load_fixture
+
+
+def _minify(text: str) -> str:
+    """Minify string by removing whitespace and new lines."""
+    return text.replace(" ", "").replace("\n", "")
+
+
+def test_fixture_functions() -> None:
+    """Test functions related to the fixture."""
+    diagnostics_data = json.loads(load_fixture("zwave_js/device_diagnostics.json"))
+    state = extract_fixture_data(copy.deepcopy(diagnostics_data))
+    assert isinstance(state["values"], list)
+    assert (
+        get_fixtures_dir_path(state)
+        == Path(__file__).parents[1] / "fixtures" / "zooz_zse44_state.json"
+    )
+
+    old_diagnostics_format_data = copy.deepcopy(diagnostics_data)
+    old_diagnostics_format_data["data"]["state"]["values"] = list(
+        old_diagnostics_format_data["data"]["state"]["values"].values()
+    )
+    assert (
+        extract_fixture_data(old_diagnostics_format_data)
+        == old_diagnostics_format_data["data"]["state"]
+    )
+
+    with pytest.raises(ValueError):
+        extract_fixture_data({})
+
+
+def test_load_file() -> None:
+    """Test load file."""
+    assert load_file(
+        Path(__file__).parents[1] / "fixtures" / "device_diagnostics.json"
+    ) == json.loads(load_fixture("zwave_js/device_diagnostics.json"))
+
+
+def test_main(capfd: pytest.CaptureFixture[str]) -> None:
+    """Test main function."""
+    Path(__file__).parents[1] / "fixtures" / "zooz_zse44_state.json"
+    fixture_str = load_fixture("zwave_js/zooz_zse44_state.json")
+    fixture_dict = json.loads(fixture_str)
+
+    # Test dump to stdout
+    args = [
+        "homeassistant/components/zwave_js/scripts/convert_device_diagnostics_to_fixture.py",
+        str(Path(__file__).parents[1] / "fixtures" / "device_diagnostics.json"),
+    ]
+    with patch.object(sys, "argv", args):
+        main()
+
+    captured = capfd.readouterr()
+    assert _minify(captured.out) == _minify(fixture_str)
+
+    # Check file dump
+    args.append("--file")
+    with patch.object(sys, "argv", args), patch(
+        "homeassistant.components.zwave_js.scripts.convert_device_diagnostics_to_fixture.Path.write_text"
+    ) as write_text_mock:
+        main()
+
+    assert len(write_text_mock.call_args_list) == 1
+    assert write_text_mock.call_args[0][0] == json.dumps(fixture_dict, indent=2)