def test_request_instance_with_accelerator(self):
"""
Test requesting an instance with GCE accelerators
"""
self.vm_.update({
"ex_accelerator_type": "foo",
"ex_accelerator_count": 42
})
call_kwargs = {
"ex_disk_type": "pd-standard",
"ex_metadata": {
"items": [{
"value": None,
"key": "salt-cloud-profile"
}]
},
"ex_accelerator_count": 42,
"name": "new",
"ex_service_accounts": None,
"external_ip": "ephemeral",
"ex_accelerator_type": "foo",
"ex_tags": None,
"ex_disk_auto_delete": True,
"ex_network": "default",
"ex_disks_gce_struct": None,
"ex_preemptible": False,
"ex_can_ip_forward": False,
"ex_on_host_maintenance": "TERMINATE",
"location": self.location,
"ex_subnetwork": None,
"image": "myimage",
"size": 1234,
}
with patch("salt.cloud.clouds.gce.get_conn",
MagicMock(return_value=self.conn)), patch(
"salt.cloud.clouds.gce.show_instance",
MagicMock()), patch(
"salt.cloud.clouds.gce.LIBCLOUD_VERSION_INFO",
(2, 3, 0)):
gce.request_instance(self.vm_)
self.conn.create_node.assert_called_once_with(**call_kwargs)
def test_request_instance_with_accelerator(config, location, conn,
fake_libcloud_2_5_0):
"""
Test requesting an instance with GCE accelerators
"""
config.update({"ex_accelerator_type": "foo", "ex_accelerator_count": 42})
call_kwargs = {
"ex_disk_type": "pd-standard",
"ex_metadata": {
"items": [{
"value": None,
"key": "salt-cloud-profile"
}]
},
"ex_accelerator_count": 42,
"name": "new",
"ex_service_accounts": None,
"external_ip": "ephemeral",
"ex_accelerator_type": "foo",
"ex_tags": None,
"ex_disk_auto_delete": True,
"ex_network": "default",
"ex_disks_gce_struct": None,
"ex_preemptible": False,
"ex_can_ip_forward": False,
"ex_on_host_maintenance": "TERMINATE",
"location": location,
"ex_subnetwork": None,
"image": "myimage",
"size": 1234,
}
gce.request_instance(config)
conn.create_node.assert_called_once_with(**call_kwargs)