add support for check_mode for modules k8s_scale and k8s_rollback (#255)

k8s_scale, k8s_rollback - add support for check_mode 

SUMMARY

closes #243 and #244

ISSUE TYPE


Bugfix Pull Request

COMPONENT NAME

k8s_scale
k8s_rollback
ADDITIONAL INFORMATION

Reviewed-by: Alina Buzachis <None>
Reviewed-by: None <None>
Reviewed-by: Mike Graves <mgraves@redhat.com>

plugins/modules/k8s_rollback.py

@@ -115,11 +115,13 @@ def execute_module(module, k8s_ansible_mixin):
         module.params["field_selectors"],
     )
 
+    changed = False
     for resource in resources["resources"]:
         result = perform_action(module, k8s_ansible_mixin, resource)
+        changed = result["changed"] or changed
         results.append(result)
 
-    module.exit_json(**{"changed": True, "rollback_info": results})
+    module.exit_json(**{"changed": changed, "rollback_info": results})
 
 
 def perform_action(module, k8s_ansible_mixin, resource):
@@ -143,6 +145,13 @@ def perform_action(module, k8s_ansible_mixin, resource):
     prev_managed_resource = get_previous_revision(
         managed_resources["resources"], current_revision
     )
+    if not prev_managed_resource:
+        warn = "No rollout history found for resource %s/%s" % (
+            module.params["kind"],
+            resource["metadata"]["name"],
+        )
+        result = {"changed": False, "warnings": [warn]}
+        return result
 
     if module.params["kind"] == "Deployment":
         del prev_managed_resource["spec"]["template"]["metadata"]["labels"][
@@ -174,22 +183,24 @@ def perform_action(module, k8s_ansible_mixin, resource):
         api_target = "daemonsets"
         content_type = "application/strategic-merge-patch+json"
 
-    rollback = k8s_ansible_mixin.client.request(
-        "PATCH",
-        "/apis/{0}/namespaces/{1}/{2}/{3}".format(
-            module.params["api_version"],
-            module.params["namespace"],
-            api_target,
-            module.params["name"],
-        ),
-        body=resource_patch,
-        content_type=content_type,
-    )
+    rollback = resource
+    if not module.check_mode:
+        rollback = k8s_ansible_mixin.client.request(
+            "PATCH",
+            "/apis/{0}/namespaces/{1}/{2}/{3}".format(
+                module.params["api_version"],
+                module.params["namespace"],
+                api_target,
+                module.params["name"],
+            ),
+            body=resource_patch,
+            content_type=content_type,
+        ).to_dict()
 
     result = {"changed": True}
     result["method"] = "patch"
     result["body"] = resource_patch
-    result["resources"] = rollback.to_dict()
+    result["resources"] = rollback
     return result
 
 

plugins/modules/k8s_scale.py

@@ -236,7 +236,7 @@ def execute_module(
             module.exit_json(warning=warn, **return_attributes)
 
     for existing in existing_items:
-        if module.params["kind"] == "job":
+        if module.params["kind"].lower() == "job":
             existing_count = existing.spec.parallelism
         elif hasattr(existing.spec, "replicas"):
             existing_count = existing.spec.replicas
@@ -271,22 +271,25 @@ def execute_module(
             continue
 
         if existing_count != replicas:
-            if not module.check_mode:
-                if module.params["kind"] == "job":
-                    existing.spec.parallelism = replicas
-                    result = resource.patch(existing.to_dict()).to_dict()
+            if module.params["kind"].lower() == "job":
+                existing.spec.parallelism = replicas
+                result = {"changed": True}
+                if module.check_mode:
+                    result["result"] = existing.to_dict()
                 else:
-                    result = scale(
-                        module,
-                        k8s_ansible_mixin,
-                        resource,
-                        existing,
-                        replicas,
-                        wait,
-                        wait_time,
-                        wait_sleep,
-                    )
-                    changed = changed or result["changed"]
+                    result["result"] = resource.patch(existing.to_dict()).to_dict()
+            else:
+                result = scale(
+                    module,
+                    k8s_ansible_mixin,
+                    resource,
+                    existing,
+                    replicas,
+                    wait,
+                    wait_time,
+                    wait_sleep,
+                )
+                changed = changed or result["changed"]
         else:
             name = existing.metadata.name
             namespace = existing.metadata.namespace
@@ -342,25 +345,34 @@ def scale(
 
     existing = resource.get(name=name, namespace=namespace)
 
-    try:
-        resource.scale.patch(body=scale_obj)
-    except Exception as exc:
-        module.fail_json(msg="Scale request failed: {0}".format(exc))
-
-    k8s_obj = resource.get(name=name, namespace=namespace).to_dict()
-    match, diffs = k8s_ansible_mixin.diff_objects(existing.to_dict(), k8s_obj)
     result = dict()
-    result["result"] = k8s_obj
+    if module.check_mode:
+        k8s_obj = copy.deepcopy(existing.to_dict())
+        k8s_obj["spec"]["replicas"] = replicas
+        match, diffs = k8s_ansible_mixin.diff_objects(existing.to_dict(), k8s_obj)
+        if wait:
+            result["duration"] = 0
+        result["result"] = k8s_obj
+    else:
+        try:
+            resource.scale.patch(body=scale_obj)
+        except Exception as exc:
+            module.fail_json(msg="Scale request failed: {0}".format(exc))
+
+        k8s_obj = resource.get(name=name, namespace=namespace).to_dict()
+        result["result"] = k8s_obj
+        if wait and not module.check_mode:
+            success, result["result"], result["duration"] = k8s_ansible_mixin.wait(
+                resource, scale_obj, wait_sleep, wait_time
+            )
+            if not success:
+                module.fail_json(msg="Resource scaling timed out", **result)
+
+    match, diffs = k8s_ansible_mixin.diff_objects(existing.to_dict(), k8s_obj)
     result["changed"] = not match
     if module._diff:
         result["diff"] = diffs
 
-    if wait:
-        success, result["result"], result["duration"] = k8s_ansible_mixin.wait(
-            resource, scale_obj, wait_sleep, wait_time
-        )
-        if not success:
-            module.fail_json(msg="Resource scaling timed out", **result)
     return result