Merge pull request #658 from eskerda/feat/helper-extend

lunarmodules · Aug 25, 2022 · 279dbc8 · 279dbc8
2 parents 4982187 + 6a65a9a
commit 279dbc8
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/busted/modules/helper_loader.lua b/busted/modules/helper_loader.lua
@@ -6,14 +6,21 @@ return function()
   local loadHelper = function(busted, helper, options)
     local old_arg = _G.arg
     local success, err = pcall(function()
+      local fn
+
       utils.copy_interpreter_args(options.arguments)
       _G.arg = options.arguments
+
       if helper:match('%.lua$') then
-        dofile(path.normpath(helper))
+        fn = dofile(path.normpath(helper))
       elseif hasMoon and helper:match('%.moon$') then
-        moonscript.dofile(path.normpath(helper))
+        fn = moonscript.dofile(path.normpath(helper))
       else
-        require(helper)
+        fn = require(helper)
+      end
+
+      if type(fn) == 'function' then
+        assert(fn(busted, helper, options))
       end
     end)