fix(shrinkCluster): ensure we are injecting disable before shrink (#3194

)
spinnaker · Sep 24, 2019 · 9aa603a · 9aa603a
1 parent c104a23
commit 9aa603a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...naker/orca/clouddriver/pipeline/cluster/AbstractClusterWideClouddriverOperationStage.java b/...naker/orca/clouddriver/pipeline/cluster/AbstractClusterWideClouddriverOperationStage.java
@@ -58,6 +58,16 @@ protected static String getStepName(String taskClassSimpleName) {
     return taskClassSimpleName;
   }
 
+  @Override
+  public final void beforeStages(@Nonnull Stage parent, @Nonnull StageGraphBuilder graph) {
+    addAdditionalBeforeStages(parent, graph);
+  }
+
+  @Override
+  public final void afterStages(@Nonnull Stage parent, @Nonnull StageGraphBuilder graph) {
+    addAdditionalAfterStages(parent, graph);
+  }
+
   protected void addAdditionalBeforeStages(
       @Nonnull Stage parent, @Nonnull StageGraphBuilder graph) {}