Fixed #23883 -- Stopped flatatt modifying its argument

This commit is contained in:
Tim Heap 2014-11-21 12:35:58 +11:00 committed by Claude Paroz
parent d8f00e1918
commit 5b17dcd8ef
2 changed files with 25 additions and 7 deletions

View File

@ -29,16 +29,17 @@ def flatatt(attrs):
The result is passed through 'mark_safe'. The result is passed through 'mark_safe'.
""" """
key_value_attrs = []
boolean_attrs = [] boolean_attrs = []
for attr, value in list(attrs.items()): for attr, value in attrs.items():
if value is True: if isinstance(value, bool):
if value:
boolean_attrs.append((attr,)) boolean_attrs.append((attr,))
del attrs[attr] else:
elif value is False: key_value_attrs.append((attr, value))
del attrs[attr]
return ( return (
format_html_join('', ' {0}="{1}"', sorted(attrs.items())) + format_html_join('', ' {0}="{1}"', sorted(key_value_attrs)) +
format_html_join('', ' {0}', sorted(boolean_attrs)) format_html_join('', ' {0}', sorted(boolean_attrs))
) )

View File

@ -27,6 +27,23 @@ class FormsUtilTestCase(TestCase):
self.assertEqual(flatatt({'class': "news", 'title': "Read this", 'required': False}), ' class="news" title="Read this"') self.assertEqual(flatatt({'class': "news", 'title': "Read this", 'required': False}), ' class="news" title="Read this"')
self.assertEqual(flatatt({}), '') self.assertEqual(flatatt({}), '')
def test_flatatt_no_side_effects(self):
"""
Fixes #23883 -- Check that flatatt does not modify the dict passed in
"""
attrs = {'foo': 'bar', 'true': True, 'false': False}
attrs_copy = copy.copy(attrs)
self.assertEqual(attrs, attrs_copy)
first_run = flatatt(attrs)
self.assertEqual(attrs, attrs_copy)
self.assertEqual(first_run, ' foo="bar" true')
second_run = flatatt(attrs)
self.assertEqual(attrs, attrs_copy)
self.assertEqual(first_run, second_run)
def test_validation_error(self): def test_validation_error(self):
################### ###################
# ValidationError # # ValidationError #