Module:WikitextParser/testcases: Difference between revisions

Revision as of 12:11, 18 March 2025

Latest revision as of 14:28, 25 March 2025

(12 intermediate revisions by the same user not shown)

Line 1:

local Wparser = require( 'Module:WikitextParser' )

local testpage = mw.title.new( 'Module:WikitextParser/testpage' ):getContent()

local ScribuntoUnit = require( 'Module:ScribuntoUnit' )

local Suite = ScribuntoUnit:new()

function Suite:testReplaceElements()

self:assertEquals( '@@@0@@@', W.replaceElements( 'a', { 'a' } ) )

self:assertEquals( '@@@1@@@b@@@2@@@', W.replaceElements( 'abc', { 'a', 'c' } ) )

end

function Suite:testRestoreElements()

self:assertEquals( 'a', W.restoreElements( '@@@0@@@' ) )

--self:assertEquals( 'abc', W.restoreElements( '@@@1@@@b@@@2@@@' ) )

end

function Suite:testLead()

self:assertEquals( '', Wparser.getLead( '' ) )

self:assertEquals( '', Wparser.getLead( '==b==' ) )

self:assertEquals( 'a', Wparser.getLead( 'a' ) )

self:assertEquals( 'a', Wparser.getLead( 'a', 'a\n==b==' ) )

self:assertEquals( 'This test page interacts with [[Module:WikitextParser/testcases]].', Wparser.getLead( testpage ) )

end

function Suite:testSection()

self:assertEquals( nil, Wparser.getSection( 'a', 'a' ) )

self:assertEquals( '', Wparser.getSection( '==a==', 'a' ) )

self:assertEquals( 'b', Wparser.getSection( '==a==\nb', 'a' ) )

self:assertEquals( 'b', Wparser.getSection( '==a==\nb\n==c==\nd', 'a' ) )

self:assertEquals( 'b', Wparser.getSection( '==a==\nb\n==a==\nd', 'a' ) )

self:assertEquals( 'b\n===c===\nd', Wparser.getSection( '==a==\nb\n===c===\nd', 'a' ) )

self:assertEquals( 'd', Wparser.getSection( '==a==\nb\n===c===\nd', 'c' ) )

self:assertEquals( 'a\n===Subsection===\nb', Wparser.getSection( testpage, 'Section' ) )

self:assertEquals( 'b', Wparser.getSection( testpage, 'Subsection' ) )

end

function Suite:testSectionTag()

self:assertEquals( nil, Wparser.getSectionTag( 'a', 'a' ) )

self:assertEquals( nil, Wparser.getSectionTag( 'a<section begin="b" /><section end="b" />c', 'a' ) )

self:assertEquals( '', Wparser.getSectionTag( 'a<section begin="b" /><section end="b" />c', 'b' ) )

self:assertEquals( 'c', Wparser.getSectionTag( 'a<section begin="b" />c<section end="b" />d', 'b' ) )

self:assertEquals( '', Wparser.getSectionTag( testpage, 'a' ) )

self:assertEquals( 'b', Wparser.getSectionTag( testpage, 'b' ) )

self:assertEquals( '{|\n!c\n|}', Wparser.getSectionTag( testpage, 'c' ) )

self:assertEquals( nil, Wparser.getSectionTag( testpage, 'd' ) )

end

function Suite:testSections()

self:assertDeepEquals( {}, Wparser.getSections( '' ) )

self:assertDeepEquals( {}, Wparser.getSections( 'a' ) )

self:assertDeepEquals( { ['a']='' }, Wparser.getSections( '==a==' ) )

self:assertDeepEquals( { ['b']='' }, Wparser.getSections( 'a\n==b==' ) )

self:assertDeepEquals( { ['a']='', ['b']='' }, Wparser.getSections( '==a==\n==b==' ) )

self:assertDeepEquals( { ['a']='b' }, Wparser.getSections( '==a==\nb' ) )

self:assertDeepEquals( { ['b']='c' }, Wparser.getSections( 'a\n==b==\nc' ) )

