public_git
/
postgres-operator
mirror of https://github.com/zalando/postgres-operator.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix flake8 warnings

This commit is contained in:
Sergey Dudoladov 2019-05-02 12:04:01 +02:00
parent 30e7445c80
commit 25fea48f8a
1 changed files with 40 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
e2e/tests/test_smoke.py
View File

@ -1,20 +1,21 @@
import unittest, yaml, time
from kubernetes import client, config, utils
from pprint import pprint
import unittest
import time
import timeout_decorator
import subprocess
import git
from kubernetes import client, config, utils
class K8sApi:
def __init__(self):
self.config = config.load_kube_config()
self.k8s_client = client.ApiClient()
self.core_v1 = client.CoreV1Api()
self.crd_api = client.CustomObjectsApi()
self.apps_v1 = client.AppsV1Api()
self.config = config.load_kube_config()
self.k8s_client = client.ApiClient()
self.core_v1 = client.CoreV1Api()
self.crd = client.CustomObjectsApi()
self.apps_v1 = client.AppsV1Api()
class SmokeTestCase(unittest.TestCase):
@ -34,7 +35,8 @@ class SmokeTestCase(unittest.TestCase):
'''
# HACK
# 1. creating RBAC entites with a separate client fails with "AttributeError: object has no attribute 'select_header_accept'"
# 1. creating RBAC entites with a separate client fails
# with "AttributeError: object has no attribute 'select_header_accept'"
# 2. utils.create_from_yaml cannot create multiple entites from a single file
subprocess.run(["kubectl", "create", "-f", "manifests/operator-service-account-rbac.yaml"])
@ -53,10 +55,10 @@ class SmokeTestCase(unittest.TestCase):
"template": {
"spec": {
"containers": [
{
"name" : "postgres-operator",
"image": "registry.opensource.zalan.do/acid/postgres-operator:" + version
}
{
"name": "postgres-operator",
"image": "registry.opensource.zalan.do/acid/postgres-operator:" + version
}
]
}
}
@ -68,8 +70,8 @@ class SmokeTestCase(unittest.TestCase):
while pod_phase != 'Running':
pods = k8s_api.core_v1.list_namespaced_pod('default', label_selector='name=postgres-operator').items
if pods:
operator_pod = pods[0]
pod_phase = operator_pod.status.phase
operator_pod = pods[0]
pod_phase = operator_pod.status.phase
print("Waiting for the operator pod to start. Current phase of pod lifecycle: " + str(pod_phase))
time.sleep(5)
@ -80,8 +82,8 @@ class SmokeTestCase(unittest.TestCase):
while pod_phase != 'Running':
pods = k8s_api.core_v1.list_namespaced_pod('default', label_selector='spilo-role=master').items
if pods:
operator_pod = pods[0]
pod_phase = operator_pod.status.phase
operator_pod = pods[0]
pod_phase = operator_pod.status.phase
print("Waiting for the Spilo master pod to start. Current phase: " + str(pod_phase))
time.sleep(5)
@ -91,8 +93,9 @@ class SmokeTestCase(unittest.TestCase):
Check that there is a single pod in the k8s cluster with the label "spilo-role=master".
"""
k8s = K8sApi()
master_pods = k8s.core_v1.list_namespaced_pod('default', label_selector='spilo-role=master,version=acid-minimal-cluster').items
self.assertEqual(len(master_pods), 1, "Expected 1 master pod,found " + str(len(master_pods)))
labels = 'spilo-role=master,version=acid-minimal-cluster'
master_pods = k8s.core_v1.list_namespaced_pod('default', label_selector=labels).items
self.assertEqual(len(master_pods), 1, "Expected 1 master pod, found " + str(len(master_pods)))
@timeout_decorator.timeout(240)
def test_scaling(self):
@ -106,12 +109,14 @@ class SmokeTestCase(unittest.TestCase):
"numberOfInstances": 3
}
}
_ = k8s.crd_api.patch_namespaced_custom_object("acid.zalan.do", "v1", "default", "postgresqls", "acid-minimal-cluster", body)
_ = k8s.crd.patch_namespaced_custom_object("acid.zalan.do",
"v1", "default", "postgresqls", "acid-minimal-cluster", body)
while len(k8s.core_v1.list_namespaced_pod('default', label_selector='version=acid-minimal-cluster').items) != 3:
labels = 'version=acid-minimal-cluster'
while len(k8s.core_v1.list_namespaced_pod('default', label_selector=labels).items) != 3:
print("Waiting for the cluster to scale up to 3 pods.")
time.sleep(5)
self.assertEqual(3, len(k8s.core_v1.list_namespaced_pod('default', label_selector='version=acid-minimal-cluster').items))
self.assertEqual(3, len(k8s.core_v1.list_namespaced_pod('default', label_selector=labels).items))
body = {
"spec": {
@ -119,12 +124,14 @@ class SmokeTestCase(unittest.TestCase):
}
}
_ = k8s.crd_api.patch_namespaced_custom_object("acid.zalan.do", "v1", "default", "postgresqls", "acid-minimal-cluster", body)
_ = k8s.crd.patch_namespaced_custom_object("acid.zalan.do",
"v1", "default", "postgresqls", "acid-minimal-cluster", body)
while len(k8s.core_v1.list_namespaced_pod('default', label_selector='version=acid-minimal-cluster').items) != 2:
while len(k8s.core_v1.list_namespaced_pod('default', label_selector=labels).items) != 2:
print("Waiting for the cluster to scale down to 2 pods.")
time.sleep(5)
self.assertEqual(2, len(k8s.core_v1.list_namespaced_pod('default', label_selector='version=acid-minimal-cluster').items))
self.assertEqual(2, len(k8s.core_v1.list_namespaced_pod('default', label_selector=labels).items))
if __name__ == '__main__':
unittest.main()