using System;

using System.Linq;

using Microsoft.VisualStudio.TestTools.UnitTesting;

using Microsoft.PowerShell;

namespace UnitTestPSReadLine

{

// Disgusting language hack to make it easier to read a sequence of keys.

using _ = Keys;

public partial class UnitTest

{

[TestMethod]

public void TestInput()

{

TestSetup(KeyMode.Cmd);

Test("exit", Keys(

"exit",

_.Enter,

CheckThat(() => AssertCursorLeftIs(0))));

}

[TestMethod]

public void TestRevertLine()

{

// Add one test for chords

TestSetup(KeyMode.Cmd, new KeyHandler("Ctrl+x,Escape", PSConsoleReadLine.RevertLine));

Test("ls", Keys("di", _.Escape, "ls"));

Test("ls", Keys("di", _.CtrlX, _.Escape, "ls"));

TestSetup(KeyMode.Emacs);

Test("ls", Keys("di", _.Escape, _.R, "ls"));

Test("ls", Keys("di", _.AltR, "ls"));

}

[TestMethod]

public void TestCancelLine()

{

TestSetup(KeyMode.Cmd, new KeyHandler("Ctrl+C", PSConsoleReadLine.CancelLine));

Test("", Keys("oops", _.CtrlC,

CheckThat(() => AssertScreenIs(1,

TokenClassification.Command, "oops",

Tuple.Create(ConsoleColor.Red, _console.BackgroundColor), "^C")),

InputAcceptedNow));

Test("", Keys("exit", _.CtrlC,

CheckThat(() => AssertScreenIs(1,

TokenClassification.Keyword, "exit",

Tuple.Create(ConsoleColor.Red, _console.BackgroundColor), "^C")),

InputAcceptedNow));

// Test near/at/over buffer width input

var width = _console.BufferWidth;

var line1 = new string('a', width - 2);

Test("", Keys(line1, _.CtrlC,

CheckThat(() => AssertScreenIs(1,

TokenClassification.Command, line1,

Tuple.Create(ConsoleColor.Red, _console.BackgroundColor), "^C")),

InputAcceptedNow));

var line2 = new string('a', width - 1);

Test("", Keys(line2, _.CtrlC,

CheckThat(() => AssertScreenIs(2,

TokenClassification.Command, line2,

Tuple.Create(ConsoleColor.Red, _console.BackgroundColor), "^C")),

InputAcceptedNow));

var line3 = new string('a', width);

Test("", Keys(line3, _.CtrlC,

CheckThat(() => AssertScreenIs(2,

TokenClassification.Command, line3,

Tuple.Create(ConsoleColor.Red, _console.BackgroundColor), "^C")),

InputAcceptedNow));

}

[TestMethod]

public void TestForwardDeleteLine()

{

TestSetup(KeyMode.Cmd);

// Empty input (does nothing but don't crash)

Test("", Keys("", _.CtrlEnd));

// at end of input - doesn't change anything

Test("abc", Keys("abc", _.CtrlEnd));

// More normal usage - actually delete stuff

Test("a", Keys("abc", _.LeftArrow, _.LeftArrow, _.CtrlEnd));

}

[TestMethod]

public void TestBackwardDeleteLine()

{

TestSetup(KeyMode.Cmd);

// Empty input (does nothing but don't crash)

Test("", Keys(_.CtrlHome));

// at beginning of input - doesn't change anything

Test("abc", Keys("abc", _.Home, _.CtrlHome));

// More typical usage

Test("c", Keys("abc", _.LeftArrow, _.CtrlHome));

Test("", Keys("abc", _.CtrlHome));

}

[TestMethod]

public void TestBackwardDeleteChar()

{

TestSetup(KeyMode.Cmd);

// Empty input (does nothing but don't crash)

Test("", Keys("", _.Backspace));

// At end, delete all input

Test("", Keys("a", _.Backspace));

// At end, delete all input with extra backspaces

Test("", Keys("a", _.Backspace, _.Backspace));

// Delete first character

Test("b", Keys("ab", _.LeftArrow, _.Backspace));

// Delete first character with extra backspaces

Test("b", Keys("ab", _.LeftArrow, _.Backspace, _.Backspace));

// Delete middle character

Test("ac", Keys("abc", _.LeftArrow, _.Backspace));

}

[TestMethod]

public void TestDeleteChar()

{

TestSetup(KeyMode.Cmd);

// Empty input (does nothing, but don't crash)

Test("", Keys(_.Delete));

// At end but input not empty (does nothing, but don't crash)

Test("a", Keys('a', _.Delete));

// Delete last character

Test("a", Keys("ab", _.LeftArrow, _.Delete));

// Delete first character

Test("b", Keys("ab", _.Home, _.Delete));

// Delete middle character

Test("ac", Keys("abc", _.Home, _.RightArrow, _.Delete));

}

[TestMethod]

public void TestDeleteCharOrExit()

{

TestSetup(KeyMode.Emacs);

Test("exit", Keys(_.CtrlD, InputAcceptedNow));

Test("foo", Keys("foo", _.CtrlD));

Test("oo", Keys("foo", _.Home, _.CtrlD));

Test("exit", Keys("foo", _.Home, Enumerable.Repeat(_.CtrlD, 4), InputAcceptedNow));

}

[TestMethod]

public void TestSwapCharacters()

{

TestSetup(KeyMode.Emacs);

Test("abc", Keys(

"abc", CheckThat(() => AssertLineIs("abc")),

_.CtrlT, CheckThat(() => AssertLineIs("acb")),

_.CtrlUnderbar

));

Test("abcd", Keys(

"abcd", CheckThat(() => AssertLineIs("abcd")),

_.CtrlA, _.CtrlT, CheckThat(() => AssertLineIs("abcd")),

_.CtrlF, Enumerable.Repeat(_.CtrlT, 3), CheckThat(() => AssertLineIs("bcda")),

_.CtrlT, CheckThat(() => AssertLineIs("bcad")),

Enumerable.Repeat(_.CtrlUnderbar, 4)

));

}

[TestMethod]

public void TestAcceptAndGetNext()

{

TestSetup(KeyMode.Emacs);

// No history

Test("", Keys(_.CtrlO, InputAcceptedNow));

// One item in history

SetHistory("echo 1");

Test("", Keys(_.CtrlO, InputAcceptedNow));

// Two items in history, make sure after Ctrl+O, second history item

// is recalled.

SetHistory("echo 1", "echo 2");

Test("echo 1", Keys(_.UpArrow, _.UpArrow, _.CtrlO, InputAcceptedNow));

Test("echo 2", Keys(_.Enter));

// Test that the current saved line is saved after AcceptAndGetNext

SetHistory("echo 1", "echo 2");

Test("echo 1", Keys("e", _.UpArrow, _.UpArrow, _.CtrlO, InputAcceptedNow));

Test("e", Keys(_.DownArrow, _.DownArrow, _.Enter));

// Test that we can edit after recalling the current line

SetHistory("echo 1", "echo 2");

Test("echo 1", Keys("e", _.UpArrow, _.UpArrow, _.CtrlO, InputAcceptedNow));

Test("eee", Keys(_.DownArrow, _.DownArrow, "ee", _.Enter));

}

[TestMethod]

public void TestAcceptAndGetNextWithHistorySearch()

{

TestSetup(KeyMode.Emacs,

new KeyHandler("UpArrow", PSConsoleReadLine.HistorySearchBackward),

new KeyHandler("DownArrow", PSConsoleReadLine.HistorySearchForward));

// Test that after AcceptAndGetNext, the previous search is not applied

SetHistory("echo 1", "echo 2", "zzz");

Test("echo 1", Keys("e", _.UpArrow, _.UpArrow, _.CtrlO, InputAcceptedNow));

Test("zzz", Keys(_.DownArrow, _.Enter));

}

[TestMethod]

public void TestAddLine()

{

TestSetup(KeyMode.Cmd);

Test("1\n2", Keys('1', _.ShiftEnter, '2'));

}

[TestMethod]

public void TestInsertLineAbove()

{

TestSetup(KeyMode.Cmd);

var continutationPromptLength = PSConsoleReadlineOptions.DefaultContinuationPrompt.Length;

// Test case - start with single line, cursor at end

Test("56\n1234", Keys("1234", _.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(0, 0)), "56"));

// Test case - start with single line, cursor in home position

Test("56\n1234", Keys("1234", _.Home, _.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(0, 0)), "56"));

// Test case - start with single line, cursor in middle

Test("56\n1234", Keys("1234",

_.LeftArrow, _.LeftArrow, _.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(0, 0)),

"56"));

