private static BrokeredMessage PrepareMessage(string messageType, VirtualMachine virtualMachine)
{
var message = new BrokeredMessage();
message.SetMessageType(messageType);
message.SetObject(virtualMachine);
return message;
}
public void HandleMessage_VirtualMachineExists_PowershellIsInvokedOnce()
{
var message = new BrokeredMessage();
message.SetMessageType("Create");
message.SetObject(new VirtualMachine { Name = "SomeName" });
var executor = MockRepository.GenerateStub<IPowershellExecutor>();
executor.Expect(x => x.Execute(Arg<PowerShellCommand[]>.Is.TypeOf))
.Return(new[] { new PSObject() })
.Repeat.Once();
var messageHandler = new SowMessageHandler(executor);
messageHandler.HandleMessage(message);
executor.VerifyAllExpectations();
}
public void HandleMessage_VirtualMachineDoesNotExist_PowershellIsInvokedOnce()
{
var message = new BrokeredMessage();
message.SetMessageType("Delete");
message.SetObject(new VirtualMachine {Name = "SomeName"});
var executor = MockRepository.GenerateStub<IPowershellExecutor>();
executor.Expect(x => x.Execute(Arg<PowerShellCommand[]>.Is.TypeOf))
.Return(Enumerable.Empty<PSObject>())
.Repeat.Once();
var messageHandler = new ReapMessageHandler(executor);
messageHandler.HandleMessage(message);
executor.VerifyAllExpectations();
}
public void SetAndGetVirtualMachine_AdminUserNameIsPersisted()
{
var message = new BrokeredMessage();
var expected = new VirtualMachine
{
Name = "Name",
AdminUserName = "******",
AdminPassword = "******",
Location = "Location",
Size = AzureVirtualMachineSize.ExtraSmall,
TimeZone = "TimeZone"
};
message.SetObject(expected);
var actual = message.GetObject<VirtualMachine>();
Assert.That(actual.AdminUserName, Is.EqualTo(expected.AdminUserName));
}