def test_objects_get_removed_after_truncate_command(self):
attacker = Player(name='AttackerTest')
attacker.save()
self.assertEqual(Player.objects.all().count(), 1)
victim = Player(name='VictimTest')
victim.save()
self.assertEqual(Player.objects.all().count(), 2)
game = Game(id=10, fraglimit=50)
game.save()
self.assertEqual(Game.objects.all().count(), 1)
frag = Frag(attacker=attacker,
victim=victim,
game=game,
weapon='MOD_SHOTGUN')
frag.save()
self.assertEqual(Frag.objects.all().count(), 1)
score = Scoreboard(player=attacker, score=12, game=game)
score.save()
self.assertEqual(Scoreboard.objects.all().count(), 1)
call_command('truncate')
self.assertEqual(Frag.objects.all().count(), 0)
self.assertEqual(Player.objects.all().count(), 0)
self.assertEqual(Scoreboard.objects.all().count(), 0)
self.assertEqual(Game.objects.all().count(), 0)
def add_new_game(self, game_id, parser_scoreboard):
Game(id=game_id, fraglimit=parser_scoreboard.get_fraglimit()).save()
print(
"%s: added game (%s, %s)" %
(self.database_prefix, game_id, parser_scoreboard.get_fraglimit()))
self.add_scoreboard(parser_scoreboard.get_scoreboard(), game_id)
