add tests for changes to state module

Author: raman325

tests/test_state.py

@@ -0,0 +1,43 @@
+"""Test pyscripts test module."""
+from custom_components.pyscript.state import State
+from pytest_homeassistant_custom_component.async_mock import patch
+
+from homeassistant.core import Context
+from homeassistant.helpers.state import State as HassState
+
+
+async def test_service_call(hass):
+    """Test calling a service using the entity_id as a property."""
+    with patch(
+        "custom_components.pyscript.state.async_get_all_descriptions",
+        return_value={
+            "test": {
+                "test": {"description": None, "fields": {"entity_id": "blah", "other_service_data": "blah"}}
+            }
+        },
+    ), patch.object(hass.states, "get", return_value=HassState("test.entity", "True")), patch.object(
+        hass.services, "async_call"
+    ) as call:
+        State.init(hass)
+        await State.get_service_params()
+
+        func = await State.get("test.entity.test")
+        await func(context=Context(id="test"), blocking=True, limit=1, other_service_data="test")
+        assert call.called
+        assert call.call_args[0] == (
+            "test",
+            "test",
+            {"other_service_data": "test", "entity_id": "test.entity"},
+        )
+        assert call.call_args[1] == {"context": Context(id="test"), "blocking": True, "limit": 1}
+        call.reset_mock()
+
+        func = await State.get("test.entity.test")
+        await func(context=Context(id="test"), blocking=False, other_service_data="test")
+        assert call.called
+        assert call.call_args[0] == (
+            "test",
+            "test",
+            {"other_service_data": "test", "entity_id": "test.entity"},
+        )
+        assert call.call_args[1] == {"context": Context(id="test"), "blocking": False}