sqozz/schort
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Add test for head requests

Browse source
This commit is contained in:
sqozz 2018-06-09 21:49:19 +02:00
parent 7def769bbb
commit ef2046b610
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/functional_tests.py
View file

@ -17,6 +17,11 @@ class WebTestCase(object):
self.assertEqual(req.status_code, 200, msg="Get request unsuccessful") self.assertEqual(req.status_code, 200, msg="Get request unsuccessful")
return req return req
def assertHeadReq(self, url):
req = requests.head(url)
self.assertEqual(req.status_code, 301, msg="Head request unsuccessful")
return req
def assertGetStatusReq(self, expected_status, url, params = {}): def assertGetStatusReq(self, expected_status, url, params = {}):
req = requests.get(url, params=params, allow_redirects=False) req = requests.get(url, params=params, allow_redirects=False)
self.assertEqual(req.status_code, expected_status, msg="Returned status code does not match the expected one") self.assertEqual(req.status_code, expected_status, msg="Returned status code does not match the expected one")
@ -77,6 +82,11 @@ class SchortShortLinkCase(object):
req = self.assertGetReq(BASE_URL + "/" + self.shortID, params = {"resolve" : ""}) req = self.assertGetReq(BASE_URL + "/" + self.shortID, params = {"resolve" : ""})
self.assertEqual(req.text, self.shortDest) self.assertEqual(req.text, self.shortDest)
def test_head_resolve(self):
"""Test resolving by using a HEAD request"""
req = self.assertHeadReq(BASE_URL + "/" + self.shortID)
self.assertEqual(req.headers.get("Location", ""), self.shortDest)
def test_HTMLresolve(self): def test_HTMLresolve(self):
"""Test HTML displaying of the shortened URL""" """Test HTML displaying of the shortened URL"""
HTML_keyword = "+" HTML_keyword = "+"