From ef2046b610ca5acb5530798c177bb8ccb5a2d3dd Mon Sep 17 00:00:00 2001
From: sqozz <sqozz@geekify.de>
Date: Sat, 9 Jun 2018 21:49:19 +0200
Subject: [PATCH] Add test for head requests

---
 tests/functional_tests.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/functional_tests.py b/tests/functional_tests.py
index 1c04079..5bf0d80 100755
--- a/tests/functional_tests.py
+++ b/tests/functional_tests.py
@@ -17,6 +17,11 @@ class WebTestCase(object):
 		self.assertEqual(req.status_code, 200, msg="Get request unsuccessful")
 		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 = {}):
 		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")
@@ -77,6 +82,11 @@ class SchortShortLinkCase(object):
 		req = self.assertGetReq(BASE_URL + "/" + self.shortID, params = {"resolve" : ""})
 		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):
 		"""Test HTML displaying of the shortened URL"""
 		HTML_keyword = "+"