import sys
import os

# getting the name of the directory
# where the this file is present.
current = os.path.dirname(os.path.realpath(__file__))

# Getting the parent directory name
# where the current directory is present.
parent = os.path.dirname(current)

# adding the parent directory to
# the sys.path.
sys.path.append(parent)

from src.watched import cleanup_watched, combine_watched_dicts

tv_shows_watched_list_1 = {
    frozenset(
        {
            ("tvdb", "75710"),
            ("title", "Criminal Minds"),
            ("imdb", "tt0452046"),
            ("locations", ("Criminal Minds",)),
            ("tmdb", "4057"),
        }
    ): {
        "Season 1": [
            {
                "imdb": "tt0550489",
                "tmdb": "282843",
                "tvdb": "176357",
                "title": "Extreme Aggressor",
                "locations": (
                    "Criminal Minds S01E01 Extreme Aggressor WEBDL-720p.mkv",
                ),
                "status": {"completed": True, "time": 0},
            },
            {
                "imdb": "tt0550487",
                "tmdb": "282861",
                "tvdb": "300385",
                "title": "Compulsion",
                "locations": ("Criminal Minds S01E02 Compulsion WEBDL-720p.mkv",),
                "status": {"completed": True, "time": 0},
            },
        ]
    },
    frozenset({("title", "Test"), ("locations", ("Test",))}): {
        "Season 1": [
            {
                "title": "S01E01",
                "locations": ("Test S01E01.mkv",),
                "status": {"completed": True, "time": 0},
            },
            {
                "title": "S01E02",
                "locations": ("Test S01E02.mkv",),
                "status": {"completed": True, "time": 0},
            },
            {
                "title": "S01E04",
                "locations": ("Test S01E04.mkv",),
                "status": {"completed": False, "time": 5},
            },
        ]
    },
}

tv_shows_watched_list_2 = {
    frozenset(
        {
            ("tvdb", "75710"),
            ("title", "Criminal Minds"),
            ("imdb", "tt0452046"),
            ("locations", ("Criminal Minds",)),
            ("tmdb", "4057"),
        }
    ): {
        "Season 1": [
            {
                "imdb": "tt0550487",
                "tmdb": "282861",
                "tvdb": "300385",
                "title": "Compulsion",
                "locations": ("Criminal Minds S01E02 Compulsion WEBDL-720p.mkv",),
                "status": {"completed": True, "time": 0},
            },
            {
                "imdb": "tt0550498",
                "tmdb": "282865",
                "tvdb": "300474",
                "title": "Won't Get Fooled Again",
                "locations": (
                    "Criminal Minds S01E03 Won't Get Fooled Again WEBDL-720p.mkv",
                ),
                "status": {"completed": True, "time": 0},
            },
        ]
    },
    frozenset({("title", "Test"), ("locations", ("Test",))}): {
        "Season 1": [
            {
                "title": "S01E02",
                "locations": ("Test S01E02.mkv",),
                "status": {"completed": False, "time": 10},
            },
            {
                "title": "S01E03",
                "locations": ("Test S01E03.mkv",),
                "status": {"completed": True, "time": 0},
            },
            {
                "title": "S01E04",
                "locations": ("Test S01E04.mkv",),
                "status": {"completed": False, "time": 10},
            },
        ]
    },
}

expected_tv_show_watched_list_1 = {
    frozenset(
        {
            ("tvdb", "75710"),
            ("title", "Criminal Minds"),
            ("imdb", "tt0452046"),
            ("locations", ("Criminal Minds",)),
            ("tmdb", "4057"),
        }
    ): {
        "Season 1": [
            {
                "imdb": "tt0550489",
                "tmdb": "282843",
                "tvdb": "176357",
                "title": "Extreme Aggressor",
                "locations": (
                    "Criminal Minds S01E01 Extreme Aggressor WEBDL-720p.mkv",
                ),
                "status": {"completed": True, "time": 0},
            }
        ]
    },
    frozenset({("title", "Test"), ("locations", ("Test",))}): {
        "Season 1": [
            {
                "title": "S01E01",
                "locations": ("Test S01E01.mkv",),
                "status": {"completed": True, "time": 0},
            },
            {
                "title": "S01E02",
                "locations": ("Test S01E02.mkv",),
                "status": {"completed": True, "time": 0},
            },
        ]
    },
}

