import pytest
from json import loads
import os

from solo_tool.session_manager import SessionManager
from fixtures import soloTool, mockPlayer, testSongs, sessionPath

@pytest.fixture
def testSessionFile(sessionPath, testSongs):
    contents = """[
    {
        "path" : "test.flac",
        "key_points" : []
    },
    {
        "path" : "test.mp3",
        "key_points" : [0.1, 0.3]
    }
]"""
    sessionFile = sessionPath / "test-session.json"
    with open(sessionFile, "w") as f:
        f.write(contents)
    return sessionFile

@pytest.fixture
def sessionManager(sessionPath):
    return SessionManager(str(sessionPath))

def test_loadSession(sessionManager, mockPlayer, testSessionFile):
    sessions = sessionManager.getSessions()
    assert sessions == [testSessionFile.stem]

    soloTool = sessionManager.loadSession(sessions[0], player=mockPlayer)
    assert soloTool.songs == ["test.flac", "test.mp3"]

    soloTool.song = 0
    assert soloTool.keyPoints == []

    soloTool.song = 1
    assert soloTool.keyPoints == [0.1, 0.3]

def test_saveSession(sessionManager, soloTool, testSessionFile, sessionPath):
    soloTool.addSong("test.flac")
    soloTool.addSong("test.mp3")
    soloTool.song = 1
    soloTool.keyPoints = [0.1, 0.3]

    sessionKey = "test_session_saved"
    sessionManager.saveSession(soloTool, sessionKey)

    with open(sessionPath / f"{sessionKey}.json", "r") as f:
        savedSession = loads(f.read())

    with open(testSessionFile, "r") as f:
        testSession = loads(f.read())

    assert savedSession == testSession

def test_loadAndSaveEmptySession(sessionManager, sessionPath, soloTool, tmp_path):
    emptySession = "empty_session"

    sessionManager.saveSession(soloTool, emptySession)
    reloadedTool = sessionManager.loadSession(emptySession)

    assert reloadedTool.songs == []