import json
import os
import unittest

import vcr

from athletico.athletico import Athletico


class TestAthletico(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        with open("tests/data/input/sample_data.json") as f:
            cls.tokens = json.load(f)["tokens"]

    @vcr.use_cassette(cassette_library_dir="tests/fixtures")
    def test_extract(self):
        a = Athletico()
        episodes = a.fetch_episodes_for_access_code(self.tokens[0])
        self.assertIn("id", episodes[0])
        self.assertIn("program", episodes[0])
        self.assertIn("program_exercises", episodes[0]["program"])

    @vcr.use_cassette(cassette_library_dir="tests/fixtures")
    def test_save_extraction_to_local(self):
        a = Athletico()
        print(os.path.abspath(os.path.curdir))
        for token in self.tokens:
            episodes = a.fetch_episodes_for_access_code(token)
            a.save_episodes(episodes, destination="tests/data/output")
        pass


if __name__ == "__main__":
    unittest.main()