import json
import unittest
from unittest.mock import patch, mock_open

from swiftstory.cards import Cards


class TestCards(unittest.TestCase):
    def test_get_cards_supported_language(self):
        white_cards = json.dumps([
            "White Card 1", "White Card 2",
            "White Card 3", "White Card 4",
        ])
        black_cards = json.dumps(["Black Card 1", "Black Card 2"])

        with patch("swiftstory.cards.open", mock_open(read_data=white_cards)):
            self.assertIn("White Card 2", Cards.get_white_cards("en"))
        with patch("swiftstory.cards.open", mock_open(read_data=black_cards)):
            self.assertIn("Black Card 1", Cards.get_black_cards("en"))

    def test_get_cards_unknown_language(self):
        with patch("swiftstory.cards.open", side_effect=FileNotFoundError):
            with self.assertRaises(FileNotFoundError):
                Cards.get_white_cards("zz")
            with self.assertRaises(FileNotFoundError):
                Cards.get_black_cards("zz")