1 files changed, 108 insertions, 0 deletions
diff --git a/testing/robot/lib/FDSLibrary.py b/testing/robot/lib/FDSLibrary.py
new file mode 100644
index 0000000..2e29d6c
--- /dev/null
+++ b/testing/robot/lib/FDSLibrary.py
@@ -0,0 +1,108 @@
+from neutronclient.v2_0 import client as neutron
+from novaclient import client as nova
+from novaclient.exceptions import NotFound
+import time
+import datetime
+import os
+import subprocess
+
+class FDSLibrary():
+    def __init__(self):
+        self.neutron_client = neutron.Client(username=os.getenv('OS_USERNAME'),
+                                             password=os.getenv('OS_PASSWORD'),
+                                             tenant_name=os.getenv('OS_TENANT_NAME'),
+                                             auth_url=os.getenv('OS_AUTH_URL'))
+
+        self.nova_client = nova.Client('2',
+                                       os.getenv('OS_USERNAME'),
+                                       os.getenv('OS_PASSWORD'),
+                                       os.getenv('OS_TENANT_NAME'),
+                                       os.getenv('OS_AUTH_URL'))
+
+    def create_network(self, name):
+        body = {'network': {'name': name}}
+        response = self.neutron_client.create_network(body=body)
+        return response
+
+    def create_subnet(self, name, network_id, cidr, dhcp=True, ip_version=4):
+        body = {'subnet': {
+            'name': name,
+            'enable_dhcp': dhcp,
+            'network_id': network_id,
+            'ip_version': ip_version,
+            'cidr': cidr
+        }}
+        response = self.neutron_client.create_subnet(body=body)
+        return response
+
+    def create_port(self, name, net_id, subnet_id, ip_address, security_groups=None):
+        body = {'port': {
+            'name': name,
+            'network_id': net_id,
+            'fixed_ips': [
+                {
+                    'subnet_id': subnet_id,
+                    'ip_address': ip_address
+                }
+            ]
+        }}
+        if security_groups is not None:
+            # this is a list
+            body['port']['security_groups'] = security_groups
+        response = self.neutron_client.create_port(body=body)
+        return response
+
+    def create_server(self, name, image, flavor, port_ids, security_groups=None):
+        image = self.nova_client.images.find(name=image)
+        flavor = self.nova_client.flavors.find(name=flavor)
+        nics = [{'port-id': port_id} for port_id in port_ids]
+        response = self.nova_client.servers.create(name=name, image=image.id, flavor=flavor.id,
+                                                   security_groups=security_groups, nics=nics)
+        for key in dir(response):
+            print key, getattr(response, key)
+        return response
+
+    def poll_server(self, vm_id, status, timeout=300):
+        try:
+            start = datetime.datetime.now()
+            delta = datetime.timedelta(seconds=timeout)
+            vm_obj = self.nova_client.servers.get(vm_id)
+            current_state = getattr(vm_obj, 'OS-EXT-STS:vm_state', None)
+            print "Found state '{}' while looking for state '{}'".format(current_state, status)
+            while datetime.datetime.now() - start < delta and self._states_are_not_equal(current_state, status):
+                time.sleep(1)
+                vm_obj = self.nova_client.servers.get(vm_id)
+                current_state = getattr(vm_obj, 'OS-EXT-STS:vm_state', None)
+                print "Found state '{}' while looking for state '{}'".format(current_state, status)
+
+            if self._states_are_not_equal(current_state, status):
+                raise Exception("Vm with id '{}' is in state '{}' after timeout of {} seconds"
+                                "".format(vm_id, getattr(vm_obj, 'OS-EXT-STS:vm_state'), timeout))
+        except NotFound as e:
+            if status is not None:
+                raise e
+            else:
+                print "VM not found, which means it's been deleted"
+
+    def _states_are_not_equal(self, current_state, expected_state):
+        return current_state.lower() != expected_state
+
+    def delete_server(self, vm_id):
+        response = self.nova_client.servers.delete(vm_id)
+        return response
+
+    def delete_port(self, port_id):
+        response = self.neutron_client.delete_port(port_id)
+        return response
+
+    def delete_net(self, net_id):
+        response = self.neutron_client.delete_network(net_id)
+        return response
+
+    def ping_vm(self, ip_address):
+        try:
+            output = subprocess.check_output(['ping', '-c', '4', ip_address])
+            print output
+        except subprocess.CalledProcessError as e:
+            print e.output
+            raise Exception('Ping on ip {} failed'.format(ip_address))