fixes @task_unique function arg - see #761

Author: craigbarratt

custom_components/pyscript/eval.py

@@ -382,7 +382,7 @@ async def trigger_init(self, trig_ctx, func_name):
             "state_active": {"arg_cnt": {1}},
             "state_trigger": {"arg_cnt": {"*"}, "type": {list, set}, "rep_ok": True},
             "service": {"arg_cnt": {0, "*"}},
-            "task_unique": {"arg_cnt": {1, 2}},
+            "task_unique": {"arg_cnt": {1}},
             "time_active": {"arg_cnt": {"*"}},
             "time_trigger": {"arg_cnt": {0, "*"}, "rep_ok": True},
             "webhook_trigger": {"arg_cnt": {1, 2}, "rep_ok": True},

tests/test_unique.py

@@ -78,6 +78,7 @@ async def test_task_unique(hass, caplog):
 seq_num = 0
 
 @time_trigger("startup")
+@task_unique("func6")
 def funcStartupSync():
     global seq_num, funcStartupSync_id
 
@@ -88,9 +89,11 @@ def funcStartupSync():
     #
     # stick around so the task.unique() still applies
     #
-    task.unique("func6")
     assert task.current_task() == task.name2id("func6")
     assert task.current_task() == task.name2id()["func6"]
+    task.unique("func7")
+    assert task.current_task() == task.name2id("func7")
+    assert task.current_task() == task.name2id()["func7"]
     task.sleep(10000)
 
 @service