Moving lp:~rackspace-titan/nova/extensions-xml-serialization to new branch based off of trunk. To remove dep on another branch.

1 files changed, 110 insertions, 3 deletions

diff --git a/nova/tests/api/openstack/test_extensions.py b/nova/tests/api/openstack/test_extensions.py
index 697c62e5c..6d3c1b1be 100644
--- a/nova/tests/api/openstack/test_extensions.py
+++ b/nova/tests/api/openstack/test_extensions.py
@@ -16,10 +16,11 @@
 #    under the License.
 
 import json
+import os.path
 import stubout
 import unittest
 import webob
-import os.path
+from xml.etree import ElementTree
 
 from nova import context
 from nova import flags
@@ -80,19 +81,73 @@ class StubExtensionManager(object):
 
 class ExtensionControllerTest(unittest.TestCase):
 
-    def test_index(self):
+    def setUp(self):
+        FLAGS.osapi_extensions_path = os.path.join(
+            os.path.dirname(__file__), "extensions")
+
+    def test_list_extensions_json(self):
         app = openstack.APIRouterV11()
         ext_midware = extensions.ExtensionMiddleware(app)
         request = webob.Request.blank("/extensions")
         response = request.get_response(ext_midware)
         self.assertEqual(200, response.status_int)
 
-    def test_get_by_alias(self):
+        # Make sure we have all the extensions.
+        data = json.loads(response.body)
+        names = [x['name'] for x in data['extensions']]
+        names.sort()
+        self.assertEqual(names, ["FlavorExtraSpecs", "Floating_ips",
+            "Fox In Socks", "Hosts", "Multinic", "Volumes"])
+
+        # Make sure that at least Fox in Sox is correct.
+        (fox_ext,) = [
+            x for x in data['extensions'] if x['alias'] == 'FOXNSOX']
+        self.assertEqual(fox_ext, {
+                "namespace" : "http://www.fox.in.socks/api/ext/pie/v1.0",
+                "name" : "Fox In Socks",
+                "updated" : "2011-01-22T13:25:27-06:00",
+                "description" : "The Fox In Socks Extension",
+                "alias" : "FOXNSOX",
+                "links" : []
+            }
+        )
+
+    def test_get_extension_json(self):
+        app = openstack.APIRouterV11()
+        ext_midware = extensions.ExtensionMiddleware(app)
+        request = webob.Request.blank("/extensions/FOXNSOX")
+        response = request.get_response(ext_midware)
+        self.assertEqual(200, response.status_int)
+
+        data = json.loads(response.body)
+        self.assertEqual(data['extension'], {
+                "namespace" : "http://www.fox.in.socks/api/ext/pie/v1.0",
+                "name" : "Fox In Socks",
+                "updated" : "2011-01-22T13:25:27-06:00",
+                "description" : "The Fox In Socks Extension",
+                "alias" : "FOXNSOX",
+                "links" : []
+            }
+        )
+
+    def test_get_extension_xml(self):
         app = openstack.APIRouterV11()
         ext_midware = extensions.ExtensionMiddleware(app)
         request = webob.Request.blank("/extensions/FOXNSOX")
+        request.accept = "application/xml"
         response = request.get_response(ext_midware)
         self.assertEqual(200, response.status_int)
+        print response.body
+
+        elem = ElementTree.XML(response.body.replace("  ", ""))
+        self.assertEqual(elem.get('alias'), 'FOXNSOX')
+        self.assertEqual(elem.get('name'), 'Fox In Socks')
+        self.assertEqual(elem.get('namespace'),
+            'http://www.fox.in.socks/api/ext/pie/v1.0')
+        self.assertEqual(elem.get('updated'), '2011-01-22T13:25:27-06:00')
+        ns = "{http://docs.openstack.org/compute/api/v1.1}"
+        self.assertEqual(elem.findtext('{0}description'.format(ns)),
+            'The Fox In Socks Extension')
 
 
 class ResourceExtensionTest(unittest.TestCase):
@@ -244,3 +299,55 @@ class RequestExtensionTest(unittest.TestCase):
         response_data = json.loads(response.body)
         self.assertEqual('newblue', response_data['flavor']['googoose'])
         self.assertEqual("Pig Bands!", response_data['big_bands'])
+
+class ExtensionsXMLSerializerTest(unittest.TestCase):
+    def test_serialize(self):
+        serializer = extensions.ExtensionsXMLSerializer()
+        data = {
+            'extension': {
+                'name': 'ext1',
+                "namespace" : "http://docs.rack.com/servers/api/ext/pie/v1.0",
+                "alias" : "RS-PIE",
+                "updated" : "2011-01-22T13:25:27-06:00",
+                "description" : "Adds the capability to share an image.",
+                "links" : [
+                    {
+                        "rel" : "describedby",
+                        "type" : "application/pdf",
+                        "href" : "http://docs.rack.com/servers/api/ext/cs.pdf"
+                    },
+                    {
+                        "rel" : "describedby",
+                        "type" : "application/vnd.sun.wadl+xml",
+                        "href" : "http://docs.rack.com/servers/api/ext/cs.wadl"
+                    }
+                ]
+            },
+        }
+
+        ns = "{http://docs.openstack.org/compute/api/v1.1}"
+        atomns = "{http://www.w3.org/2005/Atom}"
+        elem = ElementTree.XML(serializer.serialize(data, 'show'))
+        self.assertEqual(elem.tag.split('extension')[0], ns)
+        self.assertEqual(elem.get('alias'), 'RS-PIE')
+        self.assertEqual(elem.get('name'), 'ext1')
+        self.assertEqual(elem.get('namespace'),
+            'http://docs.rack.com/servers/api/ext/pie/v1.0')
+        self.assertEqual(elem.get('updated'), '2011-01-22T13:25:27-06:00')
+        self.assertEqual(elem.findtext('{0}description'.format(ns)),
+            'Adds the capability to share an image.')
+
+        link_nodes = elem.findall('{0}link'.format(atomns))
+        self.assertEqual(len(link_nodes), 2)
+        link_nodes.sort(key=lambda x: x.get('type'))
+
+        self.assertEqual(link_nodes[0].get('type'), 'application/pdf')
+        self.assertEqual(link_nodes[0].get('rel'), 'describedby')
+        self.assertEqual(link_nodes[0].get('href'), 
+            'http://docs.rack.com/servers/api/ext/cs.pdf')
+
+        self.assertEqual(link_nodes[1].get('type'), 
+            'application/vnd.sun.wadl+xml')
+        self.assertEqual(link_nodes[1].get('rel'), 'describedby')
+        self.assertEqual(link_nodes[1].get('href'), 
+            'http://docs.rack.com/servers/api/ext/cs.wadl')