Run staging-setup tagged unit tests on initial staging deployment before upgrade

1 files changed, 4 insertions, 2 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/deployment/TesterCloud.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/deployment/TesterCloud.java
index f0972643bd5..ef97ef27a72 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/deployment/TesterCloud.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/deployment/TesterCloud.java
@@ -56,13 +56,15 @@ public interface TesterCloud {
 
         system,
 
+        staging_setup,
+
         staging,
 
         production;
 
-        public static Suite of(JobType type) {
+        public static Suite of(JobType type, boolean isSetup) {
             if (type == JobType.systemTest) return system;
-            if (type == JobType.stagingTest) return  staging;
+            if (type == JobType.stagingTest) return isSetup ? staging_setup : staging;
             if (type.isProduction()) return production;
             throw new AssertionError("Unknown JobType '" + type + "'!");
         }