Test for button class

Author: bgaudino

tests/tests/tests.py

@@ -257,13 +257,16 @@ def test_better_readonly_password_widget(self):
         user = User.objects.get(username='testclient')
         form = UserChangeForm(instance=user)
 
-        self.assertIn(_('*************'), form.as_table())
+        html = form.as_table()
+        self.assertIn(_('*************'), html)
         version = django.VERSION[:2]
 
-        if version in ((4, 2), (5, 0)):
-            self.assertIn('<a href="../../{0}/password/">'.format(user.id), form.as_table())
+        if version < (4, 2):
+            self.assertIn('<a href="../password/"', html)
+        elif version < (5, 1):
+            self.assertIn('<a href="../../{0}/password/">'.format(user.id), html)
         else:
-            self.assertIn('<a href="../password/"', form.as_table())
+            self.assertIn('<a href="../password/" class="button"', html)
 
 
 class UserAdminTest(TestCase):