self:assertDeepEquals( { ['b']='c', ['d']='e' }, Wparser.getSections( 'a\n==b==\nc\n===d===\ne' ) )

end

function Suite:testLists()

self:assertDeepEquals( {}, Wparser.getLists( '' ) )

self:assertDeepEquals( {}, Wparser.getLists( 'a' ) )

self:assertDeepEquals( { '*a' }, Wparser.getLists( '*a' ) )

self:assertDeepEquals( { '*a\n*b' }, Wparser.getLists( '*a\n*b' ) )

self:assertDeepEquals( { '*b\n*c' }, Wparser.getLists( 'a\n*b\n*c\nd' ) )

self:assertDeepEquals( { '*b\n*c', '#e\n#f' }, Wparser.getLists( 'a\n*b\n*c\nd\n#e\n#f\ng' ) )

self:assertDeepEquals( {

'*b\n*c',

Line 70 ⟶ 60:

'*c\n*d',

'#h\n#i',

}, Wparser.getLists( testpage ) )

end

function Suite:testParagraphs()

self:assertDeepEquals( {}, Wparser.getParagraphs( '' ) )

self:assertDeepEquals( {}, Wparser.getParagraphs( '{{a}}' ) )

self:assertDeepEquals( {}, Wparser.getParagraphs( '{{a\n|b=c}}' ) )

self:assertDeepEquals( {}, Wparser.getParagraphs( '{{a}}\n{{b}}\n{{c}}' ) )

self:assertDeepEquals( {}, Wparser.getParagraphs( '{{a}}\n\n{{b}}\n\n{{c}}' ) )

self:assertDeepEquals( { 'a' }, Wparser.getParagraphs( 'a' ) )

self:assertDeepEquals( { 'a{{b}}' }, Wparser.getParagraphs( 'a{{b}}' ) )

self:assertDeepEquals( { '{{a}}b' }, Wparser.getParagraphs( '{{a}}b' ) )

self:assertDeepEquals( { 'a' }, Wparser.getParagraphs( 'a\n\n{{b}}' ) )

self:assertDeepEquals( { 'b' }, Wparser.getParagraphs( '{{a}}\n\nb' ) )

self:assertDeepEquals( { '{{a}}b{{c}}' }, Wparser.getParagraphs( '{{a}}b{{c}}' ) )

self:assertDeepEquals( { 'a', 'b', 'c' }, Wparser.getParagraphs( 'a\n\nb\n\nc' ) )

self:assertDeepEquals( { 'a', 'c' }, Wparser.getParagraphs( 'a\n\n{{b}}\n\nc' ) )

self:assertDeepEquals( {

'X1',

'{{X2}} X3 {{X4}}',

}, Wparser.getParagraphs( Wparser.getSection( testpage, 'Paragraphs' ) ) )

end

function Suite:testTemplates()

self:assertDeepEquals( {}, Wparser.getTemplates( 'a' ) )

self:assertDeepEquals( { '{{a}}' }, Wparser.getTemplates( '{{a}}' ) )

self:assertDeepEquals( { '{{b}}' }, Wparser.getTemplates( 'a{{b}}c' ) )

self:assertDeepEquals( { '{{a|b=c}}' }, Wparser.getTemplates( '{{a|b=c}}' ) )

self:assertDeepEquals( { '{{a\n|b=c\n}}' }, Wparser.getTemplates( '{{a\n|b=c\n}}' ) )

self:assertDeepEquals( { '{{b}}', '{{d}}' }, Wparser.getTemplates( 'a{{b}}c{{d}}e' ) )

self:assertDeepEquals( { '{{b}}', '{{d}}' }, Wparser.getTemplates( 'a{{b}}c{{d}}e' ) )

self:assertDeepEquals( { '{{a|b={{c}}}}' }, Wparser.getTemplates( '{{a|b={{c}}}}' ) )

self:assertDeepEquals( {

'{{X2}}',

Line 114 ⟶ 104:

'{{reflist|group=i}}',

'{{Infobox\n|a=1\n|b=2\n|d=4\n|c=3\n}}'

}, Wparser.getTemplates( testpage ) )

end

function Suite:testTemplate()

