def test_create_firewall_ssh(self):
"""
Test create when setting firewall_group_id and
ssh_key_names.
"""
kwargs = {
"provider": "vultr",
"enable_private_network": True,
"ssh_key_names": "key1,key2,key3",
"startup_script_id": "test_id",
"firewall_group_id": "f_id",
"image": 223,
"size": 13,
"location": 1,
"name": "test-vm",
}
patch_scripts = patch(
"salt.cloud.clouds.vultrpy.avail_scripts",
MagicMock(return_value=["test_id"]),
)
patch_firewall = patch(
"salt.cloud.clouds.vultrpy.avail_firewall_groups",
MagicMock(return_value=["f_id"]),
)
patch_keys = patch(
"salt.cloud.clouds.vultrpy.avail_keys",
MagicMock(return_value=["key3", "key2", "key1"]),
)
patch_vultrid = patch(
"salt.cloud.clouds.vultrpy._lookup_vultrid",
MagicMock(return_value="test_id"),
)
mock_query = MagicMock(return_value={"status": 200})
patch_query = patch("salt.cloud.clouds.vultrpy._query", mock_query)
patch_show = patch("salt.cloud.clouds.vultrpy.show_instance",
MagicMock())
with patch_scripts, patch_firewall, patch_keys, patch_vultrid, patch_query, patch_show:
vultr.create(kwargs)
query_ret = mock_query.call_args.kwargs["data"]
self.assertIn("SSHKEYID=key1%2Ckey2%2Ckey3", query_ret)
self.assertIn("FIREWALLGROUPID=f_id", query_ret)
def test_create_ssh_key_ids_doesnotexist(self):
"""
Test create when setting ssh_key_ids that do not
exist
"""
kwargs = {
"provider": "vultr",
"enable_private_network": True,
"startup_script_id": "test_id",
"ssh_key_names": "doesnotexist",
"image": 223,
"size": 13,
"location": 1,
"name": "test-vm",
}
patch_scripts = patch(
"salt.cloud.clouds.vultrpy.avail_scripts",
MagicMock(return_value=["test_id"]),
)
patch_firewall = patch(
"salt.cloud.clouds.vultrpy.avail_firewall_groups",
MagicMock(return_value=["f_id"]),
)
patch_keys = patch(
"salt.cloud.clouds.vultrpy.avail_keys",
MagicMock(return_value=["key3", "key2", "key1"]),
)
patch_vultrid = patch(
"salt.cloud.clouds.vultrpy._lookup_vultrid",
MagicMock(return_value="test_id"),
)
mock_query = MagicMock(return_value={"status": 200})
patch_query = patch("salt.cloud.clouds.vultrpy._query", mock_query)
patch_show = patch("salt.cloud.clouds.vultrpy.show_instance",
MagicMock())
with patch_scripts, patch_firewall, patch_keys, patch_vultrid, patch_query, patch_show:
with TstSuiteLoggingHandler() as handler:
ret = vultr.create(kwargs)
self.assertIn(
"ERROR:Your Vultr account does not have a key with ID doesnotexist",
handler.messages,
)
self.assertFalse(ret)