diff options
| -rw-r--r-- | patchwork/tests/test_bundles.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/patchwork/tests/test_bundles.py b/patchwork/tests/test_bundles.py index 63f943c..6a74409 100644 --- a/patchwork/tests/test_bundles.py +++ b/patchwork/tests/test_bundles.py @@ -10,6 +10,7 @@ import unittest from django.conf import settings from django.test import TestCase from django.urls import reverse +from django.utils.html import escape from django.utils.http import urlencode from patchwork.models import Bundle @@ -548,8 +549,8 @@ class BundleAddFromListTest(BundleTestBase): 'project_id': self.project.linkname}), params) - self.assertContains(response, 'Patch '%s' already in bundle' - % patch.name, count=1, status_code=200) + expected = escape(f"Patch '{patch.name}' already in bundle") + self.assertContains(response, expected, count=1, status_code=200) self.assertEqual(count, self.bundle.patches.count()) @@ -570,11 +571,12 @@ class BundleAddFromListTest(BundleTestBase): 'project_id': self.project.linkname}), params) - self.assertContains(response, 'Patch '%s' already in bundle' - % patch.name, count=1, status_code=200) - self.assertContains(response, 'Patch '%s' added to bundle' - % self.patches[1].name, count=1, - status_code=200) + for expected in ( + escape(f"Patch '{patch.name}' already in bundle"), + escape(f"Patch '{self.patches[1].name}' added to bundle"), + ): + self.assertContains(response, expected, count=1, status_code=200) + self.assertEqual(count + 1, self.bundle.patches.count()) |