From ef113b907fd67d7a384c38eb7b88b191c98c00c2 Mon Sep 17 00:00:00 2001
From: sqozz <sqozz@geekify.de>
Date: Sat, 7 Apr 2018 22:22:10 +0200
Subject: [PATCH] Extend wish-URL test

---
 schort_functional_tests.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/schort_functional_tests.py b/schort_functional_tests.py
index f0ee115..0f730b5 100755
--- a/schort_functional_tests.py
+++ b/schort_functional_tests.py
@@ -32,7 +32,11 @@ class SchortFunctionalTestCase(unittest.TestCase):
 		self.assertRegex(content, ".*\<div.*", msg="Didn't find any opening <div tag in the response.")
 
 	def test_custom_creation(self):
+		'''Test user supplied wish-URLs'''
+		wishId = "custom_user_supplied_url"
 		req = self.assertPostReq(BASE_URL + "/", data={"url" : "https://github.com/sqozz/schort", "wishId" : "custom_user_supplied_url"})
+		short_url = req.text
+		self.assertEqual(short_url, BASE_URL + "/" + wishId)
 
 	def test_easy_api(self):
 		'''