self:assertEquals( '{{a}}', Wparser.getTemplate( '{{a}}', 'a' ) )

self:assertEquals( '{{a|b=c}}', Wparser.getTemplate( '{{a|b=c}}', 'a' ) )

self:assertEquals( '{{a|b=c}}', Wparser.getTemplate( '{{a|b=c}}', 'A' ) )

self:assertEquals( '{{Infobox\n|a=1\n|b=2\n|d=4\n|c=3\n}}', Wparser.getTemplate( testpage, 'Infobox' ) )

end

function Suite:testParameterstestTemplateParameters()

self:assertDeepEquals( { [1]='b' }, Wparser.getParametersgetTemplateParameters( '{{a|b}}' ) )

--self:assertDeepEquals( { [1]='b' }, Wparser.getParametersgetTemplateParameters( '{{a|1=b}}' ) )

self:assertDeepEquals( { [1]='b',['c']='1' }, Wparser.getParametersgetTemplateParameters( '{{a|b|c=1}}' ) )

self:assertDeepEquals( { [1]='b',['c']='d' }, Wparser.getParametersgetTemplateParameters( '{{a|b|c=d}}' ) )

self:assertDeepEquals( { ['b']='c' }, Wparser.getParametersgetTemplateParameters( '{{a|b=c}}' ) )

self:assertDeepEquals( { ['b']='c', ['d']='e' }, Wparser.getParametersgetTemplateParameters( '{{a|b=c|d=e}}' ) )

end

function Suite:testTags()

self:assertDeepEquals( {}, Wparser.getTags( 'a' ) )

self:assertDeepEquals( { '<div>a' }, Wparser.getTags( '<div!--a-->a' ) )

self:assertDeepEquals( { '<div></div>a' }, Wparser.getTags( '<div></div>a' ) )

self:assertDeepEquals( { '<div></div>' }, Wparser.getTags( 'a<div></div>b' ) )

self:assertDeepEquals( { '<div>a</div>' }, Wparser.getTags( 'a<div>a</div>b' ) )

self:assertDeepEquals( { '<div>ba</div>' }, Wparser.getTags( 'a<div>ba</div>c' ) )

self:assertDeepEquals( { '< div >ab</ div >' }, Wparser.getTags( 'a< div >ab</ div >c' ) )

self:assertDeepEquals( { '< div class="a">ba</ div >' }, Wparser.getTags( '< div class="a">ba</ div >' ) )

self:assertDeepEquals( { '<div><div></div></div>', '<divclass="a">b</div>' }, Wparser.getTags( '<div class="a"><div></div>b</div>' ) )

self:assertDeepEquals( { '<div>b<div>c</div>d</div>', '<div>c</div>' }, Wparser.getTags( 'a<div>b<div>c</div>d</div>e' ) )

self:assertDeepEquals( { '<div>b<div><span>a</span>c</div>d</div>', '<div><span>a</span>c</div>', '<span>a</span>' }, Wparser.getTags( 'a<div>b<div><span>a</span>c</div>d</div>e' ) )

self:assertDeepEquals( { '<div><div><span>a</span></div></div>', '<div><span>a</span></div>', '<span>a</span>' }, parser.getTags( '<div><div><span>a</span></div></div>' ) )

self:assertDeepEquals( {

'<gallery>\nFile:A.png|g<ref>g</ref>\nFile:B.png\n</gallery>',

Line 168 ⟶ 159:

'<ref name="l">m</ref>',

'<noinclude>b</noinclude>',

}, Wparser.getTags( testpage ) )

end

function Suite:testReplaceElementstestTagName()

self:assertEquals( nil, parser.getTagName( 'abc' ) )

self:assertEquals( 'aref', Wparser.restoreElementsgetTagName( '@@@0@@@<ref>' ) )

self:assertEquals( '@@@0@@@ref', Wparser.replaceElementsgetTagName( 'a',< {ref />'a' } ) )

self:assertEquals( 'ref', parser.getTagName( '< ref name="a" />' ) )

--self:assertEquals( 'abcref', Wparser.restoreElementsgetTagName( '@@@1@@@b@@@2@@@<ref>a</ref>' ) )

