Fix link to work page

[djsite.git] / quotes / conftest.py

diff --git a/quotes/conftest.py b/quotes/conftest.py
index 9210e6aeb60a2049e2837372fda0ff9fffe8bd05..26ac8efe0b44e9db997681542358e62e7c125349 100644 (file)
--- a/quotes/conftest.py
+++ b/quotes/conftest.py
@@ -1,11 +1,13 @@
 import pytest
 
 import lxml.etree
+import html5lib
+import urlparse
 
 class ValidatingClient(object):
     def __init__(self, client):
         self.client = client
-    
+
     def request(self, url, method, exp_status=200, params={}):
         if not url.startswith('/quotes/'):
             url = '/quotes/' + url
@@ -23,6 +25,10 @@ class ValidatingClient(object):
         document = response.content.decode(response.charset)
         print 'For url %s got page:\n%s' % (url, document)
         lxml.etree.fromstring(document.replace('<br>', '<br/>'))
+
+        parser = html5lib.HTMLParser(strict=True)
+        parser.parse(document)
+
         assert '<script>' not in document
         return document
 
@@ -32,6 +38,17 @@ class ValidatingClient(object):
     def postPage(self, url, params, exp_status=200):
         return self.request(url, 'post', params=params, exp_status=exp_status)
 
+    def checkAllLinks(self, url, document):
+        tree = lxml.etree.fromstring(document)
+        links = tree.xpath("//a")
+        for link in links:
+            href = link.attrib['href']
+            if href == '#':
+                continue
+            fullurl = urlparse.urljoin(url, href)
+            print "asking for fullurl=", fullurl
+            assert self.getPage(fullurl)
+
 @pytest.fixture
 def c(client):
     return ValidatingClient(client)