From d32af928464afa3d1c6a233efaee3131be784761 Mon Sep 17 00:00:00 2001
From: John Eo <joon.eo@gmail.com>
Date: Mon, 9 May 2011 11:57:33 -0500
Subject: Adding unit test for the URL extension handler

---
 test/unit/test_exthandler.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 test/unit/test_exthandler.py

diff --git a/test/unit/test_exthandler.py b/test/unit/test_exthandler.py
new file mode 100644
index 00000000..f7c32fee
--- /dev/null
+++ b/test/unit/test_exthandler.py
@@ -0,0 +1,35 @@
+import os
+import sys
+# Need to access identity module
+sys.path.append(os.path.abspath(os.path.join(
+    os.getcwd(), '..', '..', 'keystone'))
+)
+from queryext.exthandler import UrlExtensionFilter
+import unittest
+
+
+class MockWsgiApp(object):
+
+    def __init__(self):
+        pass
+
+    def __call__(self, env, start_response):
+        pass
+
+
+class UrlExtensionFilterTest(unittest.TestCase):
+
+    def setUp(self):
+        self.filter = UrlExtensionFilter(MockWsgiApp(), {})
+
+    def test_xml_extension(self):
+        def _start_response():
+            pass
+        env = {'PATH_INFO': '/v1.0/someresource.xml'}
+        self.filter(env, _start_response)
+        self.assertEqual('/v1.0/someresource', env['PATH_INFO'])
+        self.assertEqual('application/xml', env['HTTP_ACCEPT'])
+
+
+if __name__ == '__main__':
+    unittest.main()
-- 
cgit