self:assertEquals( 'ref', parser.getTagName( '<ref name = a >b</ref>' ) )

end

function Suite:testRestoreElementstestTagAttribute()

self:assertEquals( nil, parser.getTagAttribute( '<ref>', 'name' ) )

self:assertEquals( '@@@1@@@b@@@2@@@a', Wparser.replaceElementsgetTagAttribute( 'abc',<ref { 'name=a>', 'cname' } ) )

self:assertEquals( 'b', parser.getTagAttribute( '<ref name="b">', 'name' ) )

self:assertEquals( 'c', parser.getTagAttribute( '< ref name = "c" />', 'name' ) )

end

function Suite:testGalleries()

self:assertDeepEquals( {}, Wparser.getGalleries( 'a' ) )

self:assertDeepEquals( { '<gallery>a</gallery>' }, Wparser.getGalleries( '<gallery>a</gallery>' ) )

self:assertDeepEquals( { '<gallery>b</gallery>' }, Wparser.getGalleries( 'a<gallery>b</gallery>c' ) )

self:assertDeepEquals( { '<gallery>\nFile:A.png\nFile:B.png\n</gallery>' }, Wparser.getGalleries( '<gallery>\nFile:A.png\nFile:B.png\n</gallery>' ) )

self:assertDeepEquals( { '<gallery>\nFile:A.png\nFile:B.png\n</gallery>' }, Wparser.getGalleries( 'a<gallery>\nFile:A.png\nFile:B.png\n</gallery>b' ) )

self:assertDeepEquals( {

'<gallery>\nFile:A.png|g<ref>g</ref>\nFile:B.png\n</gallery>',

}, Wparser.getGalleries( testpage ) )

end

function Suite:testReferences()

self:assertDeepEquals( {}, Wparser.getReferences( 'a' ) )

self:assertDeepEquals( { '<ref>a</ref>' }, Wparser.getReferences( '<ref>a</ref>' ) )

self:assertDeepEquals( { '<ref>b</ref>' }, Wparser.getReferences( 'a<ref>b</ref>c' ) )

self:assertDeepEquals( { '<ref>b</ref>', '<ref>d</ref>' }, Wparser.getReferences( 'a<ref>b</ref>c<ref>d</ref>' ) )

self:assertDeepEquals( { '<ref name="b">c</ref>', '<ref name="b" />' }, Wparser.getReferences( 'a<ref name="b">c</ref>d<ref name="b" />' ) )

self:assertDeepEquals( {

'<ref>g</ref>',

Line 196 ⟶ 203:

'<ref name="l" />',

'<ref name="l">m</ref>',

}, Wparser.getReferences( testpage ) )

end

function Suite:testTables()

self:assertDeepEquals( {}, Wparser.getTables( 'a' ) )

self:assertDeepEquals( { '{|\n!b\n|}' }, Wparser.getTables( 'a\n{|\n!b\n|}\nc' ) )

self:assertDeepEquals( { '{|\n!b\n|}', '{|\n|d\n|}', '{|id="e"\n|e\n|}' }, Wparser.getTables( 'a\n{|\n!b\n|}\nc\n{|\n|d\n|}\n{|id="e"\n|e\n|}f' ) )

self:assertDeepEquals( {

'{|\n!b\n|}',

Line 213 ⟶ 220:

'{|\n|e\n|}',

'{|\n|k\n|}',

}, Wparser.getTables( testpage ) )

end

function Suite:testTableAttribute()

self:assertEquals( nil, parser.getTableAttribute( '{|\n|a\n|}', 'id' ) )

self:assertEquals( 'a', parser.getTableAttribute( '{| id=a\n|a\n|}', 'id' ) )

self:assertEquals( 'b', parser.getTableAttribute( '{| id="b" class="foo"\n|a\n|}', 'id' ) )

self:assertEquals( 'c', parser.getTableAttribute( '{| class="foo" id="c" \n|a\n|}', 'id' ) )

end

function Suite:testLinks()

self:assertDeepEquals( {}, Wparser.getLinks( 'a' ) )

