Editing Module:Italic title/testcases

local mItalicTitle = require('Module:Italic title/sandbox')
local ScribuntoUnit = require('Module:ScribuntoUnit')
local Spy = require('Module:Lua-mock/Spy')
local suite = ScribuntoUnit:new()

local function patchCurrentTitle(page, func)
	local oldGetCurrentTitle = mw.title.getCurrentTitle
	mw.title.getCurrentTitle = function ()
		return mw.title.new(page)
	end
	func()
	mw.title.getCurrentTitle = oldGetCurrentTitle
end

local function patchDisplayTitle(func)
	local oldGetCurrentFrame = mw.getCurrentFrame
	local frame = oldGetCurrentFrame()
	local oldCallParserFunction = frame.callParserFunction
	local callParserFunctionSpy = Spy(oldCallParserFunction)
	mw.getCurrentFrame = function ()
		return frame
	end
	frame.callParserFunction = callParserFunctionSpy
	func(callParserFunctionSpy, frame)
	frame.callParserFunction = oldCallParserFunction
	mw.getCurrentFrame = oldGetCurrentFrame
end

local displayTitleTestData = {
	{
		func = mItalicTitle._main,
		description = 'test _main: Italicize all - single-word titles',
		page = "Example",
		args = {},
		expectedDisplayTitleArgs = {"<i>Example</i>"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Italicize all - multi-word titles',
		page = "Star Wars",
		args = {},
		expectedDisplayTitleArgs = {"<i>Star Wars</i>"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Text in parentheses is not italicized',
		page = "Example (dab)",
		args = {},
		expectedDisplayTitleArgs = {"<i>Example</i> (dab)"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Text in final parentheses is not italicized',
		page = "Example (not dab) (dab)",
		args = {},
		expectedDisplayTitleArgs = {"<i>Example (not dab)</i> (dab)"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Text in disambiguation with parentheses is not italicized',
		page = "Example (dab (dab))",
		args = {},
		expectedDisplayTitleArgs = {"<i>Example</i> (dab (dab))"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: When the entire title is in parentheses, it is all italicized',
		page = "(Example page)",
		args = {},
		expectedDisplayTitleArgs = {"<i>(Example page)</i>"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Only specified string argument italicized (string=)',
		page = "List of Ally McBeal episodes",
		args = {string = "Ally McBeal"},
		expectedDisplayTitleArgs = {"List of <i>Ally McBeal</i> episodes"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Only specified string argument italicized - in disambiguation (string= and all=)',
		page = "Peter (Fringe episode)",
		args = {string = "Fringe", all = "yes"},
		expectedDisplayTitleArgs = {"Peter (<i>Fringe</i> episode)"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Title and dismabiguation italicized (all=)',
		page = "Title (dismabiguation)",
		args = {all = "yes"},
		expectedDisplayTitleArgs = {"<i>Title (dismabiguation)</i>"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: For pages with a namespace prefix, the namespace prefix is not italicized',
		page = "Wikipedia:Example",
		args = {},
		expectedDisplayTitleArgs = {"Wikipedia:<i>Example</i>"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Only specified string argument italicized (string=) in title with a comma',
		page = "Morceaux de salon, Op. 6 (Rachmaninoff)",
		args = {string = "Morceaux de salon"},
		expectedDisplayTitleArgs = {"<i>Morceaux de salon</i>, Op. 6 (Rachmaninoff)"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: Only specified string argument italicized (string= and all=) in title with a comma',
		page = "Morceaux de salon, Op. 6 (Rachmaninoff)",
		args = {string = "Morceaux de salon", all = "yes"},
		expectedDisplayTitleArgs = {"<i>Morceaux de salon</i>, Op. 6 (Rachmaninoff)"},
	},
	{
		func = mItalicTitle._dabonly,
		description = 'test _dabonly: Italicize disambiguation',
		page = "The End (Lost)",
		args = {},
		expectedDisplayTitleArgs = {"The End (<i>Lost</i>)"},
	},
	{
		func = mItalicTitle._dabonly,
		description = 'test _dabonly: Italicize disambiguation with parentheses',
		page = "The Ghost Talks (Randall and Hopkirk (Deceased))",
		args = {},
		expectedDisplayTitleArgs = {"The Ghost Talks (<i>Randall and Hopkirk (Deceased)</i>)"},
	},
	{
		func = mItalicTitle._dabonly,
		description = 'test _dabonly: Italicize final parentheses only',
		page = "Title (not disambiguation) (disambiguation)",
		args = {},
		expectedDisplayTitleArgs = {"Title (not disambiguation) (<i>disambiguation</i>)"},
	},
	{
		func = mItalicTitle._dabonly,
		description = 'test _dabonly: When the entire title is in parentheses, it is not italicized',
		page = "(Example page)",
		args = {},
		expectedDisplayTitleArgs = {"(Example page)"},
	},
	{
		func = mItalicTitle._dabonly,
		description = 'test _dabonly: Only specified string argument italicized - in disambiguation (string=)',
		page = "Hell on Wheels (Hell on Wheels episode)",
		args = {string = "Hell on Wheels"},
		expectedDisplayTitleArgs = {"Hell on Wheels (<i>Hell on Wheels</i> episode)"},
	},
	{
		func = mItalicTitle._main,
		description = 'test _main: When noerror is specified as the first positional argument, it is passed to DISPLAYTITLE',
		page = "Example",
		args = {"noerror"},
		expectedDisplayTitleArgs = {"<i>Example</i>", "noerror"},
	},
	{
		func = mItalicTitle._dabonly,
		description = 'test _dabonly: When noerror is specified as the first positional argument, it is passed to DISPLAYTITLE',
		page = "Example (dab)",
		args = {"noerror"},
		expectedDisplayTitleArgs = {"Example (<i>dab</i>)", "noerror"},
	},
}

for _, testData in ipairs(displayTitleTestData) do
	suite[testData.description] = function (self)
		patchCurrentTitle(
			testData.page,
			function ()
				patchDisplayTitle(
					function (callParserFunctionSpy, frame)
						testData.func(testData.args)
						callParserFunctionSpy:assertCallMatches{
							atIndex = 1,
							arguments = {
								frame,
								'DISPLAYTITLE',
								unpack(testData.expectedDisplayTitleArgs),
							}
						}
					end
				)
			end
		)
	end
end


local categoryTestData = {
	{
		description = "test _main: Tracking category is added when the string argument does not match the title",
		func = mItalicTitle._main,
		page = "Example",
		args = {string = "foo"},
	},
	{
		description = "test _main: Tracking category is added when the string argument does not match the title or dab, and the all argument is specified",
		func = mItalicTitle._main,
		page = "Example (dab)",
		args = {string = "foo", all = "yes"},
	},
	{
		description = "test _dabonly: Tracking category is added when the string argument does not match the disambiguation",
		func = mItalicTitle._dabonly,
		page = "Example (dab)",
		args = {string = "foo"},
	},
	{
		description = "test _dabonly: Tracking category is added when the string argument does not match the disambiguation and no disambiguation is present in title",
		func = mItalicTitle._dabonly,
		page = "Example",
		args = {string = "foo"},
	},
}

for _, testData in ipairs(categoryTestData) do
	suite[testData.description] = function (self)
		patchCurrentTitle(
			testData.page,
			function ()
				self:assertStringContains(
					testData.func(testData.args),
					"[[Category:Pages using italic title with no matching string]]",
					true
				)
			end
		)
	end
end

return suite