expected_tv_show_watched_list_2 = {
    frozenset(
        {
            ("tvdb", "75710"),
            ("title", "Criminal Minds"),
            ("imdb", "tt0452046"),
            ("locations", ("Criminal Minds",)),
            ("tmdb", "4057"),
        }
    ): {
        "Season 1": [
            {
                "imdb": "tt0550498",
                "tmdb": "282865",
                "tvdb": "300474",
                "title": "Won't Get Fooled Again",
                "locations": (
                    "Criminal Minds S01E03 Won't Get Fooled Again WEBDL-720p.mkv",
                ),
                "status": {"completed": True, "time": 0},
            }
        ]
    },
    frozenset({("title", "Test"), ("locations", ("Test",))}): {
        "Season 1": [
            {
                "title": "S01E03",
                "locations": ("Test S01E03.mkv",),
                "status": {"completed": True, "time": 0},
            },
            {
                "title": "S01E04",
                "locations": ("Test S01E04.mkv",),
                "status": {"completed": False, "time": 10},
            },
        ]
    },
}

movies_watched_list_1 = [
    {
        "imdb": "tt2380307",
        "tmdb": "354912",
        "title": "Coco",
        "locations": ("Coco (2017) Remux-1080p.mkv",),
        "status": {"completed": True, "time": 0},
    },
    {
        "tmdbcollection": "448150",
        "imdb": "tt1431045",
        "tmdb": "293660",
        "title": "Deadpool",
        "locations": ("Deadpool (2016) Remux-1080p.mkv",),
        "status": {"completed": True, "time": 0},
    },
]

movies_watched_list_2 = [
    {
        "imdb": "tt2380307",
        "tmdb": "354912",
        "title": "Coco",
        "locations": ("Coco (2017) Remux-1080p.mkv",),
        "status": {"completed": True, "time": 0},
    },
    {
        "imdb": "tt0384793",
        "tmdb": "9788",
        "tvdb": "9103",
        "title": "Accepted",
        "locations": ("Accepted (2006) Remux-1080p.mkv",),
        "status": {"completed": True, "time": 0},
    },
]


expected_movie_watched_list_1 = [
    {
        "tmdbcollection": "448150",
        "imdb": "tt1431045",
        "tmdb": "293660",
        "title": "Deadpool",
        "locations": ("Deadpool (2016) Remux-1080p.mkv",),
        "status": {"completed": True, "time": 0},
    }
]

expected_movie_watched_list_2 = [
    {
        "imdb": "tt0384793",
        "tmdb": "9788",
        "tvdb": "9103",
        "title": "Accepted",
        "locations": ("Accepted (2006) Remux-1080p.mkv",),
        "status": {"completed": True, "time": 0},
    }
]

# Test to see if objects get deleted all the way up to the root.
tv_shows_2_watched_list_1 = {
    frozenset(
        {
            ("tvdb", "75710"),
            ("title", "Criminal Minds"),
            ("imdb", "tt0452046"),
            ("locations", ("Criminal Minds",)),
            ("tmdb", "4057"),
        }
    ): {
        "Season 1": [
            {
                "imdb": "tt0550489",
                "tmdb": "282843",
                "tvdb": "176357",
                "title": "Extreme Aggressor",
                "locations": (
                    "Criminal Minds S01E01 Extreme Aggressor WEBDL-720p.mkv",
                ),
                "status": {"completed": True, "time": 0},
            },
        ]
    }
}


def test_simple_cleanup_watched():
    user_watched_list_1 = {
        "user1": {
            "TV Shows": tv_shows_watched_list_1,
            "Movies": movies_watched_list_1,
            "Other Shows": tv_shows_2_watched_list_1,
        },
    }
    user_watched_list_2 = {
        "user1": {
            "TV Shows": tv_shows_watched_list_2,
            "Movies": movies_watched_list_2,
            "Other Shows": tv_shows_2_watched_list_1,
        }
    }

    expected_watched_list_1 = {
        "user1": {
            "TV Shows": expected_tv_show_watched_list_1,
            "Movies": expected_movie_watched_list_1,
        }
    }

    expected_watched_list_2 = {
        "user1": {
            "TV Shows": expected_tv_show_watched_list_2,
            "Movies": expected_movie_watched_list_2,
        }
    }

    return_watched_list_1 = cleanup_watched(user_watched_list_1, user_watched_list_2)
    return_watched_list_2 = cleanup_watched(user_watched_list_2, user_watched_list_1)

    assert return_watched_list_1 == expected_watched_list_1
    assert return_watched_list_2 == expected_watched_list_2


