def test_fill_disk_on_win(init, fetch, open):
# arrange mocks
client = MagicMock()
init.return_value = client
resource = __get_resource(os_type='Windows')
resource_list = [resource]
fetch.return_value = resource_list
# run command mocks
poller = MagicMock()
client.virtual_machines.run_command.return_value = poller
result = MagicMock(spec=RunCommandResult)
poller.result.return_value = result
result.value = [InstanceViewStatus()]
# act
fill_disk("where name=='some_windows_machine'", 60, size=100)
# assert
fetch.assert_called_with("where name=='some_windows_machine'",
"Microsoft.Compute/virtualMachines", None, None)
open.assert_called_with(AnyStringWith("fill_disk.ps1"))
client.virtual_machines.run_command.assert_called_with(
resource['resourceGroup'], resource['name'], {
'command_id':
'RunPowerShellScript',
'script': ['script'],
'parameters': [{
'name': 'duration',
'value': 60
}, {
'name': 'size',
'value': 100
}]
})
def test_burn_io_on_lnx(init, fetch, open):
# arrange mocks
client = MagicMock()
init.return_value = client
resource = __get_resource(os_type='Linux')
resource_list = [resource]
fetch.return_value = resource_list
# run command mocks
poller = MagicMock()
client.virtual_machines.run_command.return_value = poller
result = MagicMock(spec=RunCommandResult)
poller.result.return_value = result
result.value = [InstanceViewStatus()]
# act
burn_io(filter="where name=='some_linux_machine'",
duration=60,
configuration=CONFIG,
secrets=SECRETS)
# assert
fetch.assert_called_with("where name=='some_linux_machine'",
"Microsoft.Compute/virtualMachines", SECRETS,
CONFIG)
open.assert_called_with(AnyStringWith("burn_io.sh"))
client.virtual_machines.run_command.assert_called_with(
resource['resourceGroup'], resource['name'], {
'command_id': 'RunShellScript',
'script': ['script'],
'parameters': [{
'name': 'duration',
'value': 60
}]
})
def instance_view(self, resource_group, name):
statuses = []
s = InstanceViewStatus(code='PowerState/Deallocated')
statuses.append(s)
return VirtualMachineInstanceView(statuses=statuses)