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)