# -*- coding: utf-8 -*-
from pytube import playlist


def test_construct():
    ob = playlist.Playlist(
        "https://www.youtube.com/watch?v=m5q2GCsteQs&list="
        "PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw")
    expected = "https://www.youtube.com/" \
               "playlist?list=" \
               "PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw"

    assert ob.construct_playlist_url() == expected


def test_link_parse():
    ob = playlist.Playlist(
        "https://www.youtube.com/watch?v=m5q2GCsteQs&list="
        "PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw")

    expected = ["/watch?v=m5q2GCsteQs",
                "/watch?v=5YK63cXyJ2Q",
                "/watch?v=Rzt4rUPFYD4"]
    assert ob.parse_links() == expected


def test_populate():
    ob = playlist.Playlist(
        "https://www.youtube.com/watch?v=m5q2GCsteQs&list="
        "PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw")
    expected = ["https://www.youtube.com/watch?v=m5q2GCsteQs",
                "https://www.youtube.com/watch?v=5YK63cXyJ2Q",
                "https://www.youtube.com/watch?v=Rzt4rUPFYD4"]

    ob.populate_video_urls()
    assert ob.video_urls == expected


def test_download():
    ob = playlist.Playlist(
        "https://www.youtube.com/watch?v=lByG_AgKS9k&list="
        "PL525f8ds9RvuerPZ3bZygmNiYw2sP4BDk")
    ob.download_all()