import corrlib.integrity as integ
import corrlib.find as find
import datalad.api as dl
import corrlib.initialization as cinit
import sqlite3
from pathlib import Path
import os


def test_list_ensembles(tmp_path: Path) -> None:
    """
    Check against the implementation in find to check if they are the same.
    """
    os.mkdir(tmp_path / 'archive')
    os.mkdir(tmp_path / 'archive' / 'A')
    os.mkdir(tmp_path / 'archive' / 'B')
    os.mkdir(tmp_path / 'archive' / 'C')
    integ_results = integ._list_ensembles(tmp_path)
    assert len(integ_results) == 3
    find_results = find.list_ensembles(tmp_path)
    assert len(find_results) == 3
    for f,i in zip(find_results, integ_results):
        assert f == i


def test_list_projects(tmp_path: Path) -> None:
    cinit.create(tmp_path)
    db = tmp_path / "backlogger.db"
    dl.unlock(str(db), dataset=str(tmp_path))
    conn = sqlite3.connect(db)
    c = conn.cursor()

    customTags = ""
    owner = "owner"
    code = "sfcf"
    created_at = "today"
    updated_at = "today"

    id = "asdf1"
    aliases = "a1,s1,d1,f1"
    c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", (id, aliases, customTags, owner, code , created_at, updated_at))
    id = "asdf2"
    aliases = "a2,s2,d2,f2"
    c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", (id, aliases, customTags, owner, code , created_at, updated_at))
    id = "asdf3"
    aliases = "a3,s3,d3,f3"
    c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", (id, aliases, customTags, owner, code , created_at, updated_at))
    conn.commit()
    conn.close
    integ_results = integ._list_projects(tmp_path)
    assert len(integ_results) == 3
    find_results = find.list_projects(tmp_path)
    assert len(find_results) == 3
    for f,i in zip(find_results, integ_results):
        assert f == i