def test_mapping_cleanup_watched():
    user_watched_list_1 = {
        "user1": {
            "TV Shows": tv_shows_watched_list_1,
            "Movies": movies_watched_list_1,
            "Other Shows": tv_shows_2_watched_list_1,
        },
    }
    user_watched_list_2 = {
        "user2": {
            "Shows": tv_shows_watched_list_2,
            "Movies": movies_watched_list_2,
            "Other Shows": tv_shows_2_watched_list_1,
        }
    }

    expected_watched_list_1 = {
        "user1": {
            "TV Shows": expected_tv_show_watched_list_1,
            "Movies": expected_movie_watched_list_1,
        }
    }

    expected_watched_list_2 = {
        "user2": {
            "Shows": expected_tv_show_watched_list_2,
            "Movies": expected_movie_watched_list_2,
        }
    }

    user_mapping = {"user1": "user2"}
    library_mapping = {"TV Shows": "Shows"}

    return_watched_list_1 = cleanup_watched(
        user_watched_list_1,
        user_watched_list_2,
        user_mapping=user_mapping,
        library_mapping=library_mapping,
    )
    return_watched_list_2 = cleanup_watched(
        user_watched_list_2,
        user_watched_list_1,
        user_mapping=user_mapping,
        library_mapping=library_mapping,
    )

    assert return_watched_list_1 == expected_watched_list_1
    assert return_watched_list_2 == expected_watched_list_2


def test_combine_watched_dicts():
    input_watched = [
        {
            "test3": {
                "Anime Movies": [
                    {
                        "title": "Ponyo",
                        "tmdb": "12429",
                        "imdb": "tt0876563",
                        "locations": ("Ponyo (2008) Bluray-1080p.mkv",),
                        "status": {"completed": True, "time": 0},
                    },
                    {
                        "title": "Spirited Away",
                        "tmdb": "129",
                        "imdb": "tt0245429",
                        "locations": ("Spirited Away (2001) Bluray-1080p.mkv",),
                        "status": {"completed": True, "time": 0},
                    },
                    {
                        "title": "Castle in the Sky",
                        "tmdb": "10515",
                        "imdb": "tt0092067",
                        "locations": ("Castle in the Sky (1986) Bluray-1080p.mkv",),
                        "status": {"completed": True, "time": 0},
                    },
                ]
            }
        },
        {"test3": {"Anime Shows": {}}},
        {"test3": {"Cartoon Shows": {}}},
        {
            "test3": {
                "Shows": {
                    frozenset(
                        {
                            ("tmdb", "64464"),
                            ("tvdb", "301824"),
                            ("tvrage", "45210"),
                            ("title", "11.22.63"),
                            ("locations", ("11.22.63",)),
                            ("imdb", "tt2879552"),
                        }
                    ): {
                        "Season 1": [
                            {
                                "imdb": "tt4460418",
                                "title": "The Rabbit Hole",
                                "locations": (
                                    "11.22.63 S01E01 The Rabbit Hole Bluray-1080p.mkv",
                                ),
                                "status": {"completed": True, "time": 0},
                            }
                        ]
                    }
                }
            }
        },
        {"test3": {"Subbed Anime": {}}},
    ]
    expected = {
        "test3": {
            "Anime Movies": [
                {
                    "title": "Ponyo",
                    "tmdb": "12429",
                    "imdb": "tt0876563",
                    "locations": ("Ponyo (2008) Bluray-1080p.mkv",),
                    "status": {"completed": True, "time": 0},
                },
                {
                    "title": "Spirited Away",
                    "tmdb": "129",
                    "imdb": "tt0245429",
                    "locations": ("Spirited Away (2001) Bluray-1080p.mkv",),
                    "status": {"completed": True, "time": 0},
                },
                {
                    "title": "Castle in the Sky",
                    "tmdb": "10515",
                    "imdb": "tt0092067",
                    "locations": ("Castle in the Sky (1986) Bluray-1080p.mkv",),
                    "status": {"completed": True, "time": 0},
                },
            ],
            "Anime Shows": {},
            "Cartoon Shows": {},
            "Shows": {
                frozenset(
                    {
                        ("tmdb", "64464"),
                        ("tvdb", "301824"),
                        ("tvrage", "45210"),
                        ("title", "11.22.63"),
                        ("locations", ("11.22.63",)),
                        ("imdb", "tt2879552"),
                    }
                ): {
                    "Season 1": [
                        {
                            "imdb": "tt4460418",
                            "title": "The Rabbit Hole",
                            "locations": (
                                "11.22.63 S01E01 The Rabbit Hole Bluray-1080p.mkv",
                            ),
                            "status": {"completed": True, "time": 0},
                        }
                    ]
                }
            },
            "Subbed Anime": {},
        }
    }

    assert combine_watched_dicts(input_watched) == expected