def test_addCards_CorrectExpansionsAreIncluded(self): createCards(); game = Game(id = 1, expansionList = "CAHe1") game.addCards() self.assertTrue(game.gamecard_set.filter(card__expansion = 'CAHe1').count() > 0) self.assertTrue(game.gamecard_set.filter(card__expansion = 'CAHe2').count() == 0)
def test_addCards_BaseIsAlwaysIncluded(self): createCards(); game = Game(id = 1, expansionList = "") game.addCards() self.assertTrue(game.gamecard_set.filter(card__expansion = 'Base').count() > 0)