| # 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 sys |
| import unittest |
| from datetime import datetime |
| |
| from libcloud.test import MockHttp, LibcloudTestCase |
| from libcloud.utils.py3 import httplib, assertRaisesRegex |
| from libcloud.compute.base import NodeImage |
| from libcloud.test.secrets import SCALEWAY_PARAMS |
| from libcloud.utils.iso8601 import UTC |
| from libcloud.common.exceptions import BaseHTTPError |
| from libcloud.test.file_fixtures import ComputeFileFixtures |
| from libcloud.compute.drivers.scaleway import ScalewayNodeDriver |
| |
| try: |
| import simplejson as json |
| except ImportError: |
| import json # NOQA |
| |
| |
| # class ScalewayTests(unittest.TestCase, TestCaseMixin): |
| class Scaleway_Tests(LibcloudTestCase): |
| def setUp(self): |
| ScalewayNodeDriver.connectionCls.conn_class = ScalewayMockHttp |
| ScalewayMockHttp.type = None |
| self.driver = ScalewayNodeDriver(*SCALEWAY_PARAMS) |
| |
| def test_authentication(self): |
| ScalewayMockHttp.type = "UNAUTHORIZED" |
| assertRaisesRegex(self, BaseHTTPError, "Authentication error", self.driver.list_nodes) |
| |
| def test_list_locations_success(self): |
| locations = self.driver.list_locations() |
| self.assertTrue(len(locations) >= 1) |
| |
| location = locations[0] |
| self.assertEqual(location.id, "par1") |
| self.assertEqual(location.name, "Paris 1") |
| |
| def test_list_sizes_success(self): |
| sizes = self.driver.list_sizes() |
| self.assertTrue(len(sizes) >= 1) |
| |
| size = sizes[0] |
| self.assertTrue(size.id is not None) |
| self.assertEqual(size.name, "ARM64-4GB") |
| self.assertEqual(size.ram, 4096) |
| |
| size = sizes[1] |
| self.assertTrue(size.id is not None) |
| self.assertEqual(size.name, "START1-XS") |
| self.assertEqual(size.ram, 1024) |
| |
| size = sizes[2] |
| self.assertTrue(size.id is not None) |
| self.assertEqual(size.name, "X64-120GB") |
| self.assertEqual(size.ram, 122880) |
| |
| def test_list_images_success(self): |
| images = self.driver.list_images() |
| self.assertTrue(len(images) >= 1) |
| |
| image = images[0] |
| self.assertTrue(image.id is not None) |
| self.assertTrue(image.name is not None) |
| |
| def test_create_image_success(self): |
| node = self.driver.list_nodes()[0] |
| ScalewayMockHttp.type = "POST" |
| image = self.driver.create_image(node, "my_image") |
| self.assertEqual(image.name, "my_image") |
| self.assertEqual(image.id, "98bf3ac2-a1f5-471d-8c8f-1b706ab57ef0") |
| self.assertEqual(image.extra["arch"], "arm") |
| |
| def test_delete_image_success(self): |
| image = self.driver.get_image(12345) |
| ScalewayMockHttp.type = "DELETE" |
| result = self.driver.delete_image(image) |
| self.assertTrue(result) |
| |
| def test_get_image_success(self): |
| image = self.driver.get_image(12345) |
| self.assertEqual(image.name, "my_image") |
| self.assertEqual(image.id, "12345") |
| self.assertEqual(image.extra["arch"], "arm") |
| |
| def test_list_nodes_success(self): |
| nodes = self.driver.list_nodes() |
| self.assertEqual(len(nodes), 2) |
| self.assertEqual(nodes[0].name, "my_server") |
| self.assertEqual(nodes[0].public_ips, []) |
| self.assertEqual( |
| nodes[0].extra["volumes"]["0"]["id"], "c1eb8f3a-4f0b-4b95-a71c-93223e457f5a" |
| ) |
| self.assertEqual(nodes[0].extra["organization"], "000a115d-2852-4b0a-9ce8-47f1134ba95a") |
| |
| def test_list_nodes_fills_created_datetime(self): |
| nodes = self.driver.list_nodes() |
| self.assertEqual(nodes[0].created_at, datetime(2014, 5, 22, 12, 57, 22, 514298, tzinfo=UTC)) |
| |
| def test_create_node_success(self): |
| image = self.driver.list_images()[0] |
| size = self.driver.list_sizes()[0] |
| location = self.driver.list_locations()[0] |
| |
| ScalewayMockHttp.type = "POST" |
| node = self.driver.create_node(name="test", size=size, image=image, region=location) |
| self.assertEqual(node.name, "my_server") |
| self.assertEqual(node.public_ips, []) |
| self.assertEqual(node.extra["volumes"]["0"]["id"], "d9257116-6919-49b4-a420-dcfdff51fcb1") |
| self.assertEqual(node.extra["organization"], "000a115d-2852-4b0a-9ce8-47f1134ba95a") |
| |
| def test_create_node_invalid_size(self): |
| image = NodeImage(id="01234567-89ab-cdef-fedc-ba9876543210", name=None, driver=self.driver) |
| size = self.driver.list_sizes()[0] |
| location = self.driver.list_locations()[0] |
| |
| ScalewayMockHttp.type = "INVALID_IMAGE" |
| expected_msg = '" not found' |
| assertRaisesRegex( |
| self, |
| Exception, |
| expected_msg, |
| self.driver.create_node, |
| name="test", |
| size=size, |
| image=image, |
| region=location, |
| ) |
| |
| def test_reboot_node_success(self): |
| node = self.driver.list_nodes()[0] |
| ScalewayMockHttp.type = "REBOOT" |
| result = self.driver.reboot_node(node) |
| self.assertTrue(result) |
| |
| def test_destroy_node_success(self): |
| node = self.driver.list_nodes()[0] |
| ScalewayMockHttp.type = "TERMINATE" |
| result = self.driver.destroy_node(node) |
| self.assertTrue(result) |
| |
| def test_list_volumes(self): |
| volumes = self.driver.list_volumes() |
| self.assertEqual(len(volumes), 2) |
| volume = volumes[0] |
| self.assertEqual(volume.id, "f929fe39-63f8-4be8-a80e-1e9c8ae22a76") |
| self.assertEqual(volume.name, "volume-0-1") |
| self.assertEqual(volume.size, 10) |
| self.assertEqual(volume.driver, self.driver) |
| |
| def test_list_volumes_empty(self): |
| ScalewayMockHttp.type = "EMPTY" |
| volumes = self.driver.list_volumes() |
| self.assertEqual(len(volumes), 0) |
| |
| def test_list_volume_snapshots(self): |
| volume = self.driver.list_volumes()[0] |
| snapshots = self.driver.list_volume_snapshots(volume) |
| self.assertEqual(len(snapshots), 2) |
| snapshot1, snapshot2 = snapshots |
| self.assertEqual(snapshot1.id, "6f418e5f-b42d-4423-a0b5-349c74c454a4") |
| self.assertEqual(snapshot2.id, "c6ff5501-eb35-44b8-aa01-8777211a830b") |
| |
| def test_create_volume(self): |
| par1 = [r for r in self.driver.list_locations() if r.id == "par1"][0] |
| ScalewayMockHttp.type = "POST" |
| volume = self.driver.create_volume(10, "volume-0-3", par1) |
| self.assertEqual(volume.id, "c675f420-cfeb-48ff-ba2a-9d2a4dbe3fcd") |
| self.assertEqual(volume.name, "volume-0-3") |
| self.assertEqual(volume.size, 10) |
| self.assertEqual(volume.driver, self.driver) |
| |
| def test_create_volume_snapshot(self): |
| volume = self.driver.list_volumes()[0] |
| ScalewayMockHttp.type = "POST" |
| snapshot = self.driver.create_volume_snapshot(volume, "snapshot-0-1") |
| self.assertEqual(snapshot.id, "f0361e7b-cbe4-4882-a999-945192b7171b") |
| self.assertEqual(snapshot.extra["volume_type"], "l_ssd") |
| self.assertEqual(volume.driver, self.driver) |
| |
| def test_destroy_volume(self): |
| volume = self.driver.list_volumes()[0] |
| ScalewayMockHttp.type = "DELETE" |
| resp = self.driver.destroy_volume(volume) |
| self.assertTrue(resp) |
| |
| def test_destroy_volume_snapshot(self): |
| volume = self.driver.list_volumes()[0] |
| snapshot = self.driver.list_volume_snapshots(volume)[0] |
| ScalewayMockHttp.type = "DELETE" |
| result = self.driver.destroy_volume_snapshot(snapshot) |
| self.assertTrue(result) |
| |
| def test_list_key_pairs(self): |
| keys = self.driver.list_key_pairs() |
| self.assertEqual(len(keys), 1) |
| self.assertEqual(keys[0].name, "example") |
| self.assertEqual(keys[0].public_key, "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5") |
| self.assertEqual(keys[0].fingerprint, "f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d") |
| |
| def test_import_key_pair_from_string(self): |
| result = self.driver.import_key_pair_from_string( |
| name="example", key_material="ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5" |
| ) |
| self.assertTrue(result) |
| |
| def test_delete_key_pair(self): |
| key = self.driver.list_key_pairs()[0] |
| result = self.driver.delete_key_pair(key) |
| self.assertTrue(result) |
| |
| |
| class ScalewayMockHttp(MockHttp): |
| fixtures = ComputeFileFixtures("scaleway") |
| |
| def _products_servers(self, method, url, body, headers): |
| body = self.fixtures.load("list_sizes.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _products_servers_availability(self, method, url, body, headers): |
| body = self.fixtures.load("list_availability.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _servers_UNAUTHORIZED(self, method, url, body, headers): |
| body = self.fixtures.load("error.json") |
| return (httplib.UNAUTHORIZED, body, {}, httplib.responses[httplib.UNAUTHORIZED]) |
| |
| def _images(self, method, url, body, headers): |
| body = self.fixtures.load("list_images.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _images_POST(self, method, url, body, headers): |
| # create_image |
| body = self.fixtures.load("create_image.json") |
| return (httplib.CREATED, body, {}, httplib.responses[httplib.CREATED]) |
| |
| def _images_12345_DELETE(self, method, url, body, headers): |
| # delete_image |
| return (httplib.NO_CONTENT, body, {}, httplib.responses[httplib.NO_CONTENT]) |
| |
| def _images_12345(self, method, url, body, headers): |
| # get_image |
| body = self.fixtures.load("get_image.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _servers(self, method, url, body, headers): |
| body = self.fixtures.load("list_nodes.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _servers_POST(self, method, url, body, headers): |
| body = self.fixtures.load("create_node.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _servers_741db378_action_POST(self, method, url, body, headers): |
| # reboot_node |
| return (httplib.NO_CONTENT, body, {}, httplib.responses[httplib.NO_CONTENT]) |
| |
| def _servers_INVALID_IMAGE(self, method, url, body, headers): |
| body = self.fixtures.load("error_invalid_image.json") |
| return (httplib.NOT_FOUND, body, {}, httplib.responses[httplib.NOT_FOUND]) |
| |
| def _servers_741db378_action_REBOOT(self, method, url, body, headers): |
| # reboot_node |
| body = self.fixtures.load("reboot_node.json") |
| return (httplib.CREATED, body, {}, httplib.responses[httplib.CREATED]) |
| |
| def _servers_741db378_action_TERMINATE(self, method, url, body, headers): |
| # destroy_node |
| return (httplib.NO_CONTENT, body, {}, httplib.responses[httplib.NO_CONTENT]) |
| |
| def _volumes(self, method, url, body, headers): |
| body = self.fixtures.load("list_volumes.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _volumes_EMPTY(self, method, url, body, headers): |
| body = self.fixtures.load("list_volumes_empty.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _snapshots(self, method, url, body, headers): |
| body = self.fixtures.load("list_volume_snapshots.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _volumes_POST(self, method, url, body, headers): |
| body = self.fixtures.load("create_volume.json") |
| return (httplib.CREATED, body, {}, httplib.responses[httplib.CREATED]) |
| |
| def _snapshots_POST(self, method, url, body, headers): |
| body = self.fixtures.load("create_volume_snapshot.json") |
| return (httplib.CREATED, body, {}, httplib.responses[httplib.CREATED]) |
| |
| def _volumes_f929fe39_63f8_4be8_a80e_1e9c8ae22a76_DELETE(self, method, url, body, headers): |
| return (httplib.NO_CONTENT, None, {}, httplib.responses[httplib.NO_CONTENT]) |
| |
| def _snapshots_6f418e5f_b42d_4423_a0b5_349c74c454a4_DELETE(self, method, url, body, headers): |
| return (httplib.NO_CONTENT, None, {}, httplib.responses[httplib.NO_CONTENT]) |
| |
| def _tokens_token(self, method, url, body, headers): |
| body = self.fixtures.load("token_info.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| def _users_5bea0358(self, method, url, body, headers): |
| body = self.fixtures.load("user_info.json") |
| return (httplib.OK, body, {}, httplib.responses[httplib.OK]) |
| |
| |
| if __name__ == "__main__": |
| sys.exit(unittest.main()) |