Implement tests for VSRoutes

Refactor old code to get rid of unused common yamls

Author: tellet

tests/data/common/backend2-svc.yaml

@@ -0,0 +1,18 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServerRoute
+metadata:
+  name: backends
+spec:
+  host: virtual-server-route.example.com
+  upstreams:
+  - name: backend1
+    service: backend1-svc
+    port: 80
+  - name: backend3
+    service: backend3-svc
+    port: 80
+  subroutes:
+  - path: "/backends/updated-backend1"
+    upstream: backend1
+  - path: "/backends/updated-backend3"
+    upstream: backend3

tests/data/common/backend2.yaml

@@ -0,0 +1,18 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServerRoute
+metadata:
+  name: backends
+spec:
+  host: virtual-server-route.example.com
+  upstreams:
+  - name: backend1
+    service: backend1-svc
+    port: 80
+  - name: backend3
+    service: backend3-svc
+    port: 80
+  subroutes:
+  - path: "/backends/backend1"
+    upstream: backend1
+  - path: "/backends/backend3"
+    upstream: backend3

tests/data/virtual-server-route/route-multiple-updated.yaml

@@ -0,0 +1,13 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServerRoute
+metadata:
+  name: backend-orphan
+spec:
+  host: virtual-server-route.example.com
+  upstreams:
+  - name: backend2
+    service: backend2-svc
+    port: 80
+  subroutes:
+  - path: "/alone-backend2"
+    upstream: backend2

tests/data/virtual-server-route/route-multiple.yaml

@@ -0,0 +1,15 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServerRoute
+metadata:
+  name: backend2
+spec:
+  host: virtual-server-route.example.com
+  upstreams:
+  - name: backend2
+    service: backend2-svc
+    port: 80
+  subroutes:
+  - path: "/backend2"
+    upstream: backend2
+  - path: "/backend2"
+    upstream: backend2

tests/data/virtual-server-route/route-orphan.yaml

@@ -0,0 +1,13 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServerRoute
+metadata:
+  name: backend2
+spec:
+  host: invalid-host.example.com
+  upstreams:
+  - name: backend2
+    service: backend2-svc
+    port: 80
+  subroutes:
+  - path: "/backend2"
+    upstream: backend2

tests/data/virtual-server-route/route-single-duplicate-path.yaml

@@ -0,0 +1,13 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServerRoute
+metadata:
+  name: backend2
+spec:
+  host: virtual-server-route.example.com
+  upstreams:
+  - name: backend2
+    service: backend2-svc
+    port: 80
+  subroutes:
+  - path: "/backend2"
+    upstream: backend2

tests/data/virtual-server-route/route-single-invalid-host.yaml

@@ -0,0 +1,11 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: VirtualServer
+metadata:
+  name: virtual-server-route
+spec:
+  host: virtual-server-route.example.com
+  routes:
+  - path: "/backends"
+    route: backends-namespace/backends
+  - path: "/backend2"
+    route: backend2-namespace/backend2

tests/data/virtual-server-route/route-single.yaml

@@ -8,28 +8,30 @@
 from suite.resources_utils import ensure_item_removal, get_file_contents
 
 
-def create_crd_from_yaml(api_extensions_v1_beta1: ApiextensionsV1beta1Api, yaml_manifest) -> str:
+def create_crds_from_yaml(api_extensions_v1_beta1: ApiextensionsV1beta1Api, yaml_manifest) -> []:
     """
     Create a CRD based on yaml file.
 
     :param api_extensions_v1_beta1: ApiextensionsV1beta1Api
     :param yaml_manifest: an absolute path to file
-    :return: str
+    :return: []
     """
-    print("Create CRD:")
+    print("Create a CRD:")
+    names = []
     with open(yaml_manifest) as f:
