From f452d0cef62f829bbfe0d3d6e811668e38dbeb8c Mon Sep 17 00:00:00 2001
From: Haowei Cai <haoweic@google.com>
Date: Fri, 21 Feb 2020 17:37:44 -0800
Subject: [PATCH] add basic watch configmap e2e test

---
 kubernetes/e2e_test/test_watch.py | 94 +++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 kubernetes/e2e_test/test_watch.py

diff --git a/kubernetes/e2e_test/test_watch.py b/kubernetes/e2e_test/test_watch.py
new file mode 100644
index 0000000000..3530b8bbdf
--- /dev/null
+++ b/kubernetes/e2e_test/test_watch.py
@@ -0,0 +1,94 @@
+# -*- 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
+import uuid
+
+from kubernetes import watch
+from kubernetes.client import api_client
+from kubernetes.client.api import core_v1_api
+from kubernetes.e2e_test import base
+
+
+def short_uuid():
+    id = str(uuid.uuid4())
+    return id[-12:]
+
+
+def config_map_with_value(name, value):
+    return {
+        "apiVersion": "v1",
+        "kind": "ConfigMap",
+        "metadata": {
+            "name": name,
+        },
+        "data": {
+            "key": value,
+            "config": "dummy",
+        }
+    }
+
+
+class TestClient(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.config = base.get_e2e_configuration()
+
+    def test_watch_configmaps(self):
+        client = api_client.ApiClient(configuration=self.config)
+        api = core_v1_api.CoreV1Api(client)
+
+        # create a configmap
+        name_a = 'configmap-a-' + short_uuid()
+        configmap_a = config_map_with_value(name_a, "a")
+        api.create_namespaced_config_map(
+            body=configmap_a, namespace='default')
+
+        # list all configmaps and extract the resource version
+        resp = api.list_namespaced_config_map('default')
+        rv = resp.metadata.resource_version
+
+        # create another configmap
+        name_b = 'configmap-b-' + short_uuid()
+        configmap_b = config_map_with_value(name_b, "b")
+        api.create_namespaced_config_map(
+            body=configmap_b, namespace='default')
+
+        # patch configmap b
+        configmap_b['data']['config'] = "{}"
+        api.patch_namespaced_config_map(
+            name=name_b, namespace='default', body=configmap_b)
+
+        # delete all configmaps
+        api.delete_collection_namespaced_config_map(
+            namespace='default')
+
+        w = watch.Watch()
+        # expect to observe all events happened after the initial LIST
+        expect = ['ADDED', 'MODIFIED', 'DELETED', 'DELETED']
+        i = 0
+        # start watching with the resource version we got from the LIST
+        for event in w.stream(api.list_namespaced_config_map,
+                              namespace='default',
+                              resource_version=rv,
+                              timeout_seconds=5):
+            self.assertEqual(event['type'], expect[i])
+            # Kubernetes doesn't guarantee the order of the two objects
+            # being deleted
+            if i < 2:
+                self.assertEqual(event['object'].metadata.name, name_b)
+            i = i + 1
+
+        self.assertEqual(i, 4)