Module:Arguments/testcases
MyWikiBiz, Author Your Legacy — Thursday November 07, 2024
Jump to navigationJump to searchDocumentation for this module may be created at Module:Arguments/testcases/doc
local getArgs = require('Module:Arguments/sandbox').getArgs local ScribuntoUnit = require('Module:ScribuntoUnit') local suite = ScribuntoUnit:new() -------------------------------------------------------------------------- -- Default values -------------------------------------------------------------------------- local d = {} d.frameTitle = 'Frame title' d.parentTitle = 'Parent title' -- Precedence-testing values d.firstFrameArg = 'first frame argument' d.firstParentArg = 'first parent argument' d.secondParentArg = 'second parent argument' d.uniqueFrameArg = 'unique frame argument' d.uniqueFrameArgKey = 'uniqueFrameArgKey' d.uniqueParentArg = 'unique parent argument' d.uniqueParentArgKey = 'uniqueParentArgKey' -- Trimming and whitespace values. -- Whitespace gets trimmed from named parameters, so keys for these need -- to be numbers to make this a proper test. d.blankArg = '' d.blankArgKey = 100 d.spacesArg = '\n ' d.spacesArgKey = 101 d.untrimmedArg = '\n foo bar ' d.untrimmedArgKey = 102 d.trimmedArg = 'foo bar' d.valueFuncValue = 'valueFuncValue' d.defaultValueFunc = function() return d.valueFuncValue end d.translate = { foo = 'F00', bar = '8@r', baz = '8@z', qux = 'qUx' } -------------------------------------------------------------------------- -- Helper functions -------------------------------------------------------------------------- function suite.getFrames(frameTitle, frameArgs, parentTitle, parentArgs) frameTitle = frameTitle or d.frameTitle frameArgs = frameArgs or { d.firstFrameArg, [d.uniqueFrameArgKey] = d.uniqueFrameArg, [d.blankArgKey] = d.blankArg, [d.spacesArgKey] = d.spacesArg, [d.untrimmedArgKey] = d.untrimmedArg } parentTitle = parentTitle or d.parentTitle parentArgs = parentArgs or { d.firstParentArg, d.secondParentArg, [d.uniqueParentArgKey] = d.uniqueParentArg } local currentFrame = mw.getCurrentFrame() local parent = currentFrame:newChild{title = parentTitle, args = parentArgs} local frame = parent:newChild{title = frameTitle, args = frameArgs} return frame, parent end function suite.getDefaultArgs(options, frameTitle, frameArgs, parentTitle, parentArgs) local frame, parent = suite.getFrames(frameTitle, frameArgs, parentTitle, parentArgs) local args = getArgs(frame, options) return args end function suite:assertError(func, ...) -- Asserts that executing the function func results in an error. -- Parameters after func are func's arguments. local success, msg = pcall(func, ...) self:assertFalse(success) end function suite:assertNumberOfIterations(expected, iterator, t) local noIterations = 0 for k, v in iterator(t) do noIterations = noIterations + 1 end self:assertEquals(expected, noIterations) end -------------------------------------------------------------------------- -- Test precedence -------------------------------------------------------------------------- function suite:assertDefaultPrecedence(args) self:assertEquals(d.firstFrameArg, args[1]) self:assertEquals(d.secondParentArg, args[2]) self:assertEquals(d.uniqueFrameArg, args[d.uniqueFrameArgKey]) self:assertEquals(d.uniqueParentArg, args[d.uniqueParentArgKey]) end function suite:testDefaultPrecedence() self:assertDefaultPrecedence(suite.getDefaultArgs()) end function suite:testDefaultPrecedenceThroughWrapper() self:assertDefaultPrecedence(suite.getDefaultArgs{wrappers = {d.parentTitle}, parentOnly = false}) end function suite:testDefaultPrecedenceThroughNonWrapper() self:assertDefaultPrecedence(suite.getDefaultArgs({wrappers = d.parentTitle, frameOnly = false}, nil, nil, 'Not the parent title')) end function suite:assertParentFirst(args) self:assertEquals(d.firstParentArg, args[1]) self:assertEquals(d.secondParentArg, args[2]) self:assertEquals(d.uniqueFrameArg, args[d.uniqueFrameArgKey]) self:assertEquals(d.uniqueParentArg, args[d.uniqueParentArgKey]) end function suite:testParentFirst() self:assertParentFirst(suite.getDefaultArgs{parentFirst = true}) end function suite:testParentFirstThroughWrapper() self:assertParentFirst(suite.getDefaultArgs{wrappers = {d.parentTitle}, parentOnly = false, parentFirst = true}) end function suite:testParentFirstThroughNonWrapper() self:assertParentFirst(suite.getDefaultArgs({wrappers = d.parentTitle, frameOnly = false, parentFirst = true}, nil, nil, 'Not the parent title')) end function suite:assertParentOnly(args) self:assertEquals(d.firstParentArg, args[1]) self:assertEquals(d.secondParentArg, args[2]) self:assertEquals(nil, args[d.uniqueFrameArgKey]) self:assertEquals(d.uniqueParentArg, args[d.uniqueParentArgKey]) end function suite:testParentOnly() self:assertParentOnly(suite.getDefaultArgs{parentOnly = true}) end function suite:testParentOnlyThroughWrapper() self:assertParentOnly(suite.getDefaultArgs{wrappers = {d.parentTitle}}) end function suite:testParentOnlyThroughSandboxWrapper() self:assertParentOnly(suite.getDefaultArgs({wrappers = d.parentTitle}, nil, nil, d.parentTitle .. '/sandbox')) end function suite:assertFrameOnly(args) self:assertEquals(d.firstFrameArg, args[1]) self:assertEquals(nil, args[2]) self:assertEquals(d.uniqueFrameArg, args[d.uniqueFrameArgKey]) self:assertEquals(nil, args[d.uniqueParentArgKey]) end function suite:testFrameOnly() self:assertFrameOnly(suite.getDefaultArgs{frameOnly = true}) end function suite:testFrameOnlyThroughNonWrapper() self:assertFrameOnly(suite.getDefaultArgs({wrappers = d.parentTitle}, nil, nil, 'Not the parent title')) end function suite:testDefaultPrecedenceWithWhitespace() local frame, parent = suite.getFrames( d.frameTitle, {' '}, d.parentTitle, {d.firstParentArg} ) local args = getArgs(frame) self:assertEquals(d.firstParentArg, args[1]) end -------------------------------------------------------------------------- -- Test trimming and blank removal -------------------------------------------------------------------------- function suite:testDefaultTrimmingAndBlankRemoval() local args = suite.getDefaultArgs() self:assertEquals(nil, args[d.blankArgKey]) self:assertEquals(nil, args[d.spacesArgKey]) self:assertEquals(d.trimmedArg, args[d.untrimmedArgKey]) end function suite:testRemoveBlanksButNoTrimming() local args = suite.getDefaultArgs{trim = false} self:assertEquals(nil, args[d.blankArgKey]) self:assertEquals(nil, args[d.spacesArgKey]) self:assertEquals(d.untrimmedArg, args[d.untrimmedArgKey]) end function suite:testTrimButNoBlankRemoval() local args = suite.getDefaultArgs{removeBlanks = false} self:assertEquals(d.blankArg, args[d.blankArgKey]) self:assertEquals('', args[d.spacesArgKey]) self:assertEquals(d.trimmedArg, args[d.untrimmedArgKey]) end function suite:testNoTrimOrBlankRemoval() local args = suite.getDefaultArgs{trim = false, removeBlanks = false} self:assertEquals(d.blankArg, args[d.blankArgKey]) self:assertEquals(d.spacesArg, args[d.spacesArgKey]) self:assertEquals(d.untrimmedArg, args[d.untrimmedArgKey]) end -------------------------------------------------------------------------- -- Test valueFunc -------------------------------------------------------------------------- function suite:testValueFunc() local args = suite.getDefaultArgs{valueFunc = d.defaultValueFunc} self:assertEquals(d.valueFuncValue, args['some random key: sdfaliwyda']) end function suite:testValueFuncPrecedence() local args = suite.getDefaultArgs{ trim = false, removeBlanks = false, valueFunc = d.defaultValueFunc } self:assertEquals(d.valueFuncValue, args[1]) self:assertEquals(d.valueFuncValue, args['some random key: gekjabawyvy']) end function suite:testValueFuncKey() local args = suite.getDefaultArgs{valueFunc = function(key, value) return 'valueFunc key: '.. key end} self:assertEquals('valueFunc key: foo', args.foo) end function suite:testValueFuncValue() local args = suite.getDefaultArgs{valueFunc = function(key, value) return 'valueFunc value: '.. value end} self:assertEquals( 'valueFunc value: ' .. d.uniqueFrameArg, args[d.uniqueFrameArgKey] ) end -------------------------------------------------------------------------- -- Test adding new arguments -------------------------------------------------------------------------- function suite:testAddingNewArgs() local args = suite.getDefaultArgs() self:assertEquals(nil, args.newKey) args.newKey = 'some new key' self:assertEquals('some new key', args.newKey) end function suite:testAddingNewBlankArgs() local args = suite.getDefaultArgs() self:assertEquals(nil, args.newKey) args.newKey = '' self:assertEquals('', args.newKey) end function suite:testAddingNewSpacesArgs() local args = suite.getDefaultArgs() self:assertEquals(nil, args.newKey) args.newKey = ' ' self:assertEquals(' ', args.newKey) end function suite:testOverwriting() local args = suite.getDefaultArgs() self:assertEquals(d.firstFrameArg, args[1]) args[1] = 'a new first frame argument' self:assertEquals('a new first frame argument', args[1]) end function suite:testOverwritingWithNil() local args = suite.getDefaultArgs() self:assertEquals(d.firstFrameArg, args[1]) args[1] = nil self:assertEquals(nil, args[1]) end function suite:testOverwritingWithBlank() local args = suite.getDefaultArgs() self:assertEquals(d.firstFrameArg, args[1]) args[1] = '' self:assertEquals('', args[1]) end function suite:testOverwritingWithSpaces() local args = suite.getDefaultArgs() self:assertEquals(d.firstFrameArg, args[1]) args[1] = ' ' self:assertEquals(' ', args[1]) end function suite:testReadOnly() local args = suite.getDefaultArgs{readOnly = true} local function testFunc() args.newKey = 'some new value' end self:assertError(testFunc) end function suite:testNoOverwriteExistingKey() local args = suite.getDefaultArgs{noOverwrite = true} self:assertEquals(d.firstFrameArg, args[1]) local function testFunc() args[1] = 'a new first frame argument' end self:assertError(testFunc) end function suite:testNoOverwriteNewKey() local args = suite.getDefaultArgs{noOverwrite = true} self:assertEquals(nil, args.newKey) args.newKey = 'some new value' self:assertEquals('some new value', args.newKey) end -------------------------------------------------------------------------- -- Test bad input -------------------------------------------------------------------------- function suite:testBadFrameInput() self:assertError(getArgs, 'foo') self:assertError(getArgs, 9) self:assertError(getArgs, true) self:assertError(getArgs, function() return true end) end function suite:testBadOptionsInput() self:assertError(getArgs, {}, 'foo') self:assertError(getArgs, {}, 9) self:assertError(getArgs, {}, true) self:assertError(getArgs, {}, function() return true end) end function suite:testBadValueFuncInput() self:assertError(getArgs, {}, {valueFunc = 'foo'}) self:assertError(getArgs, {}, {valueFunc = 9}) self:assertError(getArgs, {}, {valueFunc = true}) self:assertError(getArgs, {}, {valueFunc = {}}) end -------------------------------------------------------------------------- -- Test iterator metamethods -------------------------------------------------------------------------- function suite:testPairs() local args = getArgs{'foo', 'bar', baz = 'qux'} self:assertNumberOfIterations(3, pairs, args) end function suite:testIpairs() local args = getArgs{'foo', 'bar', baz = 'qux'} self:assertNumberOfIterations(2, ipairs, args) end function suite:testNoNilsinPairs() -- Test that when we use pairs, we don't iterate over any nil values -- that have been memoized. local args = getArgs{''} local temp = args[1] -- Memoizes the nil self:assertNumberOfIterations(0, pairs, args) end function suite:testNoNilsinIpairs() -- Test that when we use ipairs, we don't iterate over any nil values -- that have been memoized. local args = getArgs{''} local temp = args[1] -- Memoizes the nil self:assertNumberOfIterations(0, ipairs, args) end function suite:testDeletedArgsInPairs() -- Test that when we use pairs, we don't iterate over any values that have -- been explicitly set to nil. local args = getArgs{'foo'} args[1] = nil self:assertNumberOfIterations(0, pairs, args) end function suite:testDeletedArgsInIpairs() -- Test that when we use ipairs, we don't iterate over any values that have -- been explicitly set to nil. local args = getArgs{'foo'} args[1] = nil self:assertNumberOfIterations(0, ipairs, args) end function suite:testNoNilsInPairsAfterIndex() -- Test that when we use pairs, we don't iterate over any nils that -- might have been memoized after a value that is not present in the -- original args table is indexed. local args = getArgs{} local temp = args.someRandomValue -- Memoizes the nil self:assertNumberOfIterations(0, pairs, args) end function suite:testNoNilsInPairsAfterNewindex() -- Test that when we use pairs, we don't iterate over any nils that -- might have been memoized after a value that is not present in the -- original args table is added to the args table. local args = getArgs{} args.newKey = nil -- The nil is memoized self:assertNumberOfIterations(0, pairs, args) end function suite:testNoTableLengthChangeWhileIterating() -- Test that the number of arguments doesn't change if we index the -- args table while iterating. -- (Note that the equivalent test is not needed for new arg table -- indexes, as that would be a user error - doing so produces -- undetermined behaviour in Lua's next() function.) local args = getArgs{'foo', 'bar', baz = 'qux'} self:assertNumberOfIterations(3, pairs, args) for k, v in pairs(args) do local temp = args[k .. 'foo'] end self:assertNumberOfIterations(3, pairs, args) end function suite:testPairsPrecedenceWithWhitespace() local frame, parent = suite.getFrames( d.frameTitle, {' '}, d.parentTitle, {d.firstParentArg} ) local args = getArgs(frame) local actual for k, v in pairs(args) do actual = v end self:assertEquals(d.firstParentArg, actual) -- Check that we have actually iterated. self:assertNumberOfIterations(1, pairs, args) end function suite:testPairsPrecedenceWithNil() local frame, parent = suite.getFrames( d.frameTitle, {}, d.parentTitle, {d.firstParentArg} ) local args = getArgs(frame) local actual for k, v in pairs(args) do actual = v end self:assertEquals(d.firstParentArg, actual) -- Check that we have actually iterated. self:assertNumberOfIterations(1, pairs, args) end function suite:testIpairsEarlyExit() local mt = {} function mt.__index(t, k) if k == 1 then return 'foo' elseif k == 2 then return 'bar' elseif k == 3 then error('Expanded argument 3 unnecessarily') end end function mt.__pairs(t) error('Called pairs unnecessarily') end function mt.__ipairs(t) -- Works just like the default ipairs, except respecting __index return function(t, i) local v = t[i + 1] if v ~= nil then return i + 1, v end end, t, 0 end local args = getArgs(setmetatable({}, mt)) for k,v in ipairs(args) do if v == 'bar' then break end end end -------------------------------------------------------------------------- -- Test argument translation -------------------------------------------------------------------------- function suite:testTranslationIndex() local args = getArgs({F00 = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'yep'}, {translate = d.translate}) self:assertEquals('one', args.foo) self:assertEquals('two', args.bar) self:assertEquals('three', args.baz) self:assertEquals('four', args.qux) self:assertEquals('yep', args.untranslated) end function suite:testTranslationPairsWithAutoBacktranslate() local args = getArgs({F00 = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'yep'}, {translate = d.translate}) local cleanArgs = {} for k,v in pairs(args) do cleanArgs[k] = v end self:assertDeepEquals( { foo = 'one', bar = 'two', baz = 'three', qux = 'four', untranslated = 'yep' }, cleanArgs ) end function suite:testTranslationPairsWithBacktranslate() local args = getArgs({F00 = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'yep'}, {translate = d.translate, backtranslate = {F00 = 'foo'}}) local cleanArgs = {} for k,v in pairs(args) do cleanArgs[k] = v end self:assertDeepEquals( { foo = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', untranslated = 'yep' }, cleanArgs ) end function suite:testTranslationPairsWithoutBacktranslate() local args = getArgs({F00 = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'yep'}, {translate = d.translate, backtranslate = false}) local cleanArgs = {} for k,v in pairs(args) do cleanArgs[k] = v end self:assertDeepEquals( { F00 = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'yep' }, cleanArgs ) end function suite:testTranslationNewindex() local args = getArgs({F00 = 'one', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'yep'}, {translate = d.translate, backtranslate = false}) args.foo = 'changed1' args.untranslated = 'changed2' local cleanArgs = {} for k,v in pairs(args) do cleanArgs[k] = v end self:assertDeepEquals( { F00 = 'changed1', ['8@r'] = 'two', ['8@z'] = 'three', qUx = 'four', foo = 'nope', untranslated = 'changed2' }, cleanArgs ) end function suite:test_argument() local currentFrame = mw.getCurrentFrame() currentFrame.args[5] = 555; local args = getArgs(currentFrame) self:assertEquals('nil', type(args.foo)) end return suite