adjustable namespace, create e2e tests

Author: micw523

examples/create_deployment_from_yaml.py

@@ -16,15 +16,16 @@
 
 import yaml
 
-from kubernetes import utils, config
+from kubernetes import client, utils, config
 
 
 def main():
     # Configs can be set in Configuration class directly or using helper
     # utility. If no argument provided, the config will be loaded from
     # default location.
     config.load_kube_config()
-    k8s_client = utils.create_from_yaml("./nginx-deployment.yaml")
+    k8s_client = utils.create_from_yaml("nginx-deployment.yaml")
+    k8s_client
 
 if __name__ == '__main__':
     main()

kubernetes/e2e_test/test_utils.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+import unittest
+
+from kubernetes import utils, client
+from kubernetes.e2e_test import base
+
+class TestUtils(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.config = base.get_e2e_configuration()
+
+    def test_app_yaml(self):
+        k8s_client = utils.create_from_yaml("test_yaml/app.yaml")
+        self.assertEqual("apps/v1", 
+            k8s_client.get_api_resources().group_version)
+        deployments = k8s_client.get_namespaced_deployment(
+            namespace="default").items
+        deployment_name = deployments[0].metadata.name
+        self.assertEqual("nginx-deployment-app", deployment_name)
+        resp = k8s_client.delete_namespaced_deployment(
+            name=deployment_name, namespace="default", 
+            body=client.V1DeleteOptions(
+                propagation_policy="Foreground",
+                grace_period_seconds=1))
+        
+    def test_extension_yaml(self):
+        k8s_client = utils.create_from_yaml("test_yaml/extension.yaml")
+        self.assertEqual("extensions/v1beta1", 
+            k8s_client.get_api_resources().group_version)
+        deployments = k8s_client.get_namespaced_deployment(
+            namespace="default").items
+        deployment_name = deployments[0].metadata.name
+        self.assertEqual("nginx-deployment", deployment_name)
+        resp = k8s_client.delete_namespaced_deployment(
+            name=deployment_name, namespace="default", 
+            body=client.V1DeleteOptions(
+                propagation_policy="Foreground",
+                grace_period_seconds=1))
+    
+    def test_core_pod_yaml(self):
+        k8s_client = utils.create_from_yaml("test_yaml/core-pod.yaml")
+        self.assertEqual("v1", 
+            k8s_client.get_api_resources().group_version)
+        pods = k8s_client.list_namespaced_pod(namespace="default").items
+        pod_name = pods[0].metadata.name
+        self.assertEqual("myapp-pod", pod_name)
+        resp = k8s_client.delete_namespaced_pod(
+            name=pod_name, namespace="default",
+            body=client.V1DeleteOptions(
+                propagation_policy="Foreground",
+                grace_period_seconds=1))
+
+    def test_core_service_yaml(self):
+        k8s_client = utils.create_from_yaml("test_yaml/core-service.yaml")
+        self.assertEqual("v1", 
+            k8s_client.get_api_resources().group_version)
+        svcs = k8s_client.list_namespaced_service(namespace="default").items
+        svc_name = svcs[1].metadata.name
+        self.assertEqual("my-service", svc_name)
+        resp = k8s_client.delete_namespaced_pod(
+            name=svc_name, namespace="default",
+            body=client.V1DeleteOptions(
+                propagation_policy="Foreground",
+                grace_period_seconds=1))

examples/nginx-deployment-app.yaml renamed to kubernetes/e2e_test/test_yaml/app.yaml

@@ -1,7 +1,7 @@
 apiVersion: apps/v1
 kind: Deployment
 metadata:
-  name: nginx-deployment
+  name: nginx-deployment-app
   labels:
     app: nginx
 spec:

examples/core-pod.yaml renamed to kubernetes/e2e_test/test_yaml/core-pod.yaml

@@ -0,0 +1,17 @@
+apiVersion: extensions/v1beta1
+kind: Deployment
+metadata:
+  name: nginx-deployment
+spec:
+  replicas: 3
+  template:
+    metadata:
+      labels:
+        app: nginx
+    spec:
+      containers:
+      - name: nginx
+        image: nginx:1.7.9
+        ports:
+        - containerPort: 80
+

examples/core-service.yaml renamed to kubernetes/e2e_test/test_yaml/core-service.yaml

@@ -29,8 +29,12 @@ def create_from_yaml(yaml_file):
         k8s_client = getattr(client, "%s%sApi" % (
                 api_type.capitalize(), api_version.capitalize()))()
         action_type = dep["kind"]
+        if "namespace" in dep["metadata"]:
+                dep_namespace = dep["metadata"]["namespace"]
+        else:
+                dep_namespace = "default"
         resp = getattr(k8s_client, "create_namespaced_%s" % action_type.lower())(
-                body=dep, namespace="default")
+                body=dep, namespace=dep_namespace)
         print("Deployment created. status='%s'" % str(resp.status))
         return k8s_client