Fix flake8 errors.

This commit is contained in:
Hasan Ramezani 2020-11-28 22:57:28 +01:00
parent 020e7c4da8
commit 4ec3ecdade

View file

@ -44,29 +44,29 @@ class StatusModelTests(TestCase):
accordingly when update_fields is used as an argument
and status_changed is provided
'''
with freeze_time(datetime(2020,1,1)):
with freeze_time(datetime(2020, 1, 1)):
t1 = Status.objects.create()
with freeze_time(datetime(2020,1,2)):
with freeze_time(datetime(2020, 1, 2)):
t1.status = Status.on_hold
t1.save(update_fields=['status', 'status_changed'])
self.assertEqual(t1.status_changed, datetime(2020,1,2))
self.assertEqual(t1.status_changed, datetime(2020, 1, 2))
def test_save_with_update_fields_overrides_status_changed_not_provided(self):
'''
Tests if the save method updated status_changed field
accordingly when update_fields is used as an argument
with status and status_changed is not provided
'''
with freeze_time(datetime(2020,1,1)):
with freeze_time(datetime(2020, 1, 1)):
t1 = Status.objects.create()
with freeze_time(datetime(2020,1,2)):
with freeze_time(datetime(2020, 1, 2)):
t1.status = Status.on_hold
t1.save(update_fields=['status'])
self.assertEqual(t1.status_changed, datetime(2020,1,2))
self.assertEqual(t1.status_changed, datetime(2020, 1, 2))
class StatusModelPlainTupleTests(StatusModelTests):