// Test case - start with multi-line, cursor at end of second line (end of input)

Test("1234\n9ABC\n5678", Keys("1234", _.ShiftEnter, "5678",

_.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"9ABC"));

// Test case - start with multi-line, cursor at beginning of second line

Test("1234\n9ABC\n5678", Keys("1234", _.ShiftEnter, "5678",

_.LeftArrow, _.Home, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

_.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"9ABC"));

// Test case - start with multi-line, cursor at end of first line

Test("9ABC\n1234\n5678", Keys("1234", _.ShiftEnter, "5678",

_.UpArrow, _.LeftArrow, _.End, CheckThat(() => AssertCursorLeftTopIs(4, 0)),

_.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(0, 0)),

"9ABC"));

// Test case - start with multi-line, cursor at beginning of first line - temporarily having to press Home twice to

// work around bug in home handler.

Test("9ABC\n1234\n5678", Keys("1234", _.ShiftEnter, "5678",

_.UpArrow, _.LeftArrow, _.Home, _.Home, CheckThat(() => AssertCursorLeftTopIs(0, 0)),

_.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(0, 0)),

"9ABC"));

// Test case - insert multiple blank lines

Test("1234\n9ABC\n\n5678", Keys("1234", _.ShiftEnter, "5678",

_.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

_.CtrlEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"9ABC"));

}

