mirror of
https://github.com/dredx/prole.git
synced 2026-09-24 19:14:33 +00:00
4ee2b259c9
- Add build-context helper to copy Docker context safely (ignore runtime data, keep symlinks) - Update UI and core actions to use ~/.prole/build and shared copy helper - Add/adjust tests and scripts; introduce knoe ops helpers and update manifests Co-authored-by: Junie <junie@jetbrains.com>
134 lines
3.6 KiB
Python
134 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
|
|
import json
|
|
|
|
|
|
def _kubectl_nodes_json(*, items: list[dict]) -> str:
|
|
return json.dumps({"apiVersion": "v1", "items": items})
|
|
|
|
|
|
def test_evaluate_optional_workloads_allowed_true_two_ready(monkeypatch):
|
|
from knoe.core import policy
|
|
|
|
class _Res:
|
|
returncode = 0
|
|
stderr = ""
|
|
stdout = _kubectl_nodes_json(
|
|
items=[
|
|
{
|
|
"spec": {},
|
|
"status": {
|
|
"conditions": [
|
|
{"type": "Ready", "status": "True"},
|
|
]
|
|
},
|
|
},
|
|
{
|
|
"spec": {},
|
|
"status": {
|
|
"conditions": [
|
|
{"type": "Ready", "status": "True"},
|
|
]
|
|
},
|
|
},
|
|
]
|
|
)
|
|
|
|
def _run(*_args, **_kwargs):
|
|
return _Res()
|
|
|
|
monkeypatch.setattr(policy.subprocess, "run", _run)
|
|
|
|
allowed, ready_schedulable, reason = policy.evaluate_optional_workloads_allowed(
|
|
min_nodes=2
|
|
)
|
|
assert allowed is True
|
|
assert ready_schedulable == 2
|
|
assert "optional_workloads_allowed=True" in reason
|
|
assert "min_required=2" in reason
|
|
|
|
|
|
def test_evaluate_optional_workloads_allowed_false_one_ready(monkeypatch):
|
|
from knoe.core import policy
|
|
|
|
class _Res:
|
|
returncode = 0
|
|
stderr = ""
|
|
stdout = _kubectl_nodes_json(
|
|
items=[
|
|
{
|
|
"spec": {},
|
|
"status": {
|
|
"conditions": [
|
|
{"type": "Ready", "status": "True"},
|
|
]
|
|
},
|
|
}
|
|
]
|
|
)
|
|
|
|
monkeypatch.setattr(policy.subprocess, "run", lambda *_a, **_k: _Res())
|
|
|
|
allowed, ready_schedulable, reason = policy.evaluate_optional_workloads_allowed(
|
|
min_nodes=2
|
|
)
|
|
assert allowed is False
|
|
assert ready_schedulable == 1
|
|
assert "optional_workloads_allowed=False" in reason
|
|
assert "min_required=2" in reason
|
|
|
|
|
|
def test_evaluate_optional_workloads_ignores_unschedulable(monkeypatch):
|
|
from knoe.core import policy
|
|
|
|
class _Res:
|
|
returncode = 0
|
|
stderr = ""
|
|
stdout = _kubectl_nodes_json(
|
|
items=[
|
|
{
|
|
"spec": {"unschedulable": True},
|
|
"status": {
|
|
"conditions": [
|
|
{"type": "Ready", "status": "True"},
|
|
]
|
|
},
|
|
},
|
|
{
|
|
"spec": {},
|
|
"status": {
|
|
"conditions": [
|
|
{"type": "Ready", "status": "True"},
|
|
]
|
|
},
|
|
},
|
|
]
|
|
)
|
|
|
|
monkeypatch.setattr(policy.subprocess, "run", lambda *_a, **_k: _Res())
|
|
|
|
allowed, ready_schedulable, _reason = policy.evaluate_optional_workloads_allowed(
|
|
min_nodes=2
|
|
)
|
|
assert allowed is False
|
|
assert ready_schedulable == 1
|
|
|
|
|
|
def test_evaluate_optional_workloads_kubectl_error_is_disallowed(monkeypatch):
|
|
from knoe.core import policy
|
|
|
|
class _Res:
|
|
returncode = 1
|
|
stderr = "boom"
|
|
stdout = ""
|
|
|
|
monkeypatch.setattr(policy.subprocess, "run", lambda *_a, **_k: _Res())
|
|
|
|
allowed, ready_schedulable, reason = policy.evaluate_optional_workloads_allowed(
|
|
min_nodes=2
|
|
)
|
|
assert allowed is False
|
|
assert ready_schedulable == 0
|
|
assert "failed to query nodes" in reason
|