Merge pull request #9970 from sivel/rax-integration

Add integration tests for the rax module

test/integration/Makefile

@@ -94,8 +94,7 @@ gce_cleanup:
 	python cleanup_gce.py -y --match="^$(CLOUD_RESOURCE_PREFIX)"
 
 rackspace_cleanup:
-	@echo "FIXME - cleanup_rax.py not yet implemented"
-	@# python cleanup_rax.py -y --match="^$(CLOUD_RESOURCE_PREFIX)"
+	python cleanup_rax.py -y --match="^$(CLOUD_RESOURCE_PREFIX)"
 
 $(CREDENTIALS_FILE):
 	@echo "No credentials file found.  A file named '$(CREDENTIALS_FILE)' is needed to provide credentials needed to run cloud tests.  See sample 'credentials.template' file."

test/integration/cleanup_rax.py

+#!/usr/bin/env python
+
+import os
+import yaml
+import argparse
+
+try:
+    import pyrax
+    HAS_PYRAX = True
+except ImportError:
+    HAS_PYRAX = False
+
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-y', '--yes', action='store_true', dest='assumeyes',
+                        default=False, help="Don't prompt for confirmation")
+    parser.add_argument('--match', dest='match_re',
+                        default='^ansible-testing',
+                        help='Regular expression used to find resources '
+                             '(default: %(default)s)')
+
+    return parser.parse_args()
+
+
+def authenticate():
+    try:
+        with open(os.path.realpath('./credentials.yml')) as f:
+            credentials = yaml.load(f)
+    except Exception as e:
+        raise SystemExit(e)
+
+    try:
+        pyrax.set_credentials(credentials.get('rackspace_username'),
+                              credentials.get('rackspace_api_key'))
+    except Exception as e:
+        raise SystemExit(e)
+
+
+def prompt_and_delete(item, prompt, assumeyes):
+    if not assumeyes:
+        assumeyes = raw_input(prompt).lower() == 'y'
+    assert (hasattr(item, 'delete') or hasattr(item, 'terminate'),
+            "Class <%s> has no delete or terminate attribute" % item.__class__)
+    if assumeyes:
+        if hasattr(item, 'delete'):
+            item.delete()
+            print ("Deleted %s" % item)
+        if hasattr(item, 'terminate'):
+            item.terminate()
+            print ("Terminated %s" % item)
+
+
+def delete_rax(args):
+    """Function for deleting CloudServers"""
+    for region in pyrax.identity.services.compute.regions:
+        cs = pyrax.connect_to_cloudservers(region=region)
+        servers = cs.servers.list(search_opts=dict(name='^%s' % args.match_re))
+        for server in servers:
+            prompt_and_delete(server,
+                              'Delete matching %s? [y/n]: ' % server,
+                              args.assumeyes)
+
+
+def main():
+    if not HAS_PYRAX:
+        raise SystemExit('The pyrax python module is required for this script')
+
+    args = parse_args()
+    authenticate()
+    delete_rax(args)
+
+
+if __name__ == '__main__':
+    main()

test/integration/credentials.template

 ---
+# Rackspace Credentials
+rackspace_username:
+rackspace_api_key:
+rackspace_region:
+
 # AWS Credentials
 ec2_access_key:
 ec2_secret_key:

test/integration/rackspace.yml

-- hosts: testhost
-  gather_facts: True
-  roles: []
-
+---
+- hosts: localhost
+  connection: local
+  gather_facts: false
+  tags:
+    - rackspace
+  roles:
+    - role: test_rax
+      tags: test_rax

test/integration/roles/test_rax/defaults/main.yml

+---
+rackspace_region: IAD
+resource_prefix: ansible-testing