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

	suitetestData.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

	suitetestData.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