mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-10 12:47:04 +00:00
kevinveenbirkenbachandClaude Opus 4.8
120096bccf
When a run deploys both single-node modes, the host discovery redeployed roles the compose line already covers (e.g. web-svc-html in both matrices). The single-node workflow now signals a co-running compose line via INFINITO_HOST_EXCLUDE_COMPOSE (a run-context registry key like INFINITO_DEPLOY_MODE), and the host query excludes the compose line's actually-triggered selection before its own budget cut, so freed host slots go to roles compose does not cover. Priority and regular lines each filter against their own compose counterpart through the identical whitelist/blacklist pass-through; host-only runs stay unchanged. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from cli.meta.ci import query
|
|
|
|
|
|
class MaxJobsPriorityDeduction(unittest.TestCase):
|
|
def test_no_blacklist_returns_full_budget(self):
|
|
with patch.dict("os.environ", {"INFINITO_MAX_JOBS": "10"}):
|
|
self.assertEqual(query.max_jobs("swarm"), 10)
|
|
|
|
def test_priority_consumption_is_deducted(self):
|
|
with (
|
|
patch.dict("os.environ", {"INFINITO_MAX_JOBS": "10"}),
|
|
patch.object(query, "discover", return_value=["a#0", "a#1", "b#0"]) as d,
|
|
):
|
|
self.assertEqual(query.max_jobs("swarm", blacklist="a b"), 7)
|
|
d.assert_called_once_with("swarm", whitelist="a b", lifecycles="")
|
|
|
|
def test_budget_floors_at_zero(self):
|
|
with (
|
|
patch.dict("os.environ", {"INFINITO_MAX_JOBS": "2"}),
|
|
patch.object(query, "discover", return_value=["a#0", "a#1", "b#0"]),
|
|
):
|
|
self.assertEqual(query.max_jobs("swarm", blacklist="a b"), 0)
|
|
|
|
def test_discover_short_circuits_on_zero_budget(self):
|
|
with (
|
|
patch.dict("os.environ", {"INFINITO_MAX_JOBS": "0"}),
|
|
patch.object(query, "subprocess") as sp,
|
|
):
|
|
self.assertEqual(query.discover("swarm", whitelist="x"), [])
|
|
sp.run.assert_not_called()
|
|
|
|
|
|
class HostExcludesComposeCovered(unittest.TestCase):
|
|
def test_inactive_without_flag(self):
|
|
with patch.dict("os.environ", {"INFINITO_HOST_EXCLUDE_COMPOSE": ""}):
|
|
self.assertEqual(
|
|
query.compose_covered(
|
|
"host", whitelist="", blacklist="", lifecycles=""
|
|
),
|
|
(),
|
|
)
|
|
|
|
def test_inactive_for_other_modes(self):
|
|
with patch.dict("os.environ", {"INFINITO_HOST_EXCLUDE_COMPOSE": "true"}):
|
|
self.assertEqual(
|
|
query.compose_covered(
|
|
"compose", whitelist="", blacklist="", lifecycles=""
|
|
),
|
|
(),
|
|
)
|
|
|
|
def test_covered_roles_enter_the_filter(self):
|
|
with (
|
|
patch.dict("os.environ", {"INFINITO_HOST_EXCLUDE_COMPOSE": "true"}),
|
|
patch.object(query, "discover", return_value=["web-svc-html", "a"]),
|
|
):
|
|
covered = query.compose_covered(
|
|
"host", whitelist="", blacklist="", lifecycles=""
|
|
)
|
|
self.assertEqual(covered, ("web-svc-html", "a"))
|
|
self.assertIn(
|
|
"not (name %% {web-svc-html,a})",
|
|
query.build_filter("host", covered=covered),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|