108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
import datetime
|
|
from unittest.mock import MagicMock
|
|
|
|
from runboat.db import BuildsDb
|
|
from runboat.models import Build, BuildInitStatus, BuildStatus
|
|
|
|
|
|
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
|