add extra_hooks job to existing tests

cognifloyd · cognifloyd · commit c24a4c7ed0a5 · 2022-03-18T15:02:41.000-05:00
diff --git a/tests/unit/custom_annotations_test.yaml b/tests/unit/custom_annotations_test.yaml
@@ -153,13 +153,19 @@ tests:
         annotations:
           foo: bar
           answer: "42"
+        extra_hooks:
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
           # job-st2-apply-rbac-defintions
           # job-st2-apikey-load
           # job-st2-key-load
           # job-st2-register-content
+          # extra_hooks job
 
       # job annotations
       - isNotNull:
diff --git a/tests/unit/dns_test.yaml b/tests/unit/dns_test.yaml
@@ -26,6 +26,12 @@ tests:
       st2: 
         packs: { sensors: [] } # ensure only 1 sensor
         rbac: { enabled: true } # enable rbac job
+      jobs:
+        extra_hooks: &jobs_extra_hooks
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     asserts:
       - isNull:
           path: spec.template.spec.dnsPolicy
@@ -51,6 +57,8 @@ tests:
       st2: 
         packs: { sensors: [] } # ensure only 1 sensor
         rbac: { enabled: true } # enable rbac job
+      jobs:
+        extra_hooks: *jobs_extra_hooks
     asserts:
       - equal:
           path: spec.template.spec.dnsPolicy
diff --git a/tests/unit/env_test.yaml b/tests/unit/env_test.yaml
@@ -66,9 +66,15 @@ tests:
     set:
       st2:
         rbac: { enabled: true } # enable rbac job
+      jobs:
+        extra_hooks: &extra_hooks_jobs
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
       - isNull: *is_null_env
 
   - it: Deployments accept custom env
@@ -121,9 +127,10 @@ tests:
         rbac: { enabled: true } # enable rbac job
       jobs:
         env: *env
+        extra_hooks: *extra_hooks_jobs
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
 
       - contains: *contains_env
 
@@ -156,11 +163,13 @@ tests:
     set:
       st2:
         rbac: { enabled: true } # enable rbac job
+      jobs:
+        extra_hooks: *extra_hooks_jobs
     release:
       name: st2ha
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
 
       - contains: *contains_st2_urls
         documentIndex: 1
@@ -215,6 +224,7 @@ tests:
         rbac: { enabled: true } # enable rbac job
       jobs:
         envFromSecrets: *envFromSecrets
+        extra_hooks: *extra_hooks_jobs
     asserts:
       - contains: *contains_external_secret1
       - contains: *contains_external_secret2
diff --git a/tests/unit/image_pull_test.yaml b/tests/unit/image_pull_test.yaml
@@ -44,6 +44,7 @@ tests:
         # job-st2-apikey-load
         # job-st2-key-load
         # job-st2-register-content
+        # extra_hooks job
     set:
       # image.pullPolicy defaults to IfNotPresent
       # image.pullSecret defaults to None
@@ -56,6 +57,12 @@ tests:
         packs: { sensors: [] } # ensure only 1 sensor
       st2chatops:
         enabled: true
+      jobs:
+        extra_hooks: &jobs_extra_hooks
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     asserts:
       - isNull:
           path: spec.template.spec.imagePullSecrets
@@ -84,6 +91,8 @@ tests:
         packs: { sensors: [] } # ensure only 1 sensor
       st2chatops:
         enabled: true
+      jobs:
+        extra_hooks: *jobs_extra_hooks
     asserts:
       - equal:
           path: spec.template.spec.imagePullSecrets[0].name
diff --git a/tests/unit/labels_test.yaml b/tests/unit/labels_test.yaml
@@ -101,13 +101,20 @@ tests:
       st2:
         rbac:
           enabled: true # enable rbac job
+      jobs:
+        extra_hooks:
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
           # job-st2-apply-rbac-defintions
           # job-st2-apikey-load
           # job-st2-key-load
           # job-st2-register-content
+          # extra_hooks job
 
       # unlike deployments, jobs should not have selector.matchLabels
 
diff --git a/tests/unit/packs_volumes_test.yaml b/tests/unit/packs_volumes_test.yaml
@@ -140,11 +140,17 @@ tests:
           volumes:
             enabled: false
           configs: {} # has one core.yaml config file by default (dicts get merged)
+      jobs:
+        extra_hooks: &extra_hooks_jobs
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     release:
       name: st2ha
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
 
       - notContains: *packs_vol
       - notContains: *venvs_vol
@@ -393,11 +399,13 @@ tests:
           volumes:
             enabled: false
           configs: {} # has one core.yaml config file by default (dicts get merged)
+      jobs:
+        extra_hooks: *extra_hooks_jobs
     release:
       name: st2ha
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
 
       - contains: *packs_vol
         documentIndex: 3 # register_content
@@ -687,11 +695,13 @@ tests:
             virtualenvs: *nfs_venvs_volume
             configs: *nfs_pack_configs_volume
           configs: {} # has one core.yaml config file by default (dicts get merged)
+      jobs:
+        extra_hooks: *extra_hooks_jobs
     release:
       name: st2ha
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
 
       - contains: *nfs_packs_vol
         documentIndex: 3 # register_content
@@ -960,11 +970,13 @@ tests:
             virtualenvs: *nfs_venvs_volume
             configs: *nfs_pack_configs_volume
           configs: {} # has one core.yaml config file by default (dicts get merged)
+      jobs:
+        extra_hooks: *extra_hooks_jobs
     release:
       name: st2ha
     asserts:
       - hasDocuments:
-          count: 4
+          count: 5
 
       - contains: *nfs_packs_vol
         documentIndex: 3 # register_content
diff --git a/tests/unit/placement_test.yaml b/tests/unit/placement_test.yaml
@@ -39,6 +39,12 @@ tests:
         packs: { sensors: [] } # ensure only 1 sensor
       st2chatops:
         enabled: true
+      jobs:
+        extra_hooks: &jobs_extra_hooks
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
     asserts:
       - isNull:
           path: spec.template.spec.nodeSelector
@@ -157,6 +163,7 @@ tests:
         nodeSelector: *custom_nodeSelector
         tolerations: *custom_tolerations
         affinity: *custom_affinity
+        extra_hooks: *jobs_extra_hooks
     asserts:
       - equal:
           path: spec.template.spec.nodeSelector
diff --git a/tests/unit/security_context_test.yaml b/tests/unit/security_context_test.yaml
@@ -33,12 +33,19 @@ tests:
         # job-st2-apikey-load
         # job-st2-key-load
         # job-st2-register-content
+        # extra_hooks job
     set:
       st2chatops:
         enabled: true
       st2:
         packs: { sensors: [] } # ensure only 1 sensor
         rbac: { enabled: true } # enable rbac job
+      jobs:
+        extra_hooks: &jobs_extra_hooks
+          - name: upgrade-warning
+            hook: pre-upgrade, pre-rollback
+            hook_weight: -5
+            command: ["st2", "run", "--tail", "custom_pack.warn_about_upgrade"]
 
       podSecurityContext: {}
       securityContext: {}
@@ -75,6 +82,8 @@ tests:
       st2:
         packs: { sensors: [] } # ensure only 1 sensor
         rbac: { enabled: true } # enable rbac job
+      jobs:
+        extra_hooks: *jobs_extra_hooks
 
       podSecurityContext: &global_pod_security_context
         fsGroup: 1234
