Mercurial > public > sg101

annotate core/tests/test_ssl_images.py @ 989:2908859c2fe4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Smilies now use relative links. This is for upcoming switch to SSL. Currently we do not need absolute URLs for smilies. If this changes we can add it later.
author Brian Neal <bgneal@gmail.com>
date Thu, 29 Oct 2015 20:54:34 -0500
parents 76525f5ac2b1
children
rev   line source
bgneal@868 1 """Unit tests for the ssl_images management command."""
bgneal@870 2 import re
bgneal@868 3 import unittest
bgneal@889 4 from urlparse import urlparse
bgneal@868 5
bgneal@872 6 import mock
bgneal@963 7 from django.conf import settings
bgneal@872 8
bgneal@894 9 from core.management.commands.ssl_images import html_check
bgneal@987 10 from core.management.commands.ssl_images import process_post, process_html
bgneal@889 11 import core.management.commands.ssl_images
bgneal@868 12
bgneal@868 13
bgneal@868 14 class ProcessPostTestCase(unittest.TestCase):
bgneal@868 15
bgneal@870 16 SG101_RE = re.compile(r'http://(?:www\.)?surfguitar101.com/', re.I)
bgneal@870 17
bgneal@963 18 def setUp(self):
bgneal@963 19 self.assertTrue(len(settings.USER_IMAGES_SOURCES) > 0)
bgneal@963 20 self.safe_host = settings.USER_IMAGES_SOURCES[0]
bgneal@963 21
bgneal@889 22 def tearDown(self):
bgneal@889 23 core.management.commands.ssl_images.url_cache = {}
bgneal@889 24
bgneal@868 25 def test_empty_string(self):
bgneal@868 26 s = process_post('')
bgneal@868 27 self.assertEqual(s, '')
bgneal@870 28
bgneal@870 29 def test_no_matches(self):
bgneal@870 30 test_str = """Here is a post that doesn't contain any image links at
bgneal@870 31 all. It also spans lines.
bgneal@870 32 """
bgneal@870 33 result = process_post(test_str)
bgneal@870 34 self.assertEqual(test_str, result)
bgneal@870 35
bgneal@870 36 def test_sg101_images(self):
bgneal@870 37 test_str = """An image: ![image](http://www.surfguitar101.com/img.jpg)
bgneal@870 38 And another: ![pic](HTTP://SURFGUITAR101.COM/foo/bar/img.png).
bgneal@870 39 More stuff here."""
bgneal@870 40 expected = self.SG101_RE.sub('/', test_str)
bgneal@870 41 result = process_post(test_str)
bgneal@870 42 self.assertNotEqual(test_str, expected)
bgneal@870 43 self.assertEqual(expected, result)
bgneal@870 44
bgneal@871 45 def test_sg101_with_newlines(self):
bgneal@871 46 test_str = """An image: ![image](
bgneal@871 47 http://surfguitar101.com/media/zzz.jpg
bgneal@871 48 )
bgneal@871 49 with trailing text."""
bgneal@871 50 expected = """An image: ![image](/media/zzz.jpg)
bgneal@871 51 with trailing text."""
bgneal@871 52 result = process_post(test_str)
bgneal@871 53 self.assertNotEqual(test_str, expected)
bgneal@871 54 self.assertEqual(expected, result)
bgneal@871 55
bgneal@870 56 def test_https_already(self):
bgneal@871 57 test_str = """An image that is already using https:
bgneal@963 58 ![flyer](https://{}/zzz.png)
bgneal@871 59 It's cool.
bgneal@963 60 """.format(self.safe_host)
bgneal@871 61 result = process_post(test_str)
bgneal@871 62 self.assertEqual(test_str, result)
bgneal@871 63
bgneal@871 64 def test_https_sg101(self):
bgneal@871 65 test_str = """An image that is already using https:
bgneal@871 66 ![flyer](https://www.SURFGUITAR101.com/zzz.png)
bgneal@871 67 It's cool.
bgneal@871 68 """
bgneal@871 69 expected = """An image that is already using https:
bgneal@871 70 ![flyer](/zzz.png)
bgneal@871 71 It's cool.
bgneal@871 72 """
bgneal@871 73 result = process_post(test_str)
bgneal@871 74 self.assertEqual(expected, result)
bgneal@871 75
bgneal@871 76 def test_multiple_non_http(self):
bgneal@871 77 test_str = """An image: ![image](http://www.surfguitar101.com/img.jpg)
bgneal@963 78 And another: ![pic](HTTPS://{}/foo/bar/img.png).
bgneal@963 79 More stuff here.""".format(self.safe_host)
bgneal@871 80 expected = """An image: ![image](/img.jpg)
bgneal@963 81 And another: ![pic](HTTPS://{}/foo/bar/img.png).
bgneal@963 82 More stuff here.""".format(self.safe_host)
bgneal@871 83 result = process_post(test_str)
bgneal@871 84 self.assertEqual(expected, result)
bgneal@871 85
bgneal@871 86 def test_https_already_with_title(self):
bgneal@871 87 test_str = """An image that is already using https:
bgneal@963 88 ![flyer](https://{}/zzz.png "the title")
bgneal@871 89 It's cool.
bgneal@963 90 """.format(self.safe_host)
bgneal@871 91 result = process_post(test_str)
bgneal@871 92 self.assertEqual(test_str, result)
bgneal@871 93
bgneal@871 94 def test_sg101_with_title(self):
bgneal@871 95 test_str = """An image on SG101:
bgneal@871 96 ![flyer](http://surfguitar101.com/zzz.png "the title")
bgneal@871 97 It's cool.
bgneal@871 98 """
bgneal@871 99 expected = """An image on SG101:
bgneal@871 100 ![flyer](/zzz.png "the title")
bgneal@871 101 It's cool.
bgneal@871 102 """
bgneal@871 103 result = process_post(test_str)
bgneal@871 104 self.assertEqual(expected, result)
bgneal@871 105
bgneal@871 106 def test_https_sg101_brackets(self):
bgneal@871 107 test_str = """An image that is already using https:
bgneal@871 108 ![flyer](<https://www.SURFGUITAR101.com/zzz.png>)
bgneal@871 109 It's cool.
bgneal@871 110 """
bgneal@871 111 expected = """An image that is already using https:
bgneal@871 112 ![flyer](/zzz.png)
bgneal@871 113 It's cool.
bgneal@871 114 """
bgneal@871 115 result = process_post(test_str)
bgneal@871 116 self.assertEqual(expected, result)
bgneal@871 117
bgneal@871 118 def test_https_already_brackets(self):
bgneal@871 119 test_str = """An image that is already using https:
bgneal@963 120 ![flyer](<https://{}/zzz.png>)
bgneal@871 121 It's cool.
bgneal@963 122 """.format(self.safe_host)
bgneal@871 123 expected = """An image that is already using https:
bgneal@963 124 ![flyer](https://{}/zzz.png)
bgneal@871 125 It's cool.
bgneal@963 126 """.format(self.safe_host)
bgneal@871 127 result = process_post(test_str)
bgneal@871 128 self.assertEqual(expected, result)
bgneal@872 129
bgneal@872 130 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@872 131 def test_simple_replacement(self, upload_mock):
bgneal@872 132 old_src = 'http://example.com/images/my_image.jpg'
bgneal@873 133 new_src = 'https://cloud.com/ABCDEF.jpg'
bgneal@872 134 test_str = """Here is a really cool http: based image:
bgneal@872 135 ![flyer]({})
bgneal@872 136 Cool, right?""".format(old_src)
bgneal@872 137 expected = """Here is a really cool http: based image:
bgneal@872 138 ![flyer]({})
bgneal@872 139 Cool, right?""".format(new_src)
bgneal@872 140
bgneal@872 141 upload_mock.return_value = new_src
bgneal@872 142 result = process_post(test_str)
bgneal@872 143 self.assertEqual(expected, result)
bgneal@927 144 upload_mock.assert_called_once_with(urlparse(old_src))
bgneal@873 145
bgneal@873 146 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@873 147 def test_multiple_replacement(self, upload_mock):
bgneal@873 148 old_src = [
bgneal@873 149 'http://example.com/images/my_image.jpg',
bgneal@873 150 'http://example.com/static/wow.gif',
bgneal@873 151 'http://example.com/media/a/b/c/pic.png',
bgneal@873 152 ]
bgneal@873 153 new_src = [
bgneal@873 154 'https://cloud.com/some/path/012345.jpg',
bgneal@873 155 'https://cloud.com/some/path/6789AB.gif',
bgneal@873 156 'https://cloud.com/some/path/CDEF01.png',
bgneal@873 157 ]
bgneal@873 158
bgneal@873 159 template = """Here is a really cool http: based image:
bgneal@873 160 ![flyer]({})
bgneal@873 161 Cool, right?
bgneal@873 162 Another one: ![pic]({})
bgneal@873 163 And finally
bgneal@873 164 ![an image]({})
bgneal@873 165 """
bgneal@873 166
bgneal@873 167 test_str = template.format(*old_src)
bgneal@873 168 expected = template.format(*new_src)
bgneal@873 169
bgneal@873 170 upload_mock.side_effect = new_src
bgneal@873 171 result = process_post(test_str)
bgneal@873 172 self.assertEqual(expected, result)
bgneal@927 173 expected_args = [mock.call(urlparse(c)) for c in old_src]
bgneal@873 174 self.assertEqual(upload_mock.call_args_list, expected_args)
bgneal@873 175
bgneal@873 176 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@873 177 def test_multiple_replacement_2(self, upload_mock):
bgneal@873 178 old_src = [
bgneal@873 179 'http://example.com/images/my_image.jpg',
bgneal@963 180 'https://{}/static/wow.gif'.format(self.safe_host),
bgneal@873 181 'http://www.surfguitar101.com/media/a/b/c/pic.png',
bgneal@873 182 'http://surfguitar101.com/media/a/b/c/pic2.png',
bgneal@873 183 ]
bgneal@873 184 new_src = [
bgneal@873 185 'https://cloud.com/some/path/012345.jpg',
bgneal@963 186 'https://{}/static/wow.gif'.format(self.safe_host),
bgneal@873 187 '/media/a/b/c/pic.png',
bgneal@873 188 '/media/a/b/c/pic2.png',
bgneal@873 189 ]
bgneal@873 190
bgneal@873 191 template = """Here is a really cool http: based image:
bgneal@873 192 ![flyer]({})
bgneal@873 193 Cool, right?
bgneal@873 194 Another two: ![pic]({}) ![photo]({})
bgneal@873 195 And finally
bgneal@873 196 ![an image]({}).
bgneal@873 197 """
bgneal@873 198
bgneal@873 199 test_str = template.format(*old_src)
bgneal@873 200 expected = template.format(*new_src)
bgneal@873 201
bgneal@873 202 upload_mock.side_effect = new_src
bgneal@873 203 result = process_post(test_str)
bgneal@873 204 self.assertEqual(expected, result)
bgneal@927 205 upload_mock.assert_called_once_with(urlparse(old_src[0]))
bgneal@889 206
bgneal@889 207 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@889 208 def test_caching(self, upload_mock):
bgneal@889 209 old_src = [
bgneal@889 210 'http://example.com/images/my_image.jpg',
bgneal@889 211 'http://example.com/static/wow.gif',
bgneal@889 212 'http://example.com/images/my_image.jpg',
bgneal@889 213 ]
bgneal@889 214 new_src = [
bgneal@889 215 'https://cloud.com/some/path/012345.jpg',
bgneal@889 216 'https://cloud.com/some/path/6789AB.gif',
bgneal@889 217 'https://cloud.com/some/path/012345.jpg',
bgneal@889 218 ]
bgneal@889 219
bgneal@889 220 template = """Here is a really cool http: based image:
bgneal@889 221 ![flyer]({})
bgneal@889 222 Cool, right?
bgneal@889 223 Another one: ![pic]({})
bgneal@889 224 And finally
bgneal@889 225 ![an image]({})
bgneal@889 226 """
bgneal@889 227
bgneal@889 228 test_str = template.format(*old_src)
bgneal@889 229 expected = template.format(*new_src)
bgneal@889 230
bgneal@889 231 upload_mock.side_effect = new_src
bgneal@889 232 result = process_post(test_str)
bgneal@889 233 self.assertEqual(expected, result)
bgneal@927 234 expected_args = [mock.call(urlparse(c)) for c in old_src[:2]]
bgneal@889 235 self.assertEqual(upload_mock.call_args_list, expected_args)
bgneal@889 236
bgneal@986 237 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@986 238 def test_bad_image_url(self, upload_mock):
bgneal@986 239 test_str = u'![image](http://[url=http://www.flickr.com/photos/85447101@N04/8025176073/][img]http://farm9.staticflickr.com/8456/8025176073_91ac0c3d18_b.jpg[/img][/url] [url=http://www.flickr.com/photos/85447101@N04/8025176073/]IMG_0398[/url] by [url=http://www.flickr.com/people/85447101@N04/]friendshipmaster[/url], on Flick)'
bgneal@986 240 expected = u'{bad image}'
bgneal@986 241
bgneal@986 242 result = process_post(test_str)
bgneal@986 243 self.assertEqual(expected, result)
bgneal@986 244 self.assertEqual(upload_mock.call_count, 0)
bgneal@986 245
bgneal@894 246
bgneal@894 247 class HtmlCheckTestCase(unittest.TestCase):
bgneal@894 248
bgneal@894 249 def test_empty(self):
bgneal@894 250 self.assertFalse(html_check(''))
bgneal@894 251
bgneal@894 252 def test_no_images(self):
bgneal@894 253 self.assertFalse(html_check('<p>Hi there!</p>'))
bgneal@894 254 self.assertFalse(html_check('<p>Hi <b>there</b>!</p>'))
bgneal@894 255
bgneal@894 256 def test_safe_image(self):
bgneal@894 257 self.assertFalse(html_check('<img src="https://a.jpg" />'))
bgneal@894 258 self.assertFalse(html_check('<img src="" alt="stuff" />'))
bgneal@894 259 self.assertFalse(html_check('<img src="HTTPS://a.jpg" />'))
bgneal@894 260 self.assertFalse(html_check("""
bgneal@894 261 <div>
bgneal@894 262 <p>Look: <img src="https://a.jpg" alt="a" /></p>
bgneal@894 263 <p>Look again: <img src="https://b.jpg" alt="b" /></p>
bgneal@894 264 </div>
bgneal@894 265 """))
bgneal@894 266
bgneal@894 267 def test_one_image(self):
bgneal@894 268 self.assertTrue(html_check('<img src="http://a.jpg" alt="a" />'))
bgneal@894 269 self.assertTrue(html_check(
bgneal@894 270 '<p>Look: <img src="http://a.jpg" alt="a" /></p>'))
bgneal@894 271
bgneal@894 272 def test_two_images(self):
bgneal@894 273 self.assertTrue(html_check("""
bgneal@894 274 <p>Look: <img src="https://a.jpg" alt="a" /></p>
bgneal@894 275 <p>Look again: <img src="http://b.jpg" alt="b" /></p>
bgneal@894 276 """))
bgneal@894 277 self.assertTrue(html_check("""
bgneal@894 278 <p>Look: <img src="http://a.jpg" alt="a" /></p>
bgneal@894 279 <p>Look again: <img src="http://b.jpg" alt="b" /></p>
bgneal@894 280 """))
bgneal@894 281 self.assertTrue(html_check("""
bgneal@894 282 <div>
bgneal@894 283 <p>Look: <img src="http://a.jpg" alt="a" /></p>
bgneal@894 284 <p>Look again: <img src="http://b.jpg" alt="b" /></p>
bgneal@894 285 </div>
bgneal@894 286 """))
bgneal@894 287 self.assertTrue(html_check("""
bgneal@894 288 <div>
bgneal@894 289 <p>Look: <img src="http://a.jpg" alt="a" /></p>
bgneal@894 290 <p>Look again: <img src="https://b.jpg" alt="b" /></p>
bgneal@894 291 </div>
bgneal@894 292 """))
bgneal@987 293
bgneal@987 294
bgneal@987 295 class ProcessHtmlTestCase(unittest.TestCase):
bgneal@987 296
bgneal@987 297 SG101_RE = re.compile(r'http://(?:www\.)?surfguitar101.com/', re.I)
bgneal@987 298
bgneal@987 299 def setUp(self):
bgneal@987 300 self.assertTrue(len(settings.USER_IMAGES_SOURCES) > 0)
bgneal@987 301 self.safe_host = settings.USER_IMAGES_SOURCES[0]
bgneal@987 302
bgneal@987 303 def tearDown(self):
bgneal@987 304 core.management.commands.ssl_images.url_cache = {}
bgneal@987 305
bgneal@987 306 def test_empty_string(self):
bgneal@987 307 s = process_html('')
bgneal@987 308 self.assertEqual(s, '')
bgneal@987 309
bgneal@987 310 def test_whitespace_string(self):
bgneal@987 311 s = process_html('\r\n\r\n')
bgneal@987 312 self.assertEqual(s, '')
bgneal@987 313
bgneal@987 314 def test_no_matches(self):
bgneal@987 315 test_str = """<p>Here is a post that doesn't contain any image links at
bgneal@987 316 all. It also spans lines.</p>
bgneal@987 317 """
bgneal@987 318 result = process_html(test_str)
bgneal@987 319 self.assertEqual(test_str, result)
bgneal@987 320
bgneal@987 321 def test_multiple_paragraphs(self):
bgneal@987 322 test_str = """<p>Here is a post that doesn't contain any image links at
bgneal@987 323 all. It also spans lines.</p>
bgneal@987 324 """
bgneal@987 325 test_str += test_str
bgneal@987 326 result = process_html(test_str)
bgneal@987 327 self.assertEqual(test_str, result)
bgneal@987 328
bgneal@987 329 def test_sg101_images(self):
bgneal@987 330 test_str = """<p>An image:
bgneal@987 331 <img src="http://www.surfguitar101.com/img.jpg" alt="image">
bgneal@987 332 And another: <img src="HTTP://SURFGUITAR101.COM/foo/bar/img.png" alt="pic">
bgneal@987 333 More stuff here.</p>"""
bgneal@987 334 expected = self.SG101_RE.sub('/', test_str)
bgneal@987 335 result = process_html(test_str)
bgneal@987 336 self.assertNotEqual(test_str, expected)
bgneal@987 337 self.assertEqual(expected, result)
bgneal@987 338
bgneal@987 339 def test_https_already(self):
bgneal@987 340 test_str = """<p>An image that is already using https:
bgneal@987 341 <img src="https://{}/zzz.png" alt="pic">
bgneal@987 342 It's cool.</p>""".format(self.safe_host)
bgneal@987 343 result = process_html(test_str)
bgneal@987 344 self.assertEqual(test_str, result)
bgneal@987 345
bgneal@987 346 def test_https_sg101(self):
bgneal@987 347 test_str = """<p>An image that is already using https:
bgneal@987 348 <img src="https://www.SURFGUITAR101.com/zzz.png" alt="pic">
bgneal@987 349 It's cool.</p>
bgneal@987 350 """
bgneal@987 351 expected = """<p>An image that is already using https:
bgneal@987 352 <img src="/zzz.png" alt="pic">
bgneal@987 353 It's cool.</p>"""
bgneal@987 354 result = process_html(test_str)
bgneal@987 355 self.assertEqual(expected, result)
bgneal@987 356
bgneal@987 357 def test_multiple_non_http(self):
bgneal@987 358 test_str = """<p>An image:
bgneal@987 359 <img src="http://www.surfguitar101.com/img.jpg" alt="pic">
bgneal@987 360 And another:
bgneal@987 361 <img src="HTTPS://{}/foo/bar/img.png" alt="stuff">
bgneal@987 362 More stuff here.</p>
bgneal@987 363 """.format(self.safe_host)
bgneal@987 364 expected = """<p>An image:
bgneal@987 365 <img src="/img.jpg" alt="pic">
bgneal@987 366 And another:
bgneal@987 367 <img src="HTTPS://{}/foo/bar/img.png" alt="stuff">
bgneal@987 368 More stuff here.</p>""".format(self.safe_host)
bgneal@987 369 result = process_html(test_str)
bgneal@987 370 self.assertEqual(expected, result)
bgneal@987 371
bgneal@987 372 def test_https_already_with_title(self):
bgneal@987 373 test_str = """<p>An image that is already using https:
bgneal@987 374 <img src="https://{}/zzz.png" alt="1" title="the title">
bgneal@987 375 It's cool.</p>
bgneal@987 376 """.format(self.safe_host)
bgneal@987 377 result = process_html(test_str)
bgneal@987 378 self.assertEqual(test_str, result)
bgneal@987 379
bgneal@987 380 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@987 381 def test_simple_replacement(self, upload_mock):
bgneal@987 382 old_src = 'http://example.com/images/my_image.jpg'
bgneal@987 383 new_src = 'https://cloud.com/ABCDEF.jpg'
bgneal@987 384 test_str = """<p>Here is a really cool http: based image:
bgneal@987 385 <img src="{}" alt="a">
bgneal@987 386 Cool, right?</p>""".format(old_src)
bgneal@987 387 expected = """<p>Here is a really cool http: based image:
bgneal@987 388 <img src="{}" alt="a">
bgneal@987 389 Cool, right?</p>""".format(new_src)
bgneal@987 390
bgneal@987 391 upload_mock.return_value = new_src
bgneal@987 392 result = process_html(test_str)
bgneal@987 393 self.assertEqual(expected, result)
bgneal@987 394 upload_mock.assert_called_once_with(urlparse(old_src))
bgneal@987 395
bgneal@987 396 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@987 397 def test_multiple_replacement(self, upload_mock):
bgneal@987 398 old_src = [
bgneal@987 399 'http://example.com/images/my_image.jpg',
bgneal@987 400 'http://example.com/static/wow.gif',
bgneal@987 401 'http://example.com/media/a/b/c/pic.png',
bgneal@987 402 ]
bgneal@987 403 new_src = [
bgneal@987 404 'https://cloud.com/some/path/012345.jpg',
bgneal@987 405 'https://cloud.com/some/path/6789AB.gif',
bgneal@987 406 'https://cloud.com/some/path/CDEF01.png',
bgneal@987 407 ]
bgneal@987 408
bgneal@987 409 template = """<p>Here is a really cool http: based image:
bgneal@987 410 <img src="{}" alt="a">
bgneal@987 411 Cool, right?
bgneal@987 412 Another one: <img src="{}" alt="b">
bgneal@987 413 And finally
bgneal@987 414 <img src="{}" alt="c">
bgneal@987 415 </p>"""
bgneal@987 416
bgneal@987 417 test_str = template.format(*old_src)
bgneal@987 418 expected = template.format(*new_src)
bgneal@987 419
bgneal@987 420 upload_mock.side_effect = new_src
bgneal@987 421 result = process_html(test_str)
bgneal@987 422 self.assertEqual(expected, result)
bgneal@987 423 expected_args = [mock.call(urlparse(c)) for c in old_src]
bgneal@987 424 self.assertEqual(upload_mock.call_args_list, expected_args)
bgneal@987 425
bgneal@987 426 @mock.patch('core.management.commands.ssl_images.save_image_to_cloud')
bgneal@987 427 def test_multiple_replacement_2(self, upload_mock):
bgneal@987 428 old_src = [
bgneal@987 429 'http://example.com/images/my_image.jpg',
bgneal@987 430 'https://{}/static/wow.gif'.format(self.safe_host),
bgneal@987 431 'http://www.surfguitar101.com/media/a/b/c/pic.png',
bgneal@987 432 'http://surfguitar101.com/media/a/b/c/pic2.png',
bgneal@987 433 ]
bgneal@987 434 new_src = [
bgneal@987 435 'https://cloud.com/some/path/012345.jpg',
bgneal@987 436 'https://{}/static/wow.gif'.format(self.safe_host),
bgneal@987 437 '/media/a/b/c/pic.png',
bgneal@987 438 '/media/a/b/c/pic2.png',
bgneal@987 439 ]
bgneal@987 440
bgneal@987 441 template = """<p>Here is a really cool http: based image:
bgneal@987 442 <img src="{}" alt="a">
bgneal@987 443 Cool, right?
bgneal@987 444 Another two: <img src="{}" alt="b"><img src="{}" alt="c">
bgneal@987 445 And finally
bgneal@987 446 <img src="{}" alt="d"></p>"""
bgneal@987 447
bgneal@987 448 test_str = template.format(*old_src)
bgneal@987 449 expected = template.format(*new_src)
bgneal@987 450
bgneal@987 451 upload_mock.side_effect = new_src
bgneal@987 452 result = process_html(test_str)
bgneal@987 453 self.assertEqual(expected, result)
bgneal@987 454 upload_mock.assert_called_once_with(urlparse(old_src[0]))
bgneal@987 455
bgneal@987 456 @mock.patch('core.management.commands.ssl_images.convert_to_ssl')
bgneal@987 457 def test_change_img_to_a(self, convert_mock):
bgneal@987 458 convert_mock.return_value = None
bgneal@987 459 test_str = """<p>A bad image:
bgneal@987 460 <img src="http://example.com/zzz.png" alt="1" title="the title">
bgneal@987 461 It's cool.</p>"""
bgneal@987 462
bgneal@987 463 result = process_html(test_str)
bgneal@987 464
bgneal@987 465 expected = """<p>A bad image:
bgneal@987 466 <a href="http://example.com/zzz.png">Image</a>
bgneal@987 467 It's cool.</p>"""
bgneal@987 468 self.assertEqual(result, expected)