def test_it_removes_old_users_with_zero_checks(self):
self.charlie.date_joined = self.year_ago
self.charlie.last_login = self.year_ago
self.charlie.save()
Command().handle()
self.assertEqual(User.objects.filter(username="******").count(), 0)
def test_it_leaves_team_members_alone(self):
self.bob.date_joined = self.year_ago
self.bob.last_login = self.year_ago
self.bob.save()
Command().handle()
# Bob belongs to a team so should not get removed
self.assertEqual(User.objects.filter(username="******").count(), 1)
def test_it_removes_old_never_logged_in_users(self):
self.charlie.date_joined = self.year_ago
self.charlie.save()
# Charlie has one demo check
Check(user=self.charlie).save()
Command().handle()
self.assertEqual(User.objects.filter(username="******").count(), 0)
self.assertEqual(Check.objects.count(), 0)
def test_it_removes_old_never_logged_in_users(self):
self.charlie.date_joined = self.year_ago
self.charlie.save()
# Charlie has one demo check
charlies_project = Project.objects.create(owner=self.charlie)
Check(project=charlies_project).save()
Command(stdout=Mock()).handle()
self.assertEqual(User.objects.filter(username="******").count(), 0)
self.assertEqual(Check.objects.count(), 0)