From a7610ec7be1ee7a82e26f98fa702dc25554e71e2 Mon Sep 17 00:00:00 2001 From: Karl Hobley Date: Wed, 16 Jul 2014 16:19:15 +0100 Subject: [PATCH] Added tests for signature generation --- wagtail/wagtailimages/tests.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/wagtail/wagtailimages/tests.py b/wagtail/wagtailimages/tests.py index 09bdd3e76..e26fcc500 100644 --- a/wagtail/wagtailimages/tests.py +++ b/wagtail/wagtailimages/tests.py @@ -17,7 +17,7 @@ from wagtail.wagtailimages.formats import ( from wagtail.wagtailimages.backends import get_image_backend from wagtail.wagtailimages.backends.pillow import PillowBackend -from wagtail.wagtailimages.utils import parse_filter_spec, InvalidFilterSpecError +from wagtail.wagtailimages.utils import parse_filter_spec, InvalidFilterSpecError, generate_signature, verify_signature def get_test_image_file(): @@ -498,3 +498,17 @@ class TestFilterSpecParsing(TestCase): def test_bad(self): for filter_spec in self.bad: self.assertRaises(InvalidFilterSpecError, parse_filter_spec, filter_spec) + + +class TestSignatureGeneration(TestCase): + def test_signature_generation(self): + self.assertEqual(generate_signature(100, 'fill-800x600'), b'xnZOzQyUg6pkfciqcfRJRosOrGg=') + + def test_signature_verification(self): + self.assertTrue(verify_signature(b'xnZOzQyUg6pkfciqcfRJRosOrGg=', 100, 'fill-800x600')) + + def test_signature_changes_on_image_id(self): + self.assertFalse(verify_signature(b'xnZOzQyUg6pkfciqcfRJRosOrGg=', 200, 'fill-800x600')) + + def test_signature_changes_on_filter_spec(self): + self.assertFalse(verify_signature(b'xnZOzQyUg6pkfciqcfRJRosOrGg=', 100, 'fill-800x700'))