import unittest

from swiftstory.Cards import Cards


class TestCards(unittest.TestCase):
    def test_get_cards_fr(self):
        self.assertTrue(Cards.get_white_cards("fr"))
        self.assertTrue(Cards.get_black_cards("fr"))

    def test_get_cards_en(self):
        self.assertTrue(Cards.get_white_cards("en"))
        self.assertTrue(Cards.get_black_cards("en"))

    def test_get_cards_unknown_language(self):
        try:
            Cards.get_white_cards("zz")
        except FileNotFoundError:
            pass
        else:
            self.fail("Found zz white cards.")
        try:
            Cards.get_black_cards("zz")
        except FileNotFoundError:
            pass
        else:
            self.fail("Found zz black cards.")