add scheduler.co_yield

2 files changed, 41 insertions, 0 deletions

diff --git a/tests/modules/scheduler/sleep.lua b/tests/modules/scheduler/sleep.lua
new file mode 100644
index 000000000..5c6976a07
--- /dev/null
+++ b/tests/modules/scheduler/sleep.lua
@@ -0,0 +1,26 @@
+import("core.base.scheduler")
+
+function _session2(id)
+    print("session2: %d ..", id)
+    local dt = os.mclock()
+    os.sleep(1000)
+    dt = os.mclock() - dt
+    print("session2: %d end, dt: %d ms", id, dt)
+end
+
+function _session1(id)
+    print("session1: %d ..", id)
+    local dt = os.mclock()
+    scheduler.co_sleep(1000)
+    dt = os.mclock() - dt
+    print("session1: %d end, dt: %d ms", id, dt)
+end
+
+function main()
+    for i = 1, 10 do
+        scheduler.co_start(_session1, i)
+        scheduler.co_start(_session2, i)
+    end
+    scheduler.runloop()
+end
+
diff --git a/tests/modules/scheduler/yield.lua b/tests/modules/scheduler/yield.lua
new file mode 100644
index 000000000..f884a0199
--- /dev/null
+++ b/tests/modules/scheduler/yield.lua
@@ -0,0 +1,15 @@
+import("core.base.scheduler")
+
+function _session(id)
+    print("test: %d ..", id)
+    scheduler.co_yield()
+    print("test: %d end", id)
+end
+
+function main()
+    for i = 1, 10 do
+        scheduler.co_start(_session, i)
+    end
+    scheduler.runloop()
+end
+