aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--conf.d/controller/lua.d/aft.lua48
1 files changed, 48 insertions, 0 deletions
diff --git a/conf.d/controller/lua.d/aft.lua b/conf.d/controller/lua.d/aft.lua
index 9fb06d8..27b558e 100644
--- a/conf.d/controller/lua.d/aft.lua
+++ b/conf.d/controller/lua.d/aft.lua
@@ -35,6 +35,10 @@ _AFT = {
tests_list = {},
event_history = false,
monitored_events = {},
+ beforeEach = nil,
+ afterEach = nil,
+ beforeAll = nil,
+ afterAll = nil,
}
function _AFT.enableEventHistory()
@@ -167,15 +171,19 @@ end
function _AFT.testEvtNotReceived(testName, eventName, timeout)
table.insert(_AFT.tests_list, {testName, function()
+ if _AFT.beforeEach then _AFT.beforeEach() end
if timeout then sleep(timeout) end
_AFT.assertEvtNotReceived(eventName)
+ if _AFT.afterEach then _AFT.afterEach() end
end})
end
function _AFT.testEvtReceived(testName, eventName, timeout)
table.insert(_AFT.tests_list, {testName, function()
+ if _AFT.beforeEach then _AFT.beforeEach() end
if timeout then sleep(timeout) end
_AFT.assertEvtReceived(eventName)
+ if _AFT.afterEach then _AFT.afterEach() end
end})
end
@@ -230,22 +238,60 @@ end
function _AFT.testVerb(testName, api, verb, args, cb)
table.insert(_AFT.tests_list, {testName, function()
+ if _AFT.beforeEach then _AFT.beforeEach() end
_AFT.assertVerb(api, verb, args, cb)
+ if _AFT.afterEach then _AFT.afterEach() end
end})
end
function _AFT.testVerbError(testName, api, verb, args, cb)
table.insert(_AFT.tests_list, {testName, function()
+ if _AFT.beforeEach then _AFT.beforeEach() end
_AFT.assertVerbError(api, verb, args, cb)
+ if _AFT.afterEach then _AFT.afterEach() end
end})
end
function _AFT.describe(testName, testFunction)
table.insert(_AFT.tests_list, {testName, function()
+ if _AFT.beforeEach then _AFT.beforeEach() end
testFunction()
+ if _AFT.afterEach then _AFT.afterEach() end
end})
end
+function _AFT.setbeforeEach(beforeEachTestFunction)
+ if type(beforeEachTestFunction) == "function" then
+ _AFT.beforeEach = beforeEachTestFunction
+ else
+ print("Wrong beforeEach function defined. It isn't detected as a function type")
+ end
+end
+
+function _AFT.setAfterEach(afterEachTestFunction)
+ if type(afterEachTestFunction) == "function" then
+ _AFT.afterEach = afterEachTestFunction
+ else
+ print("Wrong beforeEach function defined. It isn't detected as a function type")
+ end
+end
+
+function _AFT.setBefore(beforeAllTestsFunctions)
+ if type(beforeAllTestsFunctions) == "function" then
+ _AFT.beforeAll = beforeAllTestsFunctions
+ else
+ print("Wrong beforeAll function defined. It isn't detected as a function type")
+ end
+end
+
+function _AFT.setAfter(afterAllTestsFunctions)
+ if type(afterAllTestsFunctions) == "function" then
+ _AFT.afterAll = afterAllTestsFunctions
+ else
+ print("Wrong afterAll function defined. It isn't detected as a function type")
+ end
+end
+
--[[
Make all assertions accessible using _AFT and declare some convenients
aliases.
@@ -365,7 +411,9 @@ function _launch_test(context, args)
end
AFB:success(_AFT.context, { info = "Launching tests"})
+ if _AFT.beforeAll then _AFT.beforeAll() end
lu.LuaUnit:runSuiteByInstances(_AFT.tests_list)
+ if _AFT.afterAll then _AFT.afterAll() end
local success ="Success : "..tostring(lu.LuaUnit.result.passedCount)
local failures="Failures : "..tostring(lu.LuaUnit.result.testCount-lu.LuaUnit.result.passedCount)