From 519afbe269b671e15a1f1d2aecfe4fc579b61efc Mon Sep 17 00:00:00 2001
From: Leonard Richardson <leonardr@segfault.org>
Date: Sun, 7 Jul 2019 22:57:04 -0400
Subject: A Formatter can now decide how (or whether) to order the attributes  
  inside a tag. [bug=1812422]

---
 bs4/tests/test_tree.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'bs4/tests/test_tree.py')

diff --git a/bs4/tests/test_tree.py b/bs4/tests/test_tree.py
index a14928e..f7c5e2f 100644
--- a/bs4/tests/test_tree.py
+++ b/bs4/tests/test_tree.py
@@ -24,6 +24,7 @@ from bs4.element import (
     CData,
     Comment,
     Declaration,
+    MinimalHTMLFormatter,
     Doctype,
     NavigableString,
     SoupStrainer,
@@ -1683,6 +1684,27 @@ class TestEncoding(SoupTest):
         else:
             self.assertEqual(b'<b>\\u2603</b>', repr(soup))
 
+class TestFormatter(SoupTest):
+
+    def test_sort_attributes(self):
+        class UnsortedFormatter(MinimalHTMLFormatter):
+            def sort_attributes(self, attributes):
+                self.called_with = attributes
+                for k, v in sorted(attributes.items()):
+                    if k == 'ignore':
+                        continue
+                    yield k,v
+
+        soup = self.soup('<p cval="1" aval="2" ignore="ignored"></p>')
+        formatter = UnsortedFormatter()
+        decoded = soup.decode(formatter=formatter)
+
+        # sort_attributes() was called with all three attributes. It removed one and
+        # sorted the other two.
+        self.assertEquals(formatter.called_with, dict(cval="1", aval="2", ignore="ignored"))
+        self.assertEquals(u'<p aval="2" cval="1"></p>', decoded)
+
+
 class TestNavigableStringSubclasses(SoupTest):
 
     def test_cdata(self):
-- 
cgit v1.2.3