-        dep = yaml.safe_load(f)
-
-    try:
-        api_extensions_v1_beta1.create_custom_resource_definition(dep)
-    except Exception as ex:
-        # https://github.com/kubernetes-client/python/issues/376
-        if ex.args[0] == 'Invalid value for `conditions`, must not be `None`':
-            print("There was an insignificant exception during the CRD creation. Continue...")
-        else:
-            pytest.fail(f"An unexpected exception {ex} occurred. Exiting...")
-    print(f"CRD created with name '{dep['metadata']['name']}'")
-    return dep['metadata']['name']
+        docs = yaml.safe_load_all(f)
+        for dep in docs:
+            try:
+                api_extensions_v1_beta1.create_custom_resource_definition(dep)
+            except Exception as ex:
+                # https://github.com/kubernetes-client/python/issues/376
+                if ex.args[0] == 'Invalid value for `conditions`, must not be `None`':
+                    print("There was an insignificant exception during the CRD creation. Continue...")
+                else:
+                    pytest.fail(f"An unexpected exception {ex} occurred. Exiting...")
+            names.append(dep['metadata']['name'])
+            print(f"CRD created with name '{dep['metadata']['name']}'")
+        return names
 
 
 def delete_crd(api_extensions_v1_beta1: ApiextensionsV1beta1Api, name) -> None:
@@ -85,7 +87,7 @@ def delete_virtual_server(custom_objects: CustomObjectsApi, name, namespace) ->
 
 def patch_virtual_server_from_yaml(custom_objects: CustomObjectsApi, name, yaml_manifest, namespace) -> None:
     """
-    Update a VS based on yaml manifest
+    Patch a VS based on yaml manifest
 
     :param custom_objects: CustomObjectsApi
     :param name:
@@ -101,6 +103,24 @@ def patch_virtual_server_from_yaml(custom_objects: CustomObjectsApi, name, yaml_
     print(f"VirtualServer updated with name '{dep['metadata']['name']}'")
 
 
+def patch_v_s_route_from_yaml(custom_objects: CustomObjectsApi, name, yaml_manifest, namespace) -> None:
+    """
+    Update a VirtualServerRoute based on yaml manifest
+
+    :param custom_objects: CustomObjectsApi
+    :param name:
+    :param yaml_manifest: an absolute path to file
+    :param namespace:
+    :return:
+    """
+    print(f"Update a VirtualServerRoute: {name}")
+    with open(yaml_manifest) as f:
+        dep = yaml.safe_load(f)
+
+    custom_objects.patch_namespaced_custom_object("k8s.nginx.org", "v1alpha1", namespace, "virtualserverroutes", name, dep)
+    print(f"VirtualServerRoute updated with name '{dep['metadata']['name']}'")
+
+
 def get_vs_nginx_template_conf(v1: CoreV1Api, vs_namespace, vs_name, pod_name, pod_namespace) -> str:
     """
     Get contents of /etc/nginx/conf.d/vs_{namespace}_{vs_name}.conf in the pod.
@@ -114,3 +134,39 @@ def get_vs_nginx_template_conf(v1: CoreV1Api, vs_namespace, vs_name, pod_name, p
     """
     file_path = f"/etc/nginx/conf.d/vs_{vs_namespace}_{vs_name}.conf"
     return get_file_contents(v1, file_path, pod_name, pod_namespace)
+
+
+def create_v_s_route_from_yaml(custom_objects: CustomObjectsApi, yaml_manifest, namespace) -> str:
+    """
+    Create a VirtualServerRoute based on a yaml file.
+
+    :param custom_objects: CustomObjectsApi
+    :param yaml_manifest: an absolute path to a file
+    :param namespace:
+    :return: str
+    """
+    print("Create a VirtualServerRoute:")
+    with open(yaml_manifest) as f:
+        dep = yaml.safe_load(f)
+
+    custom_objects.create_namespaced_custom_object("k8s.nginx.org", "v1alpha1", namespace, "virtualserverroutes", dep)
+    print(f"VirtualServerRoute created with a name '{dep['metadata']['name']}'")
+    return dep['metadata']['name']
+
+
+def delete_v_s_route(custom_objects: CustomObjectsApi, name, namespace) -> None:
+    """
+    Delete a VirtualServerRoute.
+
+    :param custom_objects: CustomObjectsApi
+    :param namespace: namespace
+    :param name:
+    :return:
+    """
+    print(f"Delete a VirtualServerRoute: {name}")
+    delete_options = client.V1DeleteOptions()
+    custom_objects.delete_namespaced_custom_object("k8s.nginx.org",
+                                                   "v1alpha1", namespace, "virtualserverroutes", name, delete_options)
+    ensure_item_removal(custom_objects.get_namespaced_custom_object,
+                        "k8s.nginx.org", "v1alpha1", namespace, "virtualserverroutes", name)
+    print(f"VirtualServerRoute was removed with the name '{name}'")