[TestMethod]

public void TestInsertLineBelow()

{

TestSetup(KeyMode.Cmd);

var continutationPromptLength = PSConsoleReadlineOptions.DefaultContinuationPrompt.Length;

// Test case - start with single line, cursor at end

Test("1234\n56", Keys("1234",

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"56"));

// Test case - start with single line, cursor in home position

Test("1234\n56", Keys("1234",

_.Home, _.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"56"));

// Test case - start with single line, cursor in middle

Test("1234\n56", Keys("1234",

_.LeftArrow, _.LeftArrow, _.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"56"));

// Test case - start with multi-line, cursor at end of second line (end of input)

Test("1234\n5678\n9ABC", Keys("1234", _.ShiftEnter, "5678",

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 2)),

"9ABC"));

// Test case - start with multi-line, cursor at beginning of second line

Test("1234\n5678\n9ABC", Keys("1234", _.ShiftEnter, "5678",

_.LeftArrow, _.Home, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 2)),

"9ABC"));

// Test case - start with multi-line, cursor at end of first line

Test("1234\n9ABC\n5678", Keys("1234", _.ShiftEnter, "5678",

_.UpArrow, _.LeftArrow, _.End, CheckThat(() => AssertCursorLeftTopIs(4, 0)),

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"9ABC"));

// Test case - start with multi-line, cursor at beginning of first line - temporarily having to press Home twice to

// work around bug in home handler.

Test("1234\n9ABC\n5678", Keys("1234", _.ShiftEnter, "5678",

_.UpArrow, _.LeftArrow, _.Home, _.Home, CheckThat(() => AssertCursorLeftTopIs(0, 0)),

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 1)),

"9ABC"));

// Test case - insert multiple blank lines

Test("1234\n5678\n\n9ABC", Keys("1234", _.ShiftEnter, "5678",

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 2)),

_.CtrlShiftEnter, CheckThat(() => AssertCursorLeftTopIs(continutationPromptLength, 3)),

"9ABC"));

}

[TestMethod]

public void TestMultilineHomeBugFixed()

{

TestSetup(KeyMode.Cmd);

// Going from second line to first line, press left arrow and then home.

// That puts cursor in column 1 instead of 0. Bug? This could have something

// to do with BeginningOfLine testing against i > 1 in multiline edit instead

// of i > 0.

Test("1234\n9ABC", Keys("1234", _.ShiftEnter, "9ABC", _.UpArrow, _.LeftArrow, _.Home, CheckThat(() => AssertCursorLeftTopIs(0, 0))));

}

[TestMethod]

public void TestIgnore()

{

TestSetup(KeyMode.Emacs);

var volumeUp = new ConsoleKeyInfo('\0', ConsoleKey.VolumeUp, false, false, false);

var volumeDown = new ConsoleKeyInfo('\0', ConsoleKey.VolumeDown, false, false, false);

var volumeMute = new ConsoleKeyInfo('\0', ConsoleKey.VolumeMute, false, false, false);

Test("ab", Keys("a", volumeDown, volumeMute, volumeUp, "b"));

}

}

}