Manage tiny urls for QR codes

1 files changed, 17 insertions, 0 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 01aba1f5b..0f00322fe 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -155,6 +155,23 @@ class UtilsTest(TestCase):
                              u'data__old__traitement__constat_etat')
         self.assertEqual(res, expected)
 
+    def test_tinyfy_url(self):
+        base_url = "https://example.com"
+        tiny_url = models.TinyUrl()
+        tiny_url.link = base_url
+        tiny_url.save()
+
+        short_id = tiny_url.get_short_id()
+        db_id = models.TinyUrl.decode_id(short_id)
+        self.assertEqual(tiny_url.pk, db_id)
+
+        ty = models.TinyUrl.objects.get(id=db_id)
+        self.assertEqual(base_url, ty.link)
+        c = Client()
+        response = c.get(reverse('tiny-redirect', args=[short_id]))
+        self.assertEqual(response['Location'], base_url)
+        self.assertEqual(response.status_code, 302)
+
 
 class CommandsTestCase(TestCase):
     fixtures = [settings.ROOT_PATH +