# Licensed to the Apache Software Foundation (ASF) under one or more # contributor license agreements. See the NOTICE file distributed with # this work for additional information regarding copyright ownership. # The ASF licenses this file to You under the Apache License, Version 2.0 # (the "License"); you may not use this file except in compliance with # the License. You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import unittest import sys import random import string from libcloud.utils.py3 import httplib from libcloud.compute.drivers.gandi import GandiNodeDriver from libcloud.common.gandi import GandiException from libcloud.compute.types import NodeState from libcloud.test.file_fixtures import ComputeFileFixtures from libcloud.test.secrets import GANDI_PARAMS from libcloud.test.common.test_gandi import BaseGandiMockHttp class GandiTests(unittest.TestCase): node_name = 'test2' def setUp(self): GandiNodeDriver.connectionCls.conn_class = GandiMockHttp GandiMockHttp.type = None self.driver = GandiNodeDriver(*GANDI_PARAMS) def test_list_nodes(self): nodes = self.driver.list_nodes() self.assertTrue(len(nodes) > 0) self.assertTrue(len(nodes[0].public_ips) > 1) def test_list_locations(self): loc = list(filter(lambda x: 'france' in x.country.lower(), self.driver.list_locations()))[0] self.assertEqual(loc.country, 'France') def test_list_images(self): loc = list(filter(lambda x: 'france' in x.country.lower(), self.driver.list_locations()))[0] images = self.driver.list_images(loc) self.assertTrue(len(images) > 2) def test_list_sizes(self): sizes = self.driver.list_sizes() self.assertTrue(len(sizes) >= 1) def test_destroy_node_running(self): nodes = self.driver.list_nodes() test_node = list(filter(lambda x: x.state == NodeState.RUNNING, nodes))[0] self.assertTrue(self.driver.destroy_node(test_node)) def test_destroy_node_halted(self): nodes = self.driver.list_nodes() test_node = list(filter(lambda x: x.state == NodeState.TERMINATED, nodes))[0] self.assertTrue(self.driver.destroy_node(test_node)) def test_reboot_node(self): nodes = self.driver.list_nodes() test_node = list(filter(lambda x: x.state == NodeState.RUNNING, nodes))[0] self.assertTrue(self.driver.reboot_node(test_node)) def test_create_node(self): login = 'libcloud' passwd = ''.join(random.choice(string.ascii_letters) for i in range(10)) # Get france datacenter loc = list(filter(lambda x: 'france' in x.country.lower(), self.driver.list_locations()))[0] # Get a debian image images = self.driver.list_images(loc) images = [x for x in images if x.name.lower().startswith('debian')] img = list(filter(lambda x: '5' in x.name, images))[0] # Get a configuration size size = self.driver.list_sizes()[0] node = self.driver.create_node(name=self.node_name, login=login, password=passwd, image=img, location=loc, size=size) self.assertEqual(node.name, self.node_name) def test_create_volume(self): loc = list(filter(lambda x: 'france' in x.country.lower(), self.driver.list_locations()))[0] volume = self.driver.create_volume( size=1024, name='libcloud', location=loc) self.assertEqual(volume.name, 'libcloud') self.assertEqual(volume.size, 1024) def test_list_volumes(self): disks = self.driver.list_volumes() self.assertTrue(len(disks) > 0) def test_destroy_volume(self): volumes = self.driver.list_volumes() test_vol = list(filter(lambda x: x.name == 'test_disk', volumes))[0] self.assertTrue(self.driver.destroy_volume(test_vol)) def test_attach_volume(self): disks = self.driver.list_volumes() nodes = self.driver.list_nodes() res = self.driver.attach_volume(nodes[0], disks[0]) self.assertTrue(res) def test_detach_volume(self): disks = self.driver.list_volumes() nodes = self.driver.list_nodes() res = self.driver.detach_volume(nodes[0], disks[0]) self.assertTrue(res) def test_ex_list_interfaces(self): ifaces = self.driver.ex_list_interfaces() self.assertTrue(len(ifaces) > 0) def test_ex_attach_interface(self): ifaces = self.driver.ex_list_interfaces() nodes = self.driver.list_nodes() res = self.driver.ex_node_attach_interface(nodes[0], ifaces[0]) self.assertTrue(res) def test_ex_detach_interface(self): ifaces = self.driver.ex_list_interfaces() nodes = self.driver.list_nodes() res = self.driver.ex_node_detach_interface(nodes[0], ifaces[0]) self.assertTrue(res) def test_ex_snapshot_disk(self): disks = self.driver.list_volumes() self.assertTrue(self.driver.ex_snapshot_disk(disks[2])) self.assertRaises(GandiException, self.driver.ex_snapshot_disk, disks[0]) def test_ex_update_disk(self): disks = self.driver.list_volumes() self.assertTrue(self.driver.ex_update_disk(disks[0], new_size=4096)) def test_list_key_pairs(self): keys = self.driver.list_key_pairs() self.assertTrue(len(keys) > 0) def test_get_key_pair(self): key = self.driver.get_key_pair(10) self.assertEqual(key.name, 'testkey') def test_import_key_pair_from_string(self): key = self.driver.import_key_pair_from_string('testkey', '12345') self.assertEqual(key.name, 'testkey') self.assertEqual(key.extra['id'], 10) def test_delete_key_pair(self): response = self.driver.delete_key_pair(10) self.assertTrue(response) def test_ex_get_node(self): node = self.driver.ex_get_node(34951) self.assertEqual(node.name, "test2") def test_ex_get_volume(self): volume = self.driver.ex_get_volume(1263) self.assertEqual(volume.name, "libcloud") class GandiRatingTests(unittest.TestCase): """Tests where rating model is involved""" node_name = 'test2' def setUp(self): GandiNodeDriver.connectionCls.conn_class = GandiMockRatingHttp GandiMockRatingHttp.type = None self.driver = GandiNodeDriver(*GANDI_PARAMS) def test_list_sizes(self): sizes = self.driver.list_sizes() self.assertEqual(len(sizes), 4) def test_create_node(self): login = 'libcloud' passwd = ''.join(random.choice(string.ascii_letters) for i in range(10)) # Get france datacenter loc = list(filter(lambda x: 'france' in x.country.lower(), self.driver.list_locations()))[0] # Get a debian image images = self.driver.list_images(loc) images = [x for x in images if x.name.lower().startswith('debian')] img = list(filter(lambda x: '5' in x.name, images))[0] # Get a configuration size size = self.driver.list_sizes()[0] node = self.driver.create_node(name=self.node_name, login=login, password=passwd, image=img, location=loc, size=size) self.assertEqual(node.name, self.node_name) class GandiMockHttp(BaseGandiMockHttp): fixtures = ComputeFileFixtures('gandi') def _xmlrpc__hosting_datacenter_list(self, method, url, body, headers): body = self.fixtures.load('datacenter_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_image_list(self, method, url, body, headers): body = self.fixtures.load('image_list_dc0.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_list(self, method, url, body, headers): body = self.fixtures.load('vm_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_ip_list(self, method, url, body, headers): body = self.fixtures.load('ip_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_account_info(self, method, url, body, headers): body = self.fixtures.load('account_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_info(self, method, url, body, headers): body = self.fixtures.load('vm_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_delete(self, method, url, body, headers): body = self.fixtures.load('vm_delete.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__operation_info(self, method, url, body, headers): body = self.fixtures.load('operation_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_create_from(self, method, url, body, headers): body = self.fixtures.load('vm_create_from.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_reboot(self, method, url, body, headers): body = self.fixtures.load('vm_reboot.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_stop(self, method, url, body, headers): body = self.fixtures.load('vm_stop.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_iface_list(self, method, url, body, headers): body = self.fixtures.load('iface_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_disk_list(self, method, url, body, headers): body = self.fixtures.load('disk_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_iface_attach(self, method, url, body, headers): body = self.fixtures.load('iface_attach.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_iface_detach(self, method, url, body, headers): body = self.fixtures.load('iface_detach.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_disk_attach(self, method, url, body, headers): body = self.fixtures.load('disk_attach.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_disk_detach(self, method, url, body, headers): body = self.fixtures.load('disk_detach.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_disk_create(self, method, url, body, headers): body = self.fixtures.load('disk_create.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_disk_create_from(self, method, url, body, headers): body = self.fixtures.load('disk_create_from.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_disk_info(self, method, url, body, headers): body = self.fixtures.load('disk_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_disk_update(self, method, url, body, headers): body = self.fixtures.load('disk_update.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_disk_delete(self, method, url, body, headers): body = self.fixtures.load('disk_delete.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_ssh_info(self, method, url, body, headers): body = self.fixtures.load('ssh_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_ssh_list(self, method, url, body, headers): body = self.fixtures.load('ssh_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_ssh_create(self, method, url, body, headers): body = self.fixtures.load('ssh_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_ssh_delete(self, method, url, body, headers): body = self.fixtures.load('ssh_delete.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) class GandiMockRatingHttp(BaseGandiMockHttp): """Fixtures needed for tests related to rating model""" fixtures = ComputeFileFixtures('gandi') def _xmlrpc__hosting_datacenter_list(self, method, url, body, headers): body = self.fixtures.load('datacenter_list.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_image_list(self, method, url, body, headers): body = self.fixtures.load('image_list_dc0.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_create_from(self, method, url, body, headers): body = self.fixtures.load('vm_create_from.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__operation_info(self, method, url, body, headers): body = self.fixtures.load('operation_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) def _xmlrpc__hosting_vm_info(self, method, url, body, headers): body = self.fixtures.load('vm_info.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) # Specific to rating tests def _xmlrpc__hosting_account_info(self, method, url, body, headers): body = self.fixtures.load('account_info_rating.xml') return (httplib.OK, body, {}, httplib.responses[httplib.OK]) if __name__ == '__main__': sys.exit(unittest.main())