47 lines
3.0 KiB
Python
47 lines
3.0 KiB
Python
import json
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
import tempfile
|
|
import time
|
|
import unittest
|
|
|
|
SCRIPT = Path(__file__).resolve().parents[1] / 'scripts/placement.sh'
|
|
|
|
class PlacementTests(unittest.TestCase):
|
|
def run_case(self, status='True', age=600, missing=False, unavailable=False, cordoned=False, fallback=False, error=False):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
now = time.strftime('%Y-%m-%dT%H:%M:%SZ', time.gmtime(time.time()-age))
|
|
node = {'metadata': {'labels': {'factorio.brunner.ninja/unavailable': str(unavailable).lower()}}, 'spec': {'unschedulable': cordoned}, 'status': {'conditions':[{'type':'Ready','status':status,'lastTransitionTime':now}]}}
|
|
nodes = ['odroid-h3-'+str(i).zfill(2) for i in range(6)] if fallback else ['arschrock']
|
|
deployment={'metadata':{'resourceVersion':'123'},'spec':{'template':{'spec':{'affinity':{'nodeAffinity':{'requiredDuringSchedulingIgnoredDuringExecution':{'nodeSelectorTerms':[{'matchExpressions':[{'values':nodes}]}]}}}}}}}
|
|
(root/'node').write_text('' if missing else json.dumps(node))
|
|
(root/'deployment').write_text(json.dumps(deployment))
|
|
fake=root/'kubectl'
|
|
fake.write_text('#!/bin/sh\ncase "$*" in\n *"get node"*) '+('exit 1' if error else 'cat "$FIXTURES/node"')+' ;;\n *"get deployment"*) cat "$FIXTURES/deployment" ;;\n *"patch deployment"*) printf "%s\\n" "$*" > "$FIXTURES/patch" ;;\n *) exit 99 ;;\nesac\n')
|
|
fake.chmod(0o755)
|
|
result=subprocess.run(['sh',str(SCRIPT)],env={**os.environ,'PATH':tmp+':'+os.environ['PATH'],'FIXTURES':tmp},capture_output=True,text=True)
|
|
return result.returncode,(root/'patch').read_text() if (root/'patch').exists() else ''
|
|
|
|
def test_healthy_primary_stays(self):
|
|
self.assertEqual(self.run_case(),(0,''))
|
|
def test_notready_fails_over_after_grace(self):
|
|
code,patch=self.run_case(status='False'); self.assertEqual(code,0); self.assertIn('odroid-h3-05',patch)
|
|
def test_unreachable_fails_over(self):
|
|
self.assertIn('odroid-h3-00',self.run_case(status='Unknown')[1])
|
|
def test_missing_primary_fails_over(self):
|
|
self.assertIn('odroid-h3-00',self.run_case(missing=True)[1])
|
|
def test_transient_failure_does_not_move_world(self):
|
|
self.assertEqual(self.run_case(status='False',age=30),(0,''))
|
|
def test_recovery_waits_for_stability(self):
|
|
self.assertEqual(self.run_case(age=30,fallback=True),(0,''))
|
|
def test_stable_recovery_returns_to_primary(self):
|
|
self.assertIn('arschrock',self.run_case(fallback=True)[1])
|
|
def test_maintenance_moves_immediately(self):
|
|
self.assertIn('odroid-h3-00',self.run_case(unavailable=True,age=5)[1])
|
|
def test_cordon_moves_immediately(self):
|
|
self.assertIn('odroid-h3-00',self.run_case(cordoned=True)[1])
|
|
def test_api_error_is_not_a_missing_node(self):
|
|
code,patch=self.run_case(error=True); self.assertNotEqual(code,0); self.assertEqual(patch,'')
|