action rollout deployment implementation (#144)

* action rollout deployment implementation

---------

Signed-off-by: Jorge Tapicha <jitapichab@gmail.com>

Author: jitapichab

CHANGELOG.md

@@ -6,6 +6,10 @@
 
 ## [Unreleased][]
 
+### Added
+
+* Rollout deployment action `chaosk8s.deployment.actions.rollout_deployment`
+
 [Unreleased]: https://github.com/chaostoolkit/chaostoolkit-kubernetes/compare/0.26.4...HEAD
 
 

chaosk8s/deployment/actions.py

@@ -1,3 +1,4 @@
+import datetime
 import json
 import os.path
 
@@ -10,7 +11,12 @@
 
 from chaosk8s import create_k8s_api_client
 
-__all__ = ["create_deployment", "delete_deployment", "scale_deployment"]
+__all__ = [
+    "create_deployment",
+    "delete_deployment",
+    "scale_deployment",
+    "rollout_deployment",
+]
 
 
 def create_deployment(spec_path: str, ns: str = "default", secrets: Secrets = None):
@@ -82,3 +88,33 @@ def scale_deployment(
         raise ActivityFailed(
             f"failed to scale '{name}' to {replicas} replicas: {str(e)}"
         )
+
+
+def rollout_deployment(
+    name: str = None,
+    ns: str = "default",
+    secrets: Secrets = None,
+):
+    """
+    Rolling the deployment. The `name` is the name of the deployment.
+    """
+    api = create_k8s_api_client(secrets)
+
+    v1 = client.AppsV1Api(api)
+
+    now = datetime.datetime.now(datetime.timezone.utc)
+    now = str(now.isoformat("T") + "Z")
+    body = {
+        "spec": {
+            "template": {
+                "metadata": {"annotations": {"kubectl.kubernetes.io/restartedAt": now}}
+            }
+        }
+    }
+
+    try:
+        v1.patch_namespaced_deployment(name=name, namespace=ns, body=body)
+    except ApiException as e:
+        raise ActivityFailed(
+            f"failed to rollout the deployment '{name}'! Error: {str(e)}"
+        )

tests/test_deployment.py

@@ -1,3 +1,4 @@
+from datetime import datetime, timezone
 from unittest.mock import ANY, MagicMock, call, patch
 
 import pytest
@@ -8,6 +9,7 @@
 from chaosk8s.deployment.actions import (
     create_deployment,
     delete_deployment,
+    rollout_deployment,
     scale_deployment,
 )
 from chaosk8s.deployment.probes import (
@@ -183,3 +185,37 @@ def test_deployment_is_not_fully_available_when_it_should(cl, client, watch, has
     with pytest.raises(ActivityFailed) as x:
         deployment_fully_available("mysvc")
     assert "deployment 'mysvc' failed to recover within" in str(x.value)
+
+
+def _generate_mock_time():
+    return datetime(2023, 6, 7, 10, 30, 0, tzinfo=timezone.utc)
+
+
+@patch("chaosk8s.deployment.actions.create_k8s_api_client", autospec=True)
+@patch("chaosk8s.deployment.actions.client", autospec=True)
+def test_rollout_deployment(client, api):
+    v1 = MagicMock()
+    client.AppsV1Api.return_value = v1
+    mock_now = _generate_mock_time()
+
+    with patch("datetime.datetime") as mock_time:
+        mock_time.now.return_value = mock_now
+        mock_now_str = f'{mock_now.isoformat("T")}Z'
+
+        body = {
+            "spec": {
+                "template": {
+                    "metadata": {
+                        "annotations": {
+                            "kubectl.kubernetes.io/restartedAt": mock_now_str
+                        }
+                    }
+                }
+            }
+        }
+
+        rollout_deployment("fake", "fake_ns")
+
+        v1.patch_namespaced_deployment.assert_called_with(
+            name="fake", namespace="fake_ns", body=body
+        )