from app.playlists import Notes, PlaylistTab, Tracks
from app.models import Playlists


def test_init(qtbot, session):
    """Just check we can create a playlist_tab"""

    playlist = Playlists(session, "my playlist")
    playlist_tab = PlaylistTab(None, session, playlist)
    assert playlist_tab


def test_save_and_restore(qtbot, session):
    """Playlist with one track, one note, save and restore"""

    # Create playlist
    playlist = Playlists(session, "my playlist")
    playlist_tab = PlaylistTab(None, session, playlist)

    # Insert a note
    note_text = "my note"
    note_row = 7
    note = Notes(session, playlist.id, note_row, note_text)
    playlist_tab._insert_note(session, note)

    # Add a track
    track_path = "/a/b/c"
    track = Tracks(session, track_path)
    playlist_tab._insert_track(session, track)

    # Save playlist
    playlist_tab.save_playlist(session)

    # Retrieve playlist
    playlists = Playlists.get_open(session)
    assert len(playlists) == 1
    retrieved_playlist = playlists[0]
    assert track_path in [a.path for a in retrieved_playlist.tracks.values()]
    assert note_text in [a.note for a in retrieved_playlist.notes]