From c9e11d02b0cfef3141365018b01ad1bd4ed01b22 Mon Sep 17 00:00:00 2001
From: "Huan-Ting,Chen" <dingyiyi0226@gmail.com>
Date: Fri, 21 Jan 2022 15:43:25 +0800
Subject: [PATCH] add e2e_test for create_from_directory

---
 kubernetes/e2e_test/test_utils.py             | 22 +++++++++++++++++++
 ...multi-resource-replication-controller.yaml | 19 ++++++++++++++++
 .../multi-resource-service.yaml               | 13 +++++++++++
 3 files changed, 54 insertions(+)
 create mode 100644 kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-replication-controller.yaml
 create mode 100644 kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-service.yaml

diff --git a/kubernetes/e2e_test/test_utils.py b/kubernetes/e2e_test/test_utils.py
index 4096bf86e..98768b0c9 100644
--- a/kubernetes/e2e_test/test_utils.py
+++ b/kubernetes/e2e_test/test_utils.py
@@ -288,6 +288,28 @@ class TestUtils(unittest.TestCase):
         core_api.delete_namespaced_service(name="mock-4",
                                            namespace="default", body={})
 
+    # Tests for creating multi-resource from directory
+
+    def test_create_multi_resource_from_directory(self):
+        """
+        Should be able to create a service and a replication controller
+        from a directory
+        """
+        k8s_client = client.api_client.ApiClient(configuration=self.config)
+        utils.create_from_directory(
+            k8s_client, self.path_prefix + "multi-resource/")
+        core_api = client.CoreV1Api(k8s_client)
+        svc = core_api.read_namespaced_service(name="mock",
+                                               namespace="default")
+        self.assertIsNotNone(svc)
+        ctr = core_api.read_namespaced_replication_controller(
+            name="mock", namespace="default")
+        self.assertIsNotNone(ctr)
+        core_api.delete_namespaced_replication_controller(
+            name="mock", namespace="default", propagation_policy="Background")
+        core_api.delete_namespaced_service(name="mock",
+                                           namespace="default", body={})
+
     # Tests for multi-resource yaml objects
 
     def test_create_from_multi_resource_yaml(self):
diff --git a/kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-replication-controller.yaml b/kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-replication-controller.yaml
new file mode 100644
index 000000000..e644305ab
--- /dev/null
+++ b/kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-replication-controller.yaml
@@ -0,0 +1,19 @@
+apiVersion: v1
+kind: ReplicationController
+metadata:
+  name: mock
+spec:
+  replicas: 1
+  selector:
+    app: mock
+  template:
+    metadata:
+      labels:
+        app: mock
+    spec:
+      containers:
+      - name: mock-container
+        image: k8s.gcr.io/pause:2.0
+        ports:
+        - containerPort: 9949
+          protocol: TCP
\ No newline at end of file
diff --git a/kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-service.yaml b/kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-service.yaml
new file mode 100644
index 000000000..2e3a9fa06
--- /dev/null
+++ b/kubernetes/e2e_test/test_yaml/multi-resource/multi-resource-service.yaml
@@ -0,0 +1,13 @@
+apiVersion: v1
+kind: Service
+metadata:
+  name: mock
+  labels:
+    app: mock
+spec:
+  ports:
+  - port: 99
+    protocol: TCP
+    targetPort: 9949
+  selector:
+    app: mock
\ No newline at end of file