public void HostUserInterfacePromptForChoice()
{
Mock <NuGetRawUserInterface> mockRawUI;
Mock <NuGetHostUserInterface> mockUI;
ConsoleDispatcher dispatcher;
InteractiveHelper.InitializeConsole(out mockRawUI, out mockUI, out dispatcher);
var descriptions = new Collection <ChoiceDescription> {
new ChoiceDescription("&Yes"), // 0 (default)
new ChoiceDescription("&No") // 1
};
// test choice
InteractiveHelper.PostKeys(dispatcher, "n", appendCarriageReturn: true,
timeout: TimeSpan.FromSeconds(5));
int chosen = mockUI.Object.PromptForChoice("Test", "Test", descriptions, 0);
Assert.Equal(1, chosen);
// test default choice
dispatcher.PostKey(VsKeyInfo.Enter);
chosen = mockUI.Object.PromptForChoice("Test", "Test", descriptions, 0);
Assert.Equal(0, chosen);
}
public void HostUserInterfacePromptForMissingMandatoryParameters()
{
Mock <NuGetRawUserInterface> mockRawUI;
Mock <NuGetHostUserInterface> mockUI;
ConsoleDispatcher dispatcher;
var host = InteractiveHelper.InitializeConsole(out mockRawUI, out mockUI, out dispatcher);
Runspace rs = RunspaceFactory.CreateRunspace(host);
rs.Open();
using (PowerShell ps = PowerShell.Create())
{
ps.Runspace = rs;
// [string]$Name
InteractiveHelper.PostKeys(dispatcher, "foo", appendCarriageReturn: true);
// [int]$Count
InteractiveHelper.PostKeys(dispatcher, "42", appendCarriageReturn: true);
// [int[]]$Numbers
InteractiveHelper.PostKeys(dispatcher, "1", appendCarriageReturn: true);
InteractiveHelper.PostKeys(dispatcher, "2", appendCarriageReturn: true);
InteractiveHelper.PostKeys(dispatcher, "3", appendCarriageReturn: true);
dispatcher.PostKey(VsKeyInfo.Enter); // empty line
Hashtable result =
ps.AddScript(@"
function test-missing {
param(
[parameter(mandatory=$true)]
[string]$Name,
[parameter(mandatory=$true)]
[int]$Count,
[parameter(mandatory=$true)]
[int[]]$Numbers
);
@{
Name = $name;
Count = $count;
Numbers = $Numbers
}
}
test-missing")
.Invoke <Hashtable>()
.FirstOrDefault();
// no errors
Assert.True(ps.Streams.Error.Count == 0);
Assert.NotNull(result);
Assert.Equal(result["Name"], "foo");
Assert.Equal(result["Count"], 42);
Assert.True(
((int[])result["Numbers"])
.SequenceEqual(new[] { 1, 2, 3 }));
}
}
public void HostUserInterfaceReadLine()
{
Mock <NuGetRawUserInterface> mockRawUI;
Mock <NuGetHostUserInterface> mockUI;
ConsoleDispatcher dispatcher;
InteractiveHelper.InitializeConsole(out mockRawUI, out mockUI, out dispatcher);
InteractiveHelper.PostKeys(dispatcher, "nuget", appendCarriageReturn: true, timeout: TimeSpan.FromSeconds(5));
string line = mockUI.Object.ReadLine();
Assert.Equal("nuget", line);
}
public void HostUserInterfacePromptForConfirm()
{
Mock <NuGetRawUserInterface> mockRawUI;
Mock <NuGetHostUserInterface> mockUI;
ConsoleDispatcher dispatcher;
var host = InteractiveHelper.InitializeConsole(out mockRawUI, out mockUI, out dispatcher);
Runspace rs = RunspaceFactory.CreateRunspace(host);
rs.Open();
using (PowerShell ps = PowerShell.Create())
{
ps.Runspace = rs;
// put a "y" on the input queue
InteractiveHelper.PostKeys(dispatcher, "y", appendCarriageReturn: true,
timeout: TimeSpan.FromSeconds(5));
bool result = ps.AddScript(@"
function test-confirm {
[cmdletbinding(supportsshouldprocess=$true)]param();
$pscmdlet.shouldprocess('do', 'this')
}
[System.Threading.Thread]::CurrentThread.CurrentUICulture = [Globalization.CultureInfo]'en-US'; test-confirm -confirm")
.Invoke <bool>()
.FirstOrDefault();
// no errors
Assert.True(ps.Streams.Error.Count == 0);
// shouldprocess accepted a "y"
Assert.True(result);
// put a "n" on the input queue
InteractiveHelper.PostKeys(dispatcher, "n", appendCarriageReturn: true,
timeout: TimeSpan.FromSeconds(5));
// execute confirm again
ps.Streams.ClearStreams();
result = ps.Invoke <bool>().FirstOrDefault();
// no errors
Assert.True(ps.Streams.Error.Count == 0);
// shouldprocess accepted a "n"
Assert.False(result);
}
}
public void HostUserInterfaceReadLineAsSecureString()
{
Mock <NuGetRawUserInterface> mockRawUI;
Mock <NuGetHostUserInterface> mockUI;
ConsoleDispatcher dispatcher;
InteractiveHelper.InitializeConsole(out mockRawUI, out mockUI, out dispatcher);
InteractiveHelper.PostKeys(dispatcher, "nuget", appendCarriageReturn: true, timeout: TimeSpan.FromSeconds(5));
SecureString secure = mockUI.Object.ReadLineAsSecureString();
IntPtr bstr = Marshal.SecureStringToBSTR(secure);
string line = Marshal.PtrToStringBSTR(bstr);
Marshal.FreeBSTR(bstr);
Assert.Equal("nuget", line);
}