aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMuriithi Frederick Muriuki2018-03-12 13:00:03 +0300
committerPjotr Prins2018-03-26 09:29:29 +0000
commitdd5ec1d3080b74fa065620f6915eeacf2cca8a2d (patch)
treee9520ebbe59dd8f4339171a45a56842d5a2cbcc5
parenta29c16b770aa0bb53c632eecc1764caf02789fe7 (diff)
downloadgenenetwork2-dd5ec1d3080b74fa065620f6915eeacf2cca8a2d.tar.gz
Add tests to check links.
-rw-r--r--test/requests/link_checker.py62
1 files changed, 58 insertions, 4 deletions
diff --git a/test/requests/link_checker.py b/test/requests/link_checker.py
index f171c12a..256bf6ef 100644
--- a/test/requests/link_checker.py
+++ b/test/requests/link_checker.py
@@ -1,9 +1,63 @@
+from __future__ import print_function
+import re
import requests
+from lxml.html import parse
+from requests.exceptions import ConnectionError
+
+def is_root_link(link):
+ pattern = re.compile("^/$")
+ return pattern.match(link)
+
+def is_mailto_link(link):
+ pattern = re.compile("^mailto:.*")
+ return pattern.match(link)
+
+def is_internal_link(link):
+ pattern = re.compile("^/.*")
+ return pattern.match(link)
+
+def get_links(doc):
+ return filter(
+ lambda x: not (
+ is_root_link(x)
+ or is_mailto_link(x))
+ , map(lambda y: y.get("href")
+ , doc.cssselect("a")))
+
+def verify_link(link):
+ try:
+ result = requests.get(link, timeout=20)
+ if result.status_code == 200:
+ print(link+" ==> OK")
+ else:
+ print("ERROR: link `"+link+"` failed with status "
+ , result.status_code)
+ except ConnectionError as ex:
+ print("ERROR: ", link, ex)
+
+def check_page(host, start_url):
+ print("")
+ print("Checking links in page `"+start_url+"`")
+ doc = parse(start_url).getroot()
+ links = get_links(doc)
+ internal_links = filter(is_internal_link, links)
+ external_links = filter(lambda x: not is_internal_link(x), links)
+ external_links.append("http://somenon-existentsite.brr")
+ for link in internal_links:
+ verify_link(host+link)
+
+ for link in external_links:
+ verify_link(link)
def check_links(args_obj, parser):
print("")
print("Checking links")
- print("########################")
- print("Not implemented yet.")
- print("This is supposed to check all links in the system.")
- print("########################")
+ host = args_obj.host
+
+ # Check the home page
+ check_page(host, host)
+
+ # Check traits page
+ check_page(
+ host,
+ host+"/show_trait?trait_id=1435395_s_at&dataset=HC_M2_0606_P")