PowerShell/test/csharp/test_Runspace.cs

97 lines
3 KiB
C#
Raw Normal View History

using Xunit;
using System;
using System.Management.Automation;
using System.Management.Automation.Runspaces;
namespace PSTests
{
2015-12-01 19:45:20 +01:00
// NOTE: do not call AddCommand("out-host") after invoking or MergeMyResults,
// otherwise Invoke will not return any objects
[Collection("AssemblyLoadContext")]
public class RunspaceTests
{
2015-12-01 19:45:20 +01:00
private static int count = 3;
private static string script = String.Format($"get-command | select-object -first {count}");
2015-12-01 19:45:20 +01:00
[Fact]
public void TestRunspaceWithPipeline()
{
using (Runspace runspace = RunspaceFactory.CreateRunspace())
{
runspace.Open();
using (var pipeline = runspace.CreatePipeline(script))
{
int objCount = 0;
foreach (var result in pipeline.Invoke())
{
++objCount;
Assert.NotNull(result);
}
Assert.Equal(count, objCount);
}
runspace.Close();
}
}
[Fact]
2015-12-01 19:45:20 +01:00
public void TestRunspaceWithPowerShell()
{
using (var runspace = RunspaceFactory.CreateRunspace())
{
runspace.Open();
using (PowerShell powerShell = PowerShell.Create())
{
powerShell.Runspace = runspace;
powerShell.AddScript(script);
int objCount = 0;
foreach (var result in powerShell.Invoke())
{
++objCount;
2015-12-01 21:21:20 +01:00
Assert.NotNull(result);
}
Assert.Equal(count, objCount);
}
runspace.Close();
}
}
2016-06-10 19:03:49 +02:00
[Fact(Skip="Fails in Travis CI, need investigation")]
2015-12-01 19:45:20 +01:00
public void TestRunspaceWithPowerShellAndInitialSessionState()
{
InitialSessionState iss = InitialSessionState.CreateDefault2();
// NOTE: instantiate custom host myHost for the next line to capture stdout and stderr output
// in addition to just the PSObjects
2015-12-01 19:45:20 +01:00
using (Runspace runspace = RunspaceFactory.CreateRunspace(/*myHost,*/iss))
{
2015-12-01 19:45:20 +01:00
runspace.Open();
using (PowerShell powerShell = PowerShell.Create())
{
2015-12-01 19:45:20 +01:00
powerShell.Runspace = runspace;
2015-12-01 19:45:20 +01:00
powerShell.AddScript(script);
int objCount = 0;
2015-12-01 19:45:20 +01:00
foreach (var result in powerShell.Invoke())
{
// this is how an object would be captured here and looked at,
// each result is a PSObject with the data from the pipeline
++objCount;
Assert.NotNull(result);
}
2015-12-01 19:45:20 +01:00
Assert.Equal(count, objCount);
powerShell.Dispose();
}
}
}
}
}