From d3793166531b0ddc3f63dc2053b1d7154b234247 Mon Sep 17 00:00:00 2001
From: skudasov <f4hrenh9it@gmail.com>
Date: Sat, 20 Jan 2024 01:18:36 +0100
Subject: [PATCH] labels for automatic chaos testing

---
 .../templates/chainlink-db-deployment.yaml    |  8 ++++++
 .../templates/chainlink-node-deployment.yaml  | 25 ++++++++++++++-----
 .../templates/geth-deployment.yaml            |  8 ++++++
 3 files changed, 35 insertions(+), 6 deletions(-)

diff --git a/charts/chainlink-cluster/templates/chainlink-db-deployment.yaml b/charts/chainlink-cluster/templates/chainlink-db-deployment.yaml
index f335130ea9f..91924ba5005 100644
--- a/charts/chainlink-cluster/templates/chainlink-db-deployment.yaml
+++ b/charts/chainlink-cluster/templates/chainlink-db-deployment.yaml
@@ -24,12 +24,20 @@ spec:
   selector:
     matchLabels:
       app: {{ $.Release.Name }}-db
+      # Used for testing.
+      # havoc-component-group and havoc-network-group are used by "havoc" chaos testing tool
+      havoc-component-group: db
+      havoc-network-group: db
       instance: {{ $cfg.name }}-db
       release: {{ $.Release.Name }}
   template:
     metadata:
       labels:
         app: {{ $.Release.Name }}-db
+        # Used for testing.
+        # havoc-component-group and havoc-network-group are used by "havoc" chaos testing tool
+        havoc-component-group: db
+        havoc-network-group: db
         instance: {{ $cfg.name }}-db
         release: {{ $.Release.Name }}
         {{- range $key, $value := $.Values.labels }}
diff --git a/charts/chainlink-cluster/templates/chainlink-node-deployment.yaml b/charts/chainlink-cluster/templates/chainlink-node-deployment.yaml
index 884cf0e535b..0ce16fd475b 100644
--- a/charts/chainlink-cluster/templates/chainlink-node-deployment.yaml
+++ b/charts/chainlink-cluster/templates/chainlink-node-deployment.yaml
@@ -2,7 +2,7 @@
 apiVersion: apps/v1
 kind: Deployment
 metadata:
-  name: {{ $.Release.Name }}-{{ $cfg.name }}
+  name: {{ if eq $index 0 }}{{ $.Release.Name }}-{{ $cfg.name }}-bootstrap{{ else }}{{ $.Release.Name }}-{{ $cfg.name }}{{ end }}
 spec:
   strategy:
     # Need to recreate the pod to deal with lease lock held by old pod.
@@ -10,18 +10,31 @@ spec:
   selector:
     matchLabels:
       app: {{ $.Release.Name }}
+      # Used for testing.
+      # havoc-component-group and havoc-network-group are used by "havoc" chaos testing tool
+      {{ if eq $index 0 }}{{ else }}
+      havoc-component-group: node
+      {{ end }}
+      {{ if eq $index 0 }}{{ else }}
+      havoc-network-group: {{ if gt $index 2 }}"1"{{ else }}"2"{{ end }}
+      {{ end }}
       instance: {{ $cfg.name }}
       release: {{ $.Release.Name }}
   template:
     metadata:
       labels:
         app: {{ $.Release.Name }}
+        # Used for testing.
+        # havoc-component-group and havoc-network-group are used by "havoc" chaos testing tool
+        {{ if eq $index 0 }}{{ else }}
+        havoc-component-group: node
+        {{ end }}
+        {{ if eq $index 0 }}{{ else }}
+        havoc-network-group: {{ if gt $index 2 }}"1"{{ else }}"2"{{ end }}
+        {{ end }}
+
         instance: {{ $cfg.name }}
         release: {{ $.Release.Name }}
-        # Used for testing. Role value should either be: bootstrap or node.
-        # There should only be one "bootstrap" node, the rest should be "node".
-        # Here we set the first node to be bootstrap, the rest to be node.
-        role: {{ if eq $index 0 }}bootstrap{{ else }}node{{ end }}
         {{- range $key, $value := $.Values.labels }}
         {{ $key }}: {{ $value | quote }}
         {{- end }}
@@ -43,7 +56,7 @@ spec:
             {{- toYaml $.Values.chainlink.securityContext | nindent 12 }}
           image: {{ default "public.ecr.aws/chainlink/chainlink" $cfg.image }}
           imagePullPolicy: Always
-          command: ["bash", "-c", "while ! pg_isready -U postgres --host {{ $.Release.Name }}-db-{{ $cfg.name }} --port 5432; do echo \"waiting for database to start\"; sleep 1; done && chainlink -c /etc/node-secrets-volume/default.toml -c /etc/node-secrets-volume/overrides.toml -secrets /etc/node-secrets-volume/secrets.toml node start -d -p /etc/node-secrets-volume/node-password -a /etc/node-secrets-volume/apicredentials --vrfpassword=/etc/node-secrets-volume/apicredentials"]
+          command: [ "bash", "-c", "while ! pg_isready -U postgres --host {{ $.Release.Name }}-db-{{ $cfg.name }} --port 5432; do echo \"waiting for database to start\"; sleep 1; done && chainlink -c /etc/node-secrets-volume/default.toml -c /etc/node-secrets-volume/overrides.toml -secrets /etc/node-secrets-volume/secrets.toml node start -d -p /etc/node-secrets-volume/node-password -a /etc/node-secrets-volume/apicredentials --vrfpassword=/etc/node-secrets-volume/apicredentials" ]
           ports:
             - name: access
               containerPort: {{ $.Values.chainlink.web_port }}
diff --git a/charts/chainlink-cluster/templates/geth-deployment.yaml b/charts/chainlink-cluster/templates/geth-deployment.yaml
index 8d2d4d3c76c..abc7853d978 100644
--- a/charts/chainlink-cluster/templates/geth-deployment.yaml
+++ b/charts/chainlink-cluster/templates/geth-deployment.yaml
@@ -7,11 +7,19 @@ spec:
   selector:
     matchLabels:
       app: geth
+      # Used for testing.
+      # havoc-component-group and havoc-network-group are used by "havoc" chaos testing tool
+      havoc-component-group: "blockchain"
+      havoc-network-group: "blockchain"
       release: {{ .Release.Name }}
   template:
     metadata:
       labels:
         app: geth
+        # Used for testing.
+        # havoc-component-group and havoc-network-group are used by "havoc" chaos testing tool
+        havoc-component-group: "blockchain"
+        havoc-network-group: "blockchain"
         release: {{ .Release.Name }}
       annotations:
         {{- range $key, $value := .Values.podAnnotations }}