added more tests

Author: craigbarratt

custom_components/pyscript/function.py

@@ -69,11 +69,6 @@ def init(cls, hass):
             }
         )
 
-    @classmethod
-    async def entity_ids(cls, domain=None):
-        """Implement entity_ids."""
-        return cls.hass.states.async_entity_ids(domain)
-
     @classmethod
     async def async_sleep(cls, duration):
         """Implement task.sleep()."""
@@ -196,8 +191,6 @@ def get(cls, name):
         async def service_call(*args, **kwargs):
             await cls.hass.services.async_call(domain, service, kwargs)
 
-        # service_call = functools.partial(cls.service_call, domain, service)
-
         return service_call
 
     @classmethod

tests/test_init.py

@@ -5,6 +5,10 @@
 import pathlib
 
 from custom_components.pyscript.const import DOMAIN
+from custom_components.pyscript.event import Event
+from custom_components.pyscript.function import Function
+from custom_components.pyscript.global_ctx import GlobalContextMgr
+from custom_components.pyscript.state import State
 import custom_components.pyscript.trigger as trigger
 from pytest_homeassistant.async_mock import mock_open, patch
 
@@ -285,7 +289,11 @@ def func2(**kwargs):
 
 @service
 def call_service(domain=None, name=None, **kwargs):
-    service.call(domain, name, **kwargs)
+    if domain == "pyscript" and name == "func1":
+        task.sleep(0)
+        pyscript.func1(**kwargs)
+    else:
+        service.call(domain, name, **kwargs)
 
 """,
     )
@@ -447,3 +455,22 @@ def func5(var_name=None, value=None):
             "custom_components.pyscript.trigger.dt_now", return_value=now
         ):
             await hass.services.async_call("pyscript", "reload", {}, blocking=True)
+
+
+async def test_misc_errors(hass, caplog):
+    """Test miscellaneous errors."""
+
+    await setup_script(hass, None, dt(2020, 7, 1, 11, 59, 59, 999999), "")
+
+    Function()
+    GlobalContextMgr()
+    State()
+    Event()
+    Event.notify_del("not_in_notify_list", None)
+    trigger.TrigTime()
+
+    assert "Function class is not meant to be instantiated" in caplog.text
+    assert "GlobalContextMgr class is not meant to be instantiated" in caplog.text
+    assert "State class is not meant to be instantiated" in caplog.text
+    assert "Event class is not meant to be instantiated" in caplog.text
+    assert "TrigTime class is not meant to be instantiated" in caplog.text

tests/test_jupyter.py

@@ -506,4 +506,12 @@ async def test_jupyter_kernel_no_connection_timeout(hass, caplog):
     """Test Jupyter kernel timeout on no connection."""
     sock, port_nums = await setup_script(hass, [dt(2020, 7, 1, 11, 0, 0, 0)], "", no_connect=True)
 
-    assert "No connections to session " in caplog.text
+    #
+    # There is a race condition waiting for the log message, so we need to poll
+    #
+    for _ in range(50):
+        if "No connections to session jupyter_" in caplog.text:
+            break
+        await asyncio.sleep(2e-3)
+
+    assert "No connections to session jupyter_" in caplog.text

tests/test_unique.py

@@ -131,6 +131,14 @@ def func5():
     res = task.wait_until(state_trigger="pyscript.f5var2 == '1'")
     pyscript.done = [seq_num, "pyscript.f5var2"]
 
+
+@time_trigger("startup")
+def func6():
+    task.unique("func6")
+    task.unique("func6", kill_me=True)
+    # mess up the sequence numbers if task.unique fails to kill us
+    pyscript.done = [999]
+
 """,
     )
 

tests/test_unit_eval.py

@@ -757,6 +757,7 @@ async def test_eval(hass):
     ["continue", "Exception in test line 1 column 0: continue statement outside loop"],
     ["raise", "Exception in test line 1 column 0: No active exception to reraise"],
     ["yield", "Exception in test line 1 column 0: test: not implemented ast ast_yield"],
+    ["task.executor(5)", "Exception in test line 1 column 14: function is not callable by task.executor()"],
     [
         "import cmath; exec('xyz = cmath.sqrt(complex(3, 4))', {})",
         "Exception in test line 1 column 54: Exception in exec() line 1 column 28: 'sqrt' is not callable (got None)",