add jobgraph stub

2 files changed, 50 insertions, 0 deletions

diff --git a/tests/modules/async/jobgraph.lua b/tests/modules/async/jobgraph.lua
new file mode 100644
index 000000000..dd8cb05f8
--- /dev/null
+++ b/tests/modules/async/jobgraph.lua
@@ -0,0 +1,7 @@
+import("core.base.scheduler")
+import("async.jobgraph")
+
+function main()
+
+end
+
diff --git a/tests/modules/async/runjobs.lua b/tests/modules/async/runjobs.lua
new file mode 100644
index 000000000..4e2a98cc0
--- /dev/null
+++ b/tests/modules/async/runjobs.lua
@@ -0,0 +1,43 @@
+import("core.base.scheduler")
+import("private.async.jobpool")
+import("async.runjobs")
+
+function _jobfunc(index, total, opt)
+    print("%s: run job (%d/%d)", scheduler.co_running(), index, total)
+    local dt = os.mclock()
+    os.sleep(1000)
+    dt = os.mclock() - dt
+    print("%s: run job (%d/%d) end, progress: %s, dt: %d ms", scheduler.co_running(), index, total, opt.progress, dt)
+end
+
+function main()
+
+    -- test callback
+    print("==================================== test callback ====================================")
+    local t = os.mclock()
+    runjobs("test", _jobfunc, {total = 100, comax = 6, timeout = 1000, timer = function (running_jobs_indices)
+        print("%s: timeout (%d ms), running: %s", scheduler.co_running(), os.mclock() - t, table.concat(running_jobs_indices, ","))
+    end})
+
+    -- test jobs
+    print("==================================== test jobs ====================================")
+    local jobs = jobpool.new()
+    local root = jobs:addjob("job/root", function (index, total, opt)
+        _jobfunc(index, total, opt)
+    end)
+    for i = 1, 3 do
+        local job = jobs:addjob("job/" .. i, function (index, total, opt)
+            _jobfunc(index, total, opt)
+        end, {rootjob = root})
+        for j = 1, 50 do
+            jobs:addjob("job/" .. i .. "/" .. j, function (index, total, opt)
+                _jobfunc(index, total, opt)
+            end, {rootjob = job})
+        end
+    end
+    t = os.mclock()
+    runjobs("test", jobs, {comax = 6, timeout = 1000, timer = function (running_jobs_indices)
+        print("%s: timeout (%d ms), running: %s", scheduler.co_running(), os.mclock() - t, table.concat(running_jobs_indices, ","))
+    end})
+end
+