self:assertDeepEquals( { '[[a]]' }, Wparser.getLinks( '[[a]]' ) )

self:assertDeepEquals( { '[[a|b]]' }, Wparser.getLinks( '[[a|b]]' ) )

self:assertDeepEquals( { '[[b]]' }, Wparser.getLinks( 'a[[b]]c' ) )

self:assertDeepEquals( { '[[b]]', '[[d]]' }, Wparser.getLinks( 'a[[b]]c[[d]]e' ) )

self:assertDeepEquals( {

'[[Module:WikitextParser/testcases]]',

Line 241 ⟶ 255:

'[[Category:X2]]',

'[[Category:X3]]',

}, Wparser.getLinks( testpage ) )

end

function Suite:testCategories()

self:assertDeepEquals( {}, Wparser.getCategories( 'a' ) )

self:assertDeepEquals( {}, Wparser.getCategories( '[[a]]' ) )

self:assertDeepEquals( { '[[Category:a]]' }, Wparser.getCategories( '[[Category:a]]' ) )

self:assertDeepEquals( { '[[ category : a ]]' }, Wparser.getCategories( '[[ category : a ]]' ) )

self:assertDeepEquals( { '[[Category:b]]' }, Wparser.getCategories( 'a[[Category:b]]c' ) )

self:assertDeepEquals( { '[[Category:b]]', '[[Category:c]]' }, Wparser.getCategories( 'a[[Category:b]][[Category:c]]d' ) )

self:assertDeepEquals( {

'[[Category:X1]]',

'[[Category:X2]]',

'[[Category:X3]]',

}, Wparser.getCategories( testpage ) )

end

function Suite:testFiles()

self:assertDeepEquals( {}, Wparser.getFiles( 'a' ) )

self:assertDeepEquals( {}, Wparser.getFiles( '[[a]]' ) )

self:assertDeepEquals( { '[[File:a]]' }, Wparser.getFiles( '[[File:a]]' ) )

self:assertDeepEquals( { '[[ file : a ]]' }, Wparser.getFiles( '[[ file : a ]]' ) )

self:assertDeepEquals( { '[[File:b]]' }, Wparser.getFiles( 'a[[File:b]]c' ) )

self:assertDeepEquals( { '[[File:b]]', '[[File:c]]' }, Wparser.getFiles( 'a[[File:b]][[File:c]]d' ) )

self:assertDeepEquals( {

'[[File:Name.jpg|thumb|X1]]',

Line 273 ⟶ 287:

'[[File:A.png]]',

'[[File:f.png]]',

}, Wparser.getFiles( testpage ) )

end

function Suite:testFileName()

self:assertEquals( 'Example.jpg', parser.getFileName( '[[File:Example.jpg]]' ) )

self:assertEquals( 'Example.jpg', parser.getFileName( '[[File:Example.jpg|thumb]]' ) )

self:assertEquals( 'Example.jpg', parser.getFileName( '[[ File : Example.jpg | thumb ]]' ) )

end

function Suite:testExternalLinks()

self:assertDeepEquals( {}, Wparser.getExternalLinks( '' ) )

self:assertDeepEquals( {}, Wparser.getExternalLinks( 'a' ) )

self:assertDeepEquals( {}, Wparser.getExternalLinks( '[[a]]' ) )

self:assertDeepEquals( { '[//www.example.com]' }, Wparser.getExternalLinks( '[//www.example.com]' ) )

self:assertDeepEquals( { '[http://www.example.com]' }, Wparser.getExternalLinks( '[http://www.example.com]' ) )

self:assertDeepEquals( { '[https://www.example.com]' }, Wparser.getExternalLinks( '[https://www.example.com]' ) )

self:assertDeepEquals( { '[//www.example.com example]' }, Wparser.getExternalLinks( '[//www.example.com example]' ) )

self:assertDeepEquals( { '[http://www.example.com example]' }, Wparser.getExternalLinks( '[http://www.example.com example]' ) )

self:assertDeepEquals( { '[https://www.example.com example]' }, Wparser.getExternalLinks( '[https://www.example.com example]' ) )

end