more consistent locals, better test printing

3 files changed, 16 insertions, 3 deletions

diff --git a/src/lua.h b/src/lua.h
index 1ae8820..cfff148 100644
--- a/src/lua.h
+++ b/src/lua.h
@@ -114,6 +114,8 @@ int writer(lua_State*, const void*, size_t, void*);
   #define lua_gc(A, B) lua_gc(A, B, 0)

 

 #elif LUA_VERSION_NUM == 501

+    #define LUA_OK 0

+

     #define luaL_tolstring lua_tolstring

     

     #define lua_dump(A, B, C, D) lua_dump(A, B, C)

diff --git a/src/thread.c b/src/thread.c
index fbf5f01..f166c3f 100644
--- a/src/thread.c
+++ b/src/thread.c
@@ -203,11 +203,15 @@ int _thread_await(lua_State* L){
   pthread_mutex_lock(&*info->lock);

 

   if(info->return_count == 0) return 0;

+  lua_getglobal(L, "_locals");

+  int old_locals = lua_gettop(L);

+

+  lua_getglobal(info->L, "_stored_locals");

   lua_getglobal(info->L, "_res_locals");

+  luaI_jointable(info->L);

   luaI_deepcopy(info->L, L, SKIP_LOCALS | STRIP_GC);

-  env_table(L, 0);

+  //env_table(L, 0);

 

-  luaI_jointable(L);

   lua_setglobal(L, "_locals");

 

   lua_getglobal(info->L, "_return_table");

@@ -219,7 +223,7 @@ int _thread_await(lua_State* L){
     luaI_deepcopy(info->L, L, STRIP_GC);

   }

 

-  lua_pushnil(L);

+  lua_pushvalue(L, old_locals);

   lua_setglobal(L, "_locals");

   pthread_mutex_unlock(&*info->lock);

 

@@ -300,6 +304,9 @@ int l_async(lua_State* oL){
   luaI_copyvars(oL, L);

   luaL_openlibs(L);

 

+  lua_getglobal(L, "_locals");

+  lua_setglobal(L, "_stored_locals");

+

   struct thread_info* args = calloc(1, sizeof * args);

   args->L = L;

   args->lock = malloc(sizeof * args->lock);

diff --git a/tests/tests.lua b/tests/tests.lua
index 39c4d12..c95e072 100644
--- a/tests/tests.lua
+++ b/tests/tests.lua
@@ -21,8 +21,12 @@ local handle = assert(io.popen("find tests/units/".. search .." -type f"))
 
 for file in handle:lines() do
   total = total + 1
+  print(file)
   local f = loadfile(file)()
 
+  --move up one line and clear it
+  io.write("\27[1A\27[K")
+
   if f == true then
     yay(file)
   else