import datetime
from unittest.mock import MagicMock

from runboat.db import BuildsDb
from runboat.models import Build, BuildInitStatus, BuildStatus, Repo


def _make_build(
    name: str | None = None,
    *,
    status: BuildStatus | None = None,
    init_status: BuildInitStatus | None = None,
    repo: str | None = None,
    pr: int | None = None,
) -> Build:
    name = name or "build-a"
    return Build(
        name=name,
        deployment_name=name + "-odoo",
        repo=repo or "oca/mis-builder",
        target_branch="15.0",
        pr=pr or None,
        git_commit="0d35a10f161b410f2baa3d416a338d191b6dabc0",
        image="ghcr.io/oca/oca-ci:py3.8-odoo15.0",
        status=status or BuildStatus.starting,
        init_status=init_status or BuildInitStatus.todo,
        desired_replicas=0,
        last_scaled=datetime.datetime(2021, 10, 1, 12, 0, 0),
        created=datetime.datetime(2021, 10, 1, 11, 0, 0),
    )


def test_add() -> None:
    db = BuildsDb()
    listener = MagicMock()
    db.register_listener(listener)
    db.add(_make_build())  # new
    listener.on_build_event.assert_called()
    listener.reset_mock()
    db.add(_make_build())  # no change
    listener.on_build_event.assert_not_called()
    db.add(_make_build(status=BuildStatus.failed))
    listener.on_build_event.assert_called()


def test_remove() -> None:
    db = BuildsDb()
    listener = MagicMock()
    db.register_listener(listener)
    db.remove("not-a-build")
    listener.on_build_event.assert_not_called()
    build = _make_build()
    db.add(build)
    db.remove(build.name)
    listener.on_build_event.assert_called()


def test_get_for_commit() -> None:
    db = BuildsDb()
    build = _make_build()
    db.add(build)
    assert (
        db.get_for_commit(
            build.repo, build.target_branch, build.pr, git_commit=build.git_commit
        )
        == build
    )
    assert (
        db.get_for_commit(
            "not-a-build", build.target_branch, build.pr, git_commit=build.git_commit
        )
        is None
    )


def test_search() -> None:
    db = BuildsDb()
    db.add(build1 := _make_build(name="b1", repo="oca/repo1"))
    db.add(_make_build(name="b2", repo="oca/repo2"))
    assert len(list(db.search())) == 2
    assert list(db.search("oca/repo1")) == [build1]


def test_count_by_status() -> None:
    db = BuildsDb()
    db.add(_make_build(name="b1", status=BuildStatus.started))
    db.add(_make_build(name="b2", status=BuildStatus.stopped))
    assert db.count_by_status(BuildStatus.started) == 1
    assert db.count_by_status(BuildStatus.stopped) == 1
    assert db.count_by_status(BuildStatus.failed) == 0


def test_count_by_init_status() -> None:
    db = BuildsDb()
    db.add(_make_build(name="b1", init_status=BuildInitStatus.started))
    db.add(_make_build(name="b2", init_status=BuildInitStatus.todo))
    assert db.count_by_init_status(BuildInitStatus.started) == 1
    assert db.count_by_init_status(BuildInitStatus.todo) == 1
    assert db.count_by_init_status(BuildInitStatus.failed) == 0


def test_count_all() -> None:
    db = BuildsDb()
    assert db.count_all() == 0
    db.add(_make_build(name="b1"))
    assert db.count_all() == 1
    db.add(_make_build(name="b2"))
    assert db.count_all() == 2


def test_repos() -> None:
    db = BuildsDb()
    db.add(_make_build(name="b1", repo="oca/repo1"))
    db.add(_make_build(name="b2", repo="oca/repo2"))
    assert db.repos() == [Repo(name="oca/repo1"), Repo(name="oca/repo2")]