Coverage.

repoze · Dec 28, 2009 · 8831b89 · 8831b89
1 parent 12ee7a9
commit 8831b89
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/repoze/vhm/tests/test_utils.py b/repoze/vhm/tests/test_utils.py
@@ -143,3 +143,18 @@ def test_with_virtual_root(self):
         environ = {'repoze.vhm.virtual_root':'/abc'}
         f = self._getFUT()
         self.assertEqual(f(environ), '/abc')
+
+class Test_getVirtualURL(unittest.TestCase):
+    def _getFUT(self):
+        from repoze.vhm.utils import getVirtualURL
+        return getVirtualURL
+
+    def test_without_virtual_root(self):
+        environ = {}
+        f = self._getFUT()
+        self.assertEqual(f(environ), None)
+
+    def test_with_virtual_root(self):
+        environ = {'repoze.vhm.virtual_url':'/a/b/c'}
+        f = self._getFUT()
+        self.assertEqual(f(environ), '/a/b/c')