luau/tests/conformance/ifelseexpr.lua
Arseny Kapoulkine d01addc625 Sync to upstream/release/501 (#20)
Co-authored-by: Rodactor <rodactor@roblox.com>
2021-11-01 14:52:34 -07:00

80 lines
2.8 KiB
Lua

-- This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
-- This file is based on Lua 5.x tests -- https://github.com/lua/lua/tree/master/testes
print("testing if-else expressions")
function True()
return true
end
function False()
return false
end
function EvalElseifChain(condition1, condition2, condition3)
return if condition1 then 10 elseif condition2 then 20 elseif condition3 then 30 else 40
end
function EvalElse_IfChain(condition1, condition2, condition3)
return if condition1 then 10 else if condition2 then 20 else if condition3 then 30 else 40
end
function CheckForConditionalEvaluation(condition)
local counter = 0
local function AddToCounter(count)
counter += count
return counter
end
local result = if condition then AddToCounter(7) else AddToCounter(17)
if condition then
assert(result == 7)
else
assert(result == 17)
end
-- ensure the counter value matches the result of the clause that was evaluated
assert(counter == result)
end
-- Test expression using only constants
assert(if true then true else false)
assert(if false then false else true)
assert(if nil then false else true)
assert((7 + if true then 10 else 20) == 17)
-- Test expression using non-constant condition
assert(if True() then true else false)
assert(if False() then false else true)
-- Test evaluation of a "chain" of if/elseif/else in an expression
assert(EvalElseifChain(false, false, false) == 40)
assert(EvalElseifChain(false, false, true) == 30)
assert(EvalElseifChain(false, true, false) == 20)
assert(EvalElseifChain(false, true, true) == 20)
assert(EvalElseifChain(true, false, false) == 10)
assert(EvalElseifChain(true, false, true) == 10)
assert(EvalElseifChain(true, true, false) == 10)
assert(EvalElseifChain(true, true, true) == 10)
-- Test evaluation of a "chain" of if/"else if"/else in an expression
assert(EvalElse_IfChain(false, false, false) == 40)
assert(EvalElse_IfChain(false, false, true) == 30)
assert(EvalElse_IfChain(false, true, false) == 20)
assert(EvalElse_IfChain(false, true, true) == 20)
assert(EvalElse_IfChain(true, false, false) == 10)
assert(EvalElse_IfChain(true, false, true) == 10)
assert(EvalElse_IfChain(true, true, false) == 10)
assert(EvalElse_IfChain(true, true, true) == 10)
-- Test nesting of if-else expressions inside the condition of an if-else expression
assert((if (if True() then false else true) then 10 else 20) == 20)
assert((if if True() then false else true then 10 else 20) == 20)
-- Ensure that if/else expressions are conditionally evaluated
-- i.e. verify the evaluated expression doesn't evaluate the true and false expressions and
-- merely select the proper value.
CheckForConditionalEvaluation(true)
CheckForConditionalEvaluation(false)
return('OK')