Mercurial > public > sg101
diff core/tests/test_ssl_images.py @ 870:ee56c8c8cf98
More WIP on ssl_images management command and tests.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Mon, 15 Dec 2014 21:16:47 -0600 |
parents | 64a5acb83937 |
children | 6900040df0f8 |
line wrap: on
line diff
--- a/core/tests/test_ssl_images.py Mon Dec 15 19:51:28 2014 -0600 +++ b/core/tests/test_ssl_images.py Mon Dec 15 21:16:47 2014 -0600 @@ -1,5 +1,5 @@ """Unit tests for the ssl_images management command.""" - +import re import unittest from core.management.commands.ssl_images import process_post @@ -7,6 +7,27 @@ class ProcessPostTestCase(unittest.TestCase): + SG101_RE = re.compile(r'http://(?:www\.)?surfguitar101.com/', re.I) + def test_empty_string(self): s = process_post('') self.assertEqual(s, '') + + def test_no_matches(self): + test_str = """Here is a post that doesn't contain any image links at + all. It also spans lines. + """ + result = process_post(test_str) + self.assertEqual(test_str, result) + + def test_sg101_images(self): + test_str = """An image: ![image](http://www.surfguitar101.com/img.jpg) + And another: ![pic](HTTP://SURFGUITAR101.COM/foo/bar/img.png). + More stuff here.""" + expected = self.SG101_RE.sub('/', test_str) + result = process_post(test_str) + self.assertNotEqual(test_str, expected) + self.assertEqual(expected, result) + + def test_https_already(self): + pass