Initial support fo extension resources. Tests.

1 files changed, 83 insertions, 0 deletions

diff --git a/nova/tests/api/openstack/test_extensions.py b/nova/tests/api/openstack/test_extensions.py
new file mode 100644
index 000000000..f5332c84a
--- /dev/null
+++ b/nova/tests/api/openstack/test_extensions.py
@@ -0,0 +1,83 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright 2011 OpenStack LLC.
+# All Rights Reserved.
+#
+#    Licensed 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 webob
+
+from nova.api import openstack
+import nova.wsgi
+
+class StubController(nova.wsgi.Controller):
+
+    def __init__(self, body):
+        self.body = body
+
+    def index(self, req):
+        return self.body
+
+class StubExtensionManager(object):
+
+    def __init__(self, resources):
+        self.resources = resources
+
+    def get_resources(self):
+        return self.resources
+
+class WidgetExtensionResource(object):
+
+    def __init__(self, name, collection, wsgi_app):
+        self.name = name
+        self.collection = collection
+        self.wsgi_app = wsgi_app
+
+    def add_routes(self, mapper):
+        mapper.resource(self.name, self.collection, controller=self.wsgi_app)
+
+class ExtensionTest(unittest.TestCase):
+
+    def test_no_extension_present(self):
+        manager = StubExtensionManager([])
+        router = openstack.APIRouter(manager)
+        request = webob.Request.blank("/widgets")
+        response = request.get_response(router)
+        self.assertEqual(404, response.status_int)
+
+    def test_get_resources(self):
+        response_body = "Buy more widgets!"
+        response = webob.Response()
+        response.body = response_body
+        resource1 = WidgetExtensionResource("widget", "widgets", response)
+        manager = StubExtensionManager([resource1])
+        router = openstack.APIRouter(manager)
+        request = webob.Request.blank("/widgets")
+        response = request.get_response(router)
+        self.assertEqual(200, response.status_int)
+        self.assertEqual(response_body, response.body)
+
+    def test_get_resources_with_controller(self):
+        response_body = "Buy more widgets!"
+        controller = StubController(response_body)
+        resource1 = WidgetExtensionResource("widget", "widgets", controller)
+        manager = StubExtensionManager([resource1])
+        router = openstack.APIRouter(manager)
+        request = webob.Request.blank("/widgets")
+        response = request.get_response(router)
+        self.assertEqual(200, response.status_int)
+        self.assertEqual(response_body, response.body)
+
+