From f56b05fffee1b5e7f8815e500d4b0a5c8a48f1e0 Mon Sep 17 00:00:00 2001
From: Chin-Ya Huang <chin-ya.huang@suse.com>
Date: Wed, 22 Mar 2023 13:23:03 +0800
Subject: [PATCH] test(support-bundle): agent with taint toleration

Ref: 5614

Signed-off-by: Chin-Ya Huang <chin-ya.huang@suse.com>
---
 manager/integration/tests/node.py             | 56 +++++++++++++++++++
 .../integration/tests/test_support_bundle.py  | 47 +++++++++++++++-
 2 files changed, 102 insertions(+), 1 deletion(-)
 create mode 100644 manager/integration/tests/node.py

diff --git a/manager/integration/tests/node.py b/manager/integration/tests/node.py
new file mode 100644
index 0000000000..99883c742b
--- /dev/null
+++ b/manager/integration/tests/node.py
@@ -0,0 +1,56 @@
+import pytest
+
+from kubernetes import client as k8sclient
+
+from common import get_self_host_id
+
+
+@pytest.fixture(
+    params=[("foo/bar", "test", "NoSchedule"),
+            ("foo", "", "NoSchedule")]
+)
+def taint_nodes_exclude_self(request):
+    taint = k8sclient.V1Taint(
+        key=request.param[0],
+        value=request.param[1],
+        effect=request.param[2],
+    )
+
+    self_host_id = get_self_host_id()
+
+    api = k8sclient.CoreV1Api()
+    node_items = api.list_node().items
+    saved_nodes = []
+    for node in node_items:
+        if node.metadata.name == self_host_id:
+            continue
+
+        saved_nodes.append(node)
+
+        if node.spec.taints is None:
+            taints = [taint]
+        else:
+            taints = node.spec.taints + [taint]
+        payload = {
+            "spec": {
+                "taints": taints
+            }
+        }
+        api.patch_node(node.metadata.name, body=payload)
+
+    yield saved_nodes
+
+    for node in saved_nodes:
+        if node.metadata.name == self_host_id:
+            continue
+
+        if node.spec.taints is None:
+            taints = []
+        else:
+            taints = [node.spec.taints]
+        payload = {
+            "spec": {
+                "taints": taints
+            }
+        }
+        api.patch_node(node.metadata.name, body=payload)
diff --git a/manager/integration/tests/test_support_bundle.py b/manager/integration/tests/test_support_bundle.py
index cfa2346c2b..30f643f74a 100644
--- a/manager/integration/tests/test_support_bundle.py
+++ b/manager/integration/tests/test_support_bundle.py
@@ -1,10 +1,11 @@
 import pytest
-import time
 import os
 import zipfile
 
 from tempfile import TemporaryDirectory
 
+from node import taint_nodes_exclude_self  # NOQA
+
 from common import apps_api  # NOQA
 from common import client  # NOQA
 from common import core_api  # NOQA
@@ -21,6 +22,7 @@
 
 from common import SETTING_NODE_SELECTOR
 from common import SETTING_SUPPORT_BUNDLE_FAILED_LIMIT
+from common import SETTING_TAINT_TOLERATION
 
 
 @pytest.mark.support_bundle   # NOQA
@@ -217,3 +219,46 @@ def check_bundled_nodes_matches(node_names, zip, temp_dir):
         f'Nodes zipped in bundle do not match. \n' \
         f'Expect = {expect_node_zips}\n' \
         f'Got = {node_zips}\n'
+
+
+@pytest.mark.support_bundle   # NOQA
+def test_support_bundle_agent_with_taint_toleration(client, taint_nodes_exclude_self):  # NOQA
+    """
+    Scenario: support bundle agent should respect taint toleration
+
+    Issue: https://github.com/longhorn/longhorn/issues/5614
+
+    Given there are some tainted nodes in the cluster
+    And Longhorn tolerates the tainted nodes with setting "taint-toleration"
+
+    When a support bundle is generated
+
+    Then should be able to download the support bundle successfully
+    And support bundle should include all tainted nodes in node collection
+
+    """
+    # The taint-toleration is set up to match the "taint_nodes_exclude_self"
+    # fixture.
+    update_setting(client, SETTING_TAINT_TOLERATION,
+                   "foo/bar=test:NoSchedule; foo:NoSchedule")
+
+    resp = create_support_bundle(client)
+    node_id = resp['id']
+    name = resp['name']
+
+    wait_for_support_bundle_state("ReadyForDownload", node_id, name, client)
+
+    # The temporary directory will be automatically deleted outside of the
+    # "with" context manager.
+    with TemporaryDirectory(prefix="supportbundle-") as temp_dir:
+        download_path = f'{temp_dir}/{0}.zip'.format(name)
+        download_support_bundle(node_id, name, client,
+                                target_path=download_path)
+
+        with zipfile.ZipFile(download_path, 'r') as zip:
+            node_list = client.list_node()
+            node_names = [f"{node.name}" for node in node_list]
+            check_bundled_nodes_matches(node_names, zip, temp_dir)
+
+    wait_for_support_bundle_cleanup(client)
+    check_all_support_bundle_managers_deleted()