Unittest for get service , daemonset. cronjob and ingress details

test/resources/cronjob.json

@@ -4,6 +4,9 @@
 		"kind": "CronJob",
 		"metadata": {
 			"creationTimestamp": "2021-03-25T04:02:24Z",
+      "labels": {
+            "TestLabel": "mytestlabel"
+        },
 			"managedFields": [
 				{
 					"apiVersion": "batch/v1beta1",
@@ -95,4 +98,4 @@
 		},
 		"status": {}
 	}
-]
+]

test/resources/daemonset.json

@@ -9,8 +9,8 @@
 			"creationTimestamp": "2021-03-25T04:00:10Z",
 			"generation": 1,
 			"labels": {
-				"k8s-app": "fluentd-logging"
-			},
+            "TestLabel": "mytestlabel"
+       },
 			"managedFields": [
 				{
 					"apiVersion": "apps/v1",
@@ -233,4 +233,4 @@
 			"updatedNumberScheduled": 2
 		}
 	}
-]
+]

test/resources/ingress.json

@@ -4,6 +4,9 @@
 		"kind": "Ingress",
 		"metadata": {
 			"creationTimestamp": "2021-03-25T04:04:21Z",
+      "labels": {
+            "TestLabel": "mytestlabel"
+        },
 			"generation": 1,
 			"managedFields": [
 				{
@@ -47,4 +50,4 @@
 			"loadBalancer": {}
 		}
 	}
-]
+]

test/resources/service.json

@@ -4,6 +4,9 @@
 		"kind": "Service",
 		"metadata": {
 			"creationTimestamp": "2021-03-17T12:27:33Z",
+                        "labels": {
+                              "Test": "mytest"
+                         },
 			"managedFields": [
 				{
 					"apiVersion": "v1",
@@ -32,7 +35,7 @@
 					"time": "2021-03-17T12:27:33Z"
 				}
 			],
-			"name": "test-service",
+			"name": "my-service2",
 			"namespace": "default",
 			"resourceVersion": "970",
 			"selfLink": "/api/v1/namespaces/default/services/my-service",
@@ -55,6 +58,4 @@
 		},
 		"status": {
 			"loadBalancer": {}
-		}
-	}
-]
+

test/test_KubeLibrary.py

@@ -177,6 +177,38 @@ def mock_list_namespaced_role_bindings(namespace, watch=False):
             return list_of_role_bind
 
 
+def mock_read_daemonset_details_in_namespace(name, namespace):
+    if namespace == 'default':
+        with open('test/resources/daemonset.json') as json_file:
+            daemonset_details_content = json.load(json_file)
+            read_daemonset_details = AttributeDict({'items': daemonset_details_content})
+            return read_daemonset_details
+
+
+def mock_read_service_details_in_namespace(name, namespace):
+    if namespace == 'default':
+        with open('test/resources/service.json') as json_file:
+            service_details_content = json.load(json_file)
+            read_service_details = AttributeDict({'items': service_details_content})
+            return read_service_details
+
+
+def mock_read_ingress_details_in_namespace(name, namespace):
+    if namespace == 'default':
+        with open('test/resources/ingress.json') as json_file:
+            ingress_details_content = json.load(json_file)
+            read_ingress_details = AttributeDict({'items': ingress_details_content})
+            return read_ingress_details
+
+
+def mock_read_cron_job_details_in_namespace(name, namespace):
+    if namespace == 'default':
+        with open('test/resources/cronjob.json') as json_file:
+            cron_job_details_content = json.load(json_file)
+            read_cron_job_details = AttributeDict({'items': cron_job_details_content})
+            return read_cron_job_details
+
+
 bearer_token = 'eyJhbGciOiJSUzI1NiIsImtpZCI6IjdXVWJMOUdTaDB1TjcyNmF0Sjk4RWlzQ05RaWdSUFoyN004TmlGT1pSX28ifQ.' \
                'eyJpc3MiOiJrdWJlcm5ldGVzL3NlcnZpY2VhY2NvdW50Iiwia3ViZXJuZXRlcy5pby9zZXJ2aWNlYWNjb3VudC9uYW1' \
                'lc3BhY2UiOiJkZWZhdWx0Iiwia3ViZXJuZXRlcy5pby9zZXJ2aWNlYWNjb3VudC9zZWNyZXQubmFtZSI6Im15c2EtdG' \
@@ -404,6 +436,34 @@ def test_get_role_bindings_in_namespace(self, mock_lnp):
         role_bindings = kl.get_role_bindings_in_namespace('default')
         self.assertEqual(['read-pods'], [item for item in role_bindings])
 
+    @mock.patch('kubernetes.client.AppsV1Api.read_namespaced_daemon_set')
+    def test_get_daemonset_details_in_namespace(self, mock_lnp):
+        mock_lnp.side_effect = mock_read_daemonset_details_in_namespace
+        kl = KubeLibrary(kube_config='test/resources/k3d')
+        daemonset_details = kl.get_daemonset_details_in_namespace('fluentd-elasticsearch', 'default')
+        self.assertEqual('mytestlabel', daemonset_details.items[0].metadata.labels.TestLabel)
+
+    @mock.patch('kubernetes.client.CoreV1Api.read_namespaced_service')
+    def test_get_service_details_in_namespace(self, mock_lnp):
+        mock_lnp.side_effect = mock_read_service_details_in_namespace
+        kl = KubeLibrary(kube_config='test/resources/k3d')
+        service_details = kl.get_service_details_in_namespace('test-service', 'default')
+        self.assertEqual('mytest', service_details.items[0].metadata.labels.Test)
+
+    @mock.patch('kubernetes.client.ExtensionsV1beta1Api.read_namespaced_ingress')
+    def test_get_ingress_details_in_namespace(self, mock_lnp):
+        mock_lnp.side_effect = mock_read_ingress_details_in_namespace
+        kl = KubeLibrary(kube_config='test/resources/k3d')
+        ingress_details = kl.get_ingress_details_in_namespace('max-ingress', 'default')
+        self.assertEqual('mytestlabel', ingress_details.items[0].metadata.labels.TestLabel)
+
+    @mock.patch('kubernetes.client.BatchV1beta1Api.read_namespaced_cron_job')
+    def test_get_cron_job_details_in_namespace(self, mock_lnp):
+        mock_lnp.side_effect = mock_read_cron_job_details_in_namespace
+        kl = KubeLibrary(kube_config='test/resources/k3d')
+        cron_job_details = kl.get_cron_job_details_in_namespace('hello', 'default')
+        self.assertEqual('mytestlabel', cron_job_details.items[0].metadata.labels.TestLabel)
+
     @mock.patch('kubernetes.client.AppsV1Api.list_namespaced_deployment')
     def test_get_deployments_in_namespace(self, mock_lnp):
         mock_lnp.side_effect = mock_list_namespaced_deployments