X-Git-Url: http://gitweb.fperrin.net/?a=blobdiff_plain;f=quotes%2Fconftest.py;h=b87a6baf2a756894143ec20c6f2db398144f8f01;hb=ac8dcbf73da18db58cedc61cb79882b8bf3e74b1;hp=9e88dc0a43b2797e6618f1ea49d214defc4d454c;hpb=6bc7c03c1fa9c289e4fff457e5404bffbb595fe7;p=djsite.git

diff --git a/quotes/conftest.py b/quotes/conftest.py
index 9e88dc0..b87a6ba 100644
--- a/quotes/conftest.py
+++ b/quotes/conftest.py
@@ -7,25 +7,34 @@ class ValidatingClient(object):
         self.client = client
     
     def request(self, url, method, exp_status=200, params={}):
+        if not url.startswith('/quotes/'):
+            url = '/quotes/' + url
+
         if method == 'get':
-            response = self.client.get('/quotes/' + url)
+            response = self.client.get(url)
         elif method == 'post':
-            response = self.client.post('/quotes/' + url, params)
+            response = self.client.post(url, params)
         else:
             raise RuntimeError('Unknown method %s for %s' % (method, url))
         assert response.status_code == exp_status
+        if response.status_code != 200:
+            return None
         assert response.charset == 'utf-8'
         document = response.content.decode(response.charset)
         lxml.etree.fromstring(document)
         assert '<script>' not in document
         return document
 
-    def getPage(self, url):
-        return self.request(url, 'get')
+    def getPage(self, url, exp_status=200):
+        return self.request(url, 'get', exp_status=exp_status)
 
-    def postPage(self, url, params):
-        return self.request(url, 'post', params=params)
+    def postPage(self, url, params, exp_status=200):
+        return self.request(url, 'post', params=params, exp_status=exp_status)
 
 @pytest.fixture
 def c(client):
     return ValidatingClient(client)
+
+@pytest.fixture
+def c_adm(admin_client):
+    return ValidatingClient(admin_client)