def test_associate_pods(self):
deps = K8Kat.deps().names('d11').go()
self.assertEqual(self.the_pod_app_lbs(deps), ['d11'])
deps = K8Kat.deps().ns('n1', 'n2').go()
self.assertEqual(self.the_pod_app_lbs(deps),
['d0', 'd11', 'd12', 'd21'])
def label_matrix():
get = lambda *keys: [request.args.get(k) for k in keys]
_type, ns, name = get('matcher_type', 'matcher_ns', 'matcher_name')
collection = K8Kat.deps() if _type == 'deployment' else K8Kat.svcs()
matcher = collection.ns(ns).find(name)
result = ResUtils.label_matrix(matcher)
return jsonify(result)
def stdSetUpClass(cls, step_class):
cls.dep = K8Kat.deps().ns(TESTING_NS).find(TESTING_DEP_NM)
cls.svc = K8Kat.svcs().ns(TESTING_NS).find(TESTING_SVC_NM)
cls.step = step_class(
from_port=cls.svc.from_port,
dep_ns=TESTING_NS,
svc_name=TESTING_SVC_NM,
dep_name=TESTING_DEP_NM,
)
def test_names(self):
result = K8Kat.deps().names('d11').go()
self.assertEqual([dep.name for dep in result], ['d11'])
result = K8Kat.deps().names('d11', 'd12').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd12'])
result = K8Kat.deps().names('d11', 'd22').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd22'])
result = K8Kat.deps().names('d11', 'd22').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd22'])
def test_associate_svcs(self):
deps = K8Kat.deps().names('d0').go()
self.assertEqual(self.the_svc_names(deps), [])
deps = K8Kat.deps().names('d11').go()
self.assertEqual(self.the_svc_names(deps), ['d11'])
deps = K8Kat.deps().names('d11', 'd12').go()
self.assertEqual(self.the_svc_names(deps), ['d11'])
deps = K8Kat.deps().ns('n1', 'n2').go()
self.assertEqual(self.the_svc_names(deps), ['d11', 'd21'])
def test_lbs_inc_any(self):
result = K8Kat.deps().ns('n1').lbs_inc_any(c='c', l1='v1').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd12'])
result = K8Kat.deps().ns('n1', 'n2').lbs_inc_any(c='c', l1='v1').go()
self.assertCountEqual([dep.name for dep in result],
['d11', 'd12', 'd21'])
q = K8Kat.deps().ns('n1', 'n2').lbs_inc_any(c='c', l1='v1')
result = q.lbs_inc_each(l1='v1').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd21'])
q = K8Kat.deps().ns('n1', 'n2').lbs_inc_each(l1='v1')
result = q.lbs_inc_any(c='c', l1='v1').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd21'])
def params_to_deps():
q = K8Kat.deps()
ns_white = request.args.get('ns_filter_type', 'whitelist')
ns_white = True if ns_white == 'whitelist' else False
ns_filters = request.args.get('ns_filters')
ns_filters = ns_filters and ns_filters.split(',') or None
lb_white = request.args.get('lb_filter_type', 'blacklist')
lb_white = True if lb_white == 'whitelist' else False
lb_filters = request.args.get('lb_filters')
lb_filters = lb_filters and eq_strs_to_tups(lb_filters)
if ns_filters is not None:
q = q.ns(ns_filters) if ns_white else q.not_ns(ns_filters)
if lb_filters is not None:
q = q.lbs_inc_each(lb_filters) if lb_white else q.lbs_exc_each(
lb_filters)
deps = q.go()
if request.args.get('svcs') == 'true':
DepComposer.associate_svcs(deps)
if request.args.get('pods') == 'true':
DepComposer.associate_pods(deps)
return deps
def dep_by_ns() -> List[Dict[str, str]]:
from k8_kat.base.k8_kat import K8Kat
deps = K8Kat.deps().not_ns('kube-system').go()
output = []
for name in set([dep.name for dep in deps]):
appears_in = set([dep.ns for dep in deps if dep.name == name])
output.append(dict(name=name, namespaces=list(appears_in)))
return output
def validate_labels(ns, name):
dep = K8Kat.deps().ns(ns).find(name)
v = DepWarnings
return jsonify(
dict(template_covers=v.check_pod_template_inclusive(dep),
no_eavesdrop=v.check_no_pods_eavesdrop(dep),
labels_unique=v.check_labels_unique(dep),
pods_same_ns=v.check_pods_in_same_ns(dep)))
def perform(self):
possible_labels = [('k8s-app', 'kube-dns'), ('k8s-app', 'core-dns')]
pods = K8Kat.pods().ns("kube-system").lbs_inc_any(possible_labels).go()
running = [pod for pod in pods if pod.status == 'Running']
outputs = [f"{pod.name}: {pod.phase}" for pod in pods]
return self.record_step_performed(
outcome=len(running) > 0,
outputs=outputs,
bundle={}
)
def test_lbs_inc_each(self):
pass
result = K8Kat.deps().ns(['n1']).lbs_inc_each(l1='v1').go()
self.assertCountEqual([dep.name for dep in result], ['d11'])
result = K8Kat.deps().ns('n1').lbs_inc_each([('c', 'c')]).go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd12'])
result = K8Kat.deps().ns('n1', 'n2').lbs_inc_each(l1='v1').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd21'])
q = [('l1', 'v1'), ('c', 'c')]
result = K8Kat.deps().ns('n1', 'n2').lbs_inc_each(q).go()
self.assertCountEqual([dep.name for dep in result], ['d11'])
result = K8Kat.deps().ns('n1', 'n2').lbs_inc_each(c='c').go()
self.assertCountEqual([dep.name for dep in result], ['d11', 'd12'])
q = [('x', 'y'), ('c', 'c')]
result = K8Kat.deps().ns('n1', 'n2').lbs_inc_each(q).go()
self.assertCountEqual([dep.name for dep in result], [])
def test_delete_all(self): self.assertCountEqual(names(), ['p11', 'p12', 'p21']) K8Kat.pods().lbs_inc_each(l1='v1').delete_all() self.assertCountEqual(names(), ['p11', 'p12', 'p21'])
def stunt_pods():
broker.check_connected_or_raise()
garbage = K8Kat.pods().lbs_inc_each(StuntPod.labels()).go()
ser = garbage.serialize(PodSerialization.standard)
return jsonify(data=ser)
def names(query=None):
query = query or K8Kat.pods()
return query.pluck('name')
def find(self):
pod = K8Kat.pods().ns(self.namespace).find(self.pod_name)
return pod and pod.raw
def scale_replicas():
j = request.json
ns, name, scale_to = j['dep_namespace'], j['dep_name'], j['scale_to']
K8Kat.deps().ns(ns).find(name).scale(int(scale_to))
return jsonify(data=dict(status='working'))
def deployment_pods(ns, name):
dep = K8Kat.deps().ns(ns).find(name)
serialized = [PodSerialization.standard(pod) for pod in dep.pods()]
return jsonify(dict(data=serialized))
def find_and_assoc_pods(self):
from k8_kat.base.k8_kat import K8Kat
matchers = list(self.pod_select_labels.items())
self.assoced_pods = K8Kat.pods().ns(
self.ns).lbs_inc_each(matchers).go()
def run_command():
j = request.json
ns, pod_name, cmd = j['pod_namespace'], j['pod_name'], j['command']
pod = K8Kat.pods().ns(ns).find(pod_name)
return jsonify(data=pod.cmd(cmd))
def kill_stunt_pods():
pods = K8Kat.pods().lbs_inc_each(StuntPod.labels())
pods.delete_all()
def logs(ns, name):
pod = K8Kat.pods().ns(ns).find(name)
since_seconds = int(request.args.get('since_seconds', '5000'))
return dict(data=pod.logs(since_seconds))
def find_and_assoc_svcs(self):
from k8_kat.base.k8_kat import K8Kat
candidate_svcs = K8Kat.svcs().ns(self.ns).go()
checker = lambda svc: ResUtils.dep_matches_svc(self.raw, svc.raw)
self.assoced_svcs = [s for s in candidate_svcs if checker(s)]
def events(ns, name):
pod = K8Kat.pods().ns(ns).find(name)
serializer = EventSerialization.standard
serialized = [serializer(e) for e in pod.events()]
return jsonify(data=serialized)
def show(ns, name):
dep = K8Kat.deps().ns(ns).find(name).with_friends()
serialized = dep.serialize(Ser.with_pods_and_svcs)
return jsonify(serialized)
def kill_stunt_pods():
garbage = K8Kat.pods().lbs_inc_each(StuntPod.labels())
garbage.delete_all()
return jsonify(status='done')
def image_reload():
j = request.json
ns, name = j['dep_namespace'], j['dep_name']
K8Kat.deps().ns(ns).find(name).restart_pods()
return jsonify(data=dict(status='working'))
def __init__(self, **args):
super().__init__()
self.from_port = args['from_port']
self.dep = K8Kat.deps().ns(args['dep_ns']).find(args['dep_name'])
self.svc = K8Kat.svcs().ns(args['dep_ns']).find(args['svc_name'])
self._stunt_pod = None
def new_image():
j = request.json
ns, name, img_name = j['dep_namespace'], j['dep_name'], j['target_name']
K8Kat.deps().ns(ns).find(name).replace_image(img_name)
return jsonify(data=dict(status='working'))
def check_labels_unique(dep):
twins = K8Kat.deps().ns(dep.ns).lbs_inc_each(dep.labels)
return len(twins.go()) == 0
def deployment_services(ns, name):
dep = K8Kat.deps().ns(ns).find(name)
svcs = dep.svcs()
serialized = [SvcSerialization.with_endpoints(svc) for svc in svcs]
return jsonify(dict(data=serialized))
