aim-web2.1.0rc9

scripts/addons/webgui/tests/test_worker_v2.py

@@ -0,0 +1,27 @@
+from dataclasses import replace
+from pathlib import Path
+import json
+import time
+from aim_webgui.auth.service import Auth
+from aim_webgui.workflows import Workflows
+from aim_webgui.worker import execute_claimed
+from test_core_execution import native_fixture
+
+
+def test_worker(settings,core_root):
+    native_fixture(core_root)
+    auth=Auth(settings);auth.bootstrap()
+    with auth.store.transaction()as db:
+        db.execute("UPDATE users SET must_change_password=0 WHERE username='admin'")
+        uid=db.execute("SELECT id FROM users WHERE username='admin'").fetchone()[0]
+    flow=Workflows(settings)
+    review=flow.review(uid,'example','debug_test_connection',['test01.example'],{})
+    ident=flow.queue_review(uid,review['review_id'],idempotency_key='worker-proof')
+    with flow.store.transaction()as db:
+        db.execute("UPDATE jobs SET status='running' WHERE id=?",(ident,))
+    execute_claimed(settings,ident)
+    result=flow.job(uid,ident)['core_result']
+    assert result['status']=='succeeded' and result['counts']['ok']==1
+    assert result['remote_work_may_have_started'] is True
+    assert not list(settings.state_dir.glob('.console-*'))
+    assert not flow.plans(uid)