Fix tests

This commit is contained in:
Agrendalath
2019-11-13 18:31:00 +01:00
parent a60efa2941
commit de7b8d4216
2 changed files with 6 additions and 8 deletions

View File

@@ -917,11 +917,11 @@ class LoginFailures(models.Model):
def __str__(self):
"""Str -> Username: count - date."""
return six.text_type('{username}: {count} - {date}'.format(
return u'{username}: {count} - {date}'.format(
username=self.user.username,
count=self.failure_count,
date=self.lockout_until.isoformat() if self.lockout_until else '-'
))
)
class Meta:
verbose_name = 'Login Failure'

View File

@@ -1,3 +1,4 @@
# coding=UTF-8
"""
Tests student admin.py
"""
@@ -324,7 +325,7 @@ class LoginFailuresAdminTest(TestCase):
"""Setup."""
super(LoginFailuresAdminTest, self).setUp()
self.client.login(username=self.user.username, password='test')
self.user2 = UserFactory.create()
self.user2 = UserFactory.create(username=u'Zażółć gęślą jaźń')
LoginFailures.objects.create(user=self.user, failure_count=10, lockout_until=datetime.datetime.now())
LoginFailures.objects.create(user=self.user2, failure_count=2)
@@ -338,11 +339,8 @@ class LoginFailuresAdminTest(TestCase):
Test if `__str__` method behaves correctly for unicode username.
It shouldn't raise `TypeError`.
"""
try:
str(LoginFailures.objects.get(user=self.user))
str(LoginFailures.objects.get(user=self.user2))
except TypeError as e:
self.fail("Failed executing `__str__` with unicode: {0}".format(e))
str(LoginFailures.objects.get(user=self.user))
str(LoginFailures.objects.get(user=self.user2))
@ddt.data(
reverse('admin:student_loginfailures_changelist'),