public async Task WhenCredentialsInvalid_ThenErrorIsShownAndWindowIsClosed(
[WindowsInstance(MachineType = MachineTypeForRdp)] ResourceTask <InstanceLocator> testInstance,
[Credential(Role = PredefinedRole.IapTunnelUser)] ResourceTask <ICredential> credential)
{
var locator = await testInstance;
using (var tunnel = IapTunnel.ForRdp(
locator,
await credential))
{
var settings = InstanceConnectionSettings.CreateNew(
locator.ProjectId,
locator.Name);
settings.RdpUsername.StringValue = "wrong";
settings.RdpPassword.Value = SecureStringExtensions.FromClearText("wrong");
settings.RdpAuthenticationLevel.EnumValue = RdpAuthenticationLevel.NoServerAuthentication;
settings.RdpUserAuthenticationBehavior.EnumValue = RdpUserAuthenticationBehavior.AbortOnFailure;
settings.RdpDesktopSize.EnumValue = RdpDesktopSize.ClientSize;
var rdpService = new RemoteDesktopSessionBroker(this.serviceProvider);
var session = rdpService.Connect(
locator,
"localhost",
(ushort)tunnel.LocalPort,
settings);
AwaitEvent <SessionAbortedEvent>();
Assert.IsNotNull(this.ExceptionShown);
Assert.IsInstanceOf(typeof(RdpDisconnectedException), this.ExceptionShown);
Assert.AreEqual(2055, ((RdpDisconnectedException)this.ExceptionShown).DisconnectReason);
}
}
public async Task WhenConnected_ThenActiveSessionIsSetAndTryActivateReturnsTrue(
[WindowsInstance(MachineType = MachineTypeForRdp)] ResourceTask <InstanceLocator> testInstance,
[Credential(Role = PredefinedRole.IapTunnelUser)] ResourceTask <ICredential> credential)
{
var locator = await testInstance;
using (var tunnel = RdpTunnel.Create(
locator,
await credential))
using (var gceAdapter = new ComputeEngineAdapter(this.serviceProvider.GetService <IAuthorizationAdapter>()))
{
var credentials = await gceAdapter.ResetWindowsUserAsync(
locator,
CreateRandomUsername(),
TimeSpan.FromSeconds(60),
CancellationToken.None);
var settings = VmInstanceConnectionSettings.CreateNew(
locator.ProjectId,
locator.Name);
settings.Username.StringValue = credentials.UserName;
settings.Password.Value = credentials.SecurePassword;
// Connect
var broker = new RemoteDesktopSessionBroker(this.serviceProvider);
IRemoteDesktopSession session = null;
AssertRaisesEvent <SessionStartedEvent>(
() => session = broker.Connect(
locator,
"localhost",
(ushort)tunnel.LocalPort,
settings));
Assert.IsNull(this.ExceptionShown);
// Verify session is connected.
Assert.IsTrue(broker.IsConnected(locator));
Assert.AreSame(session, broker.ActiveSession);
Assert.IsTrue(broker.TryActivate(locator));
// Verify dummy session is not connected.
Assert.IsFalse(broker.TryActivate(SampleLocator));
Assert.IsFalse(broker.IsConnected(SampleLocator));
AssertRaisesEvent <SessionEndedEvent>(
() => session.Close());
}
}
public void WhenWrongPort_ThenErrorIsShownAndWindowIsClosed()
{
var settings = VmInstanceConnectionSettings.CreateNew(this.SampleLocator);
var rdpService = new RemoteDesktopSessionBroker(this.serviceProvider);
rdpService.Connect(
this.SampleLocator,
"localhost",
135, // That one will be listening, but it is RPC, not RDP.
settings);
AwaitEvent <SessionAbortedEvent>();
Assert.IsInstanceOf(typeof(RdpDisconnectedException), this.ExceptionShown);
Assert.AreEqual(2308, ((RdpDisconnectedException)this.ExceptionShown).DisconnectReason);
}
public void WhenServerInvalid_ThenErrorIsShownAndWindowIsClosed()
{
var settings = VmInstanceConnectionSettings.CreateNew(this.SampleLocator);
var rdpService = new RemoteDesktopSessionBroker(this.serviceProvider);
rdpService.Connect(
this.SampleLocator,
"invalid.corp",
3389,
settings);
AwaitEvent <SessionAbortedEvent>();
Assert.IsInstanceOf(typeof(RdpDisconnectedException), this.ExceptionShown);
Assert.AreEqual(260, ((RdpDisconnectedException)this.ExceptionShown).DisconnectReason);
}
public async Task WhenSigningOutPerSendKeys_ThenWindowIsClosed(
[WindowsInstance(ImageFamily = WindowsInstanceAttribute.WindowsServer2019)]
ResourceTask <InstanceLocator> testInstance,
[Credential(Role = PredefinedRole.IapTunnelUser)] ResourceTask <ICredential> credential)
{
var locator = await testInstance;
using (var tunnel = RdpTunnel.Create(
locator,
await credential))
using (var gceAdapter = new ComputeEngineAdapter(this.serviceProvider.GetService <IAuthorizationAdapter>()))
{
var credentials = await gceAdapter.ResetWindowsUserAsync(
locator,
CreateRandomUsername(),
TimeSpan.FromSeconds(60),
CancellationToken.None);
var settings = VmInstanceConnectionSettings.CreateNew(
locator.ProjectId,
locator.Name);
settings.Username.StringValue = credentials.UserName;
settings.Password.Value = credentials.SecurePassword;
settings.AuthenticationLevel.EnumValue = RdpAuthenticationLevel.NoServerAuthentication;
settings.BitmapPersistence.EnumValue = RdpBitmapPersistence.Disabled;
settings.DesktopSize.EnumValue = RdpDesktopSize.ClientSize;
var rdpService = new RemoteDesktopSessionBroker(this.serviceProvider);
var session = (RemoteDesktopPane)rdpService.Connect(
locator,
"localhost",
(ushort)tunnel.LocalPort,
settings);
AwaitEvent <SessionStartedEvent>();
Thread.Sleep(5000);
session.ShowSecurityScreen();
Thread.Sleep(1000);
session.SendKeys(Keys.Menu, Keys.S); // Sign out.
AwaitEvent <SessionEndedEvent>();
Assert.IsNull(this.ExceptionShown);
}
}
public void WhenPortNotListening_ThenErrorIsShownAndWindowIsClosed()
{
var settings = VmInstanceConnectionSettings.CreateNew(this.SampleLocator);
settings.ConnectionTimeout.IntValue = 5;
var rdpService = new RemoteDesktopSessionBroker(this.serviceProvider);
rdpService.Connect(
this.SampleLocator,
"localhost",
1,
settings);
AwaitEvent <SessionAbortedEvent>();
Assert.IsInstanceOf(typeof(RdpDisconnectedException), this.ExceptionShown);
Assert.AreEqual(516, ((RdpDisconnectedException)this.ExceptionShown).DisconnectReason);
}
public void WhenNotConnected_ThenActiveSessionReturnsNull()
{
var broker = new RemoteDesktopSessionBroker(this.serviceProvider);
Assert.IsNull(broker.ActiveSession);
}
public async Task WhenCredentialsValid_ThenConnectingSucceeds(
[Values(RdpConnectionBarState.AutoHide, RdpConnectionBarState.Off, RdpConnectionBarState.Pinned)]
RdpConnectionBarState connectionBarState,
[Values(RdpDesktopSize.ClientSize, RdpDesktopSize.ScreenSize)]
RdpDesktopSize desktopSize,
[Values(RdpAudioMode.DoNotPlay, RdpAudioMode.PlayLocally, RdpAudioMode.PlayOnServer)]
RdpAudioMode audioMode,
[Values(RdpRedirectClipboard.Disabled, RdpRedirectClipboard.Enabled)]
RdpRedirectClipboard redirectClipboard,
[WindowsInstance(MachineType = MachineTypeForRdp)] ResourceTask <InstanceLocator> testInstance,
[Credential(Role = PredefinedRole.IapTunnelUser)] ResourceTask <ICredential> credential)
{
var locator = await testInstance;
using (var tunnel = RdpTunnel.Create(
locator,
await credential))
using (var gceAdapter = new ComputeEngineAdapter(this.serviceProvider.GetService <IAuthorizationAdapter>()))
{
var credentials = await gceAdapter.ResetWindowsUserAsync(
locator,
CreateRandomUsername(),
TimeSpan.FromSeconds(60),
CancellationToken.None);
var settings = VmInstanceConnectionSettings.CreateNew(
locator.ProjectId,
locator.Name);
settings.Username.StringValue = credentials.UserName;
settings.Password.Value = credentials.SecurePassword;
settings.ConnectionBar.EnumValue = connectionBarState;
settings.DesktopSize.EnumValue = desktopSize;
settings.AudioMode.EnumValue = audioMode;
settings.RedirectClipboard.EnumValue = redirectClipboard;
settings.AuthenticationLevel.EnumValue = RdpAuthenticationLevel.NoServerAuthentication;
settings.BitmapPersistence.EnumValue = RdpBitmapPersistence.Disabled;
var rdpService = new RemoteDesktopSessionBroker(this.serviceProvider);
var session = rdpService.Connect(
locator,
"localhost",
(ushort)tunnel.LocalPort,
settings);
AwaitEvent <SessionStartedEvent>();
Assert.IsNull(this.ExceptionShown);
SessionEndedEvent expectedEvent = null;
this.serviceProvider.GetService <IEventService>()
.BindHandler <SessionEndedEvent>(e =>
{
expectedEvent = e;
});
session.Close();
Assert.IsNotNull(expectedEvent);
}
}
