diff options
Diffstat (limited to 'ishtar_common')
| -rw-r--r-- | ishtar_common/tests.py | 31 |
1 files changed, 16 insertions, 15 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index 8708d90bd..2bd4afef1 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -217,21 +217,22 @@ class UtilsTest(TestCase): 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) + for idx in range(65): # > 62 to test with 2 letters + base_url = "https://example.com#{}".format(idx) + 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): |