Add. Test for multiple prelude.

3 files changed, 40 insertions, 0 deletions

diff --git a/.travis.yml b/.travis.yml
index 6b389e4..779b236 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -37,6 +37,7 @@ script:
   - lua test_threads_ex.lua 10
   - lua test_threads_ex_arg.lua
   - lua test_threads_ex_opt.lua
+  - lua test_threads_ex_opt_2.lua
   - lua test_threads_attr.lua
   # - lua$LUA_SFX test_register_llthreads.lua
 
diff --git a/lakefile b/lakefile
index 960dd77..fab4646 100644
--- a/lakefile
+++ b/lakefile
@@ -42,6 +42,7 @@ target('test', install, function()
   run_test('test_threads_ex.lua 10')

   run_test('test_threads_ex_arg.lua')

   run_test('test_threads_ex_opt.lua')

+  run_test('test_threads_ex_opt_2.lua')

   run_test('test_threads_attr.lua')

 

 

diff --git a/test/test_threads_ex_opt_2.lua b/test/test_threads_ex_opt_2.lua
new file mode 100644
index 0000000..b01eda5
--- /dev/null
+++ b/test/test_threads_ex_opt_2.lua
@@ -0,0 +1,38 @@
+local thread_code = function(...)
+  local function assert_equal(name, a, b, ...)
+    if a == b then return b, ... end
+    print(name .. " Fail! Expected `" .. tostring(a) .. "` got `" .. tostring(b) .. "`")
+    os.exit(1)
+  end
+
+  local a,b,c,d = ...
+  assert_equal("1:", 1, a )
+  assert_equal("2:", 2, b )
+  assert_equal("3:", 3, c )
+  assert_equal("4:", 4, d )
+  assert_equal("#:", 4, select("#", ...))
+end
+
+local llthreads = require"llthreads.ex"
+
+local prelude1 = function(...) return 1, ... end
+
+local prelude2 = function(...) return 2, ... end
+
+local prelude = string.format([[
+  local loadstring = loadstring or load
+  local prelude1 = loadstring(%q)
+  local prelude2 = loadstring(%q)
+  return prelude1(prelude2(...))
+]], string.dump(prelude1), string.dump(prelude2))
+
+-- pass `prelude` function that change thread arguments
+local thread = llthreads.new({thread_code, prelude = prelude}, 3, 4)
+
+local a = assert(thread:start())
+
+assert(thread:join())
+
+assert(a == thread)
+
+print("done!")