Revert the autoscaler code to get back the needed deployment

actions/scaling/config.go

@@ -0,0 +1,10 @@
+package scaling
+
+const (
+	AutoScalingConfigurationKey = "autoScalingConfig"
+)
+
+type AutoscalingConfig struct {
+	ChartRepository string `json:"chartRepository,omitempty" yaml:"chartRepository,omitempty"`
+	Image           string `json:"image,omitempty" yaml:"image,omitempty"`
+}

validation/prime/autoscaling/auto_scaling_main_test.go

@@ -0,0 +1,67 @@
+package autoscaling
+
+import (
+	"os"
+	"testing"
+
+	"github.com/rancher/shepherd/clients/rancher"
+	"github.com/rancher/shepherd/pkg/config"
+	"github.com/rancher/shepherd/pkg/config/operations"
+	"github.com/rancher/shepherd/pkg/session"
+	"github.com/rancher/tests/actions/features"
+	"github.com/rancher/tests/actions/scaling"
+	"github.com/sirupsen/logrus"
+)
+
+func TestMain(m *testing.M) {
+	testSession := session.NewSession()
+	client, err := rancher.NewClient("", testSession)
+	if err != nil {
+		logrus.Errorf("Failed, during client setup: %s", err)
+		os.Exit(1)
+	}
+
+	cattleConfig := config.LoadConfigFromFile(os.Getenv(config.ConfigEnvironmentKey))
+
+	autoScalingConfig := new(scaling.AutoscalingConfig)
+	operations.LoadObjectFromMap(scaling.AutoScalingConfigurationKey, cattleConfig, autoScalingConfig)
+
+	configureAutoscaler, err := features.IsEnabled(client, features.ClusterAutoscaling)
+	if err != nil {
+		logrus.Errorf("Failed, to get %s state", features.ClusterAutoscaling)
+		os.Exit(1)
+	}
+
+	if !configureAutoscaler {
+		if autoScalingConfig.ChartRepository == "" || autoScalingConfig.Image == "" {
+			logrus.Errorf("Autoscaling is not configured and no autoscaling config was provided")
+			os.Exit(1)
+		}
+
+		logrus.Infof("Setting autoscaler environment variables")
+		err = features.ConfigureAutoscaler(client, autoScalingConfig.ChartRepository, autoScalingConfig.Image)
+		if err != nil {
+			logrus.Errorf("Failed, to set autoscaler environment variables: %s", err)
+			os.Exit(1)
+		}
+
+		logrus.Infof("Enabling, %s", features.ClusterAutoscaling)
+		err = features.UpdateFeatureFlag(client, features.ClusterAutoscaling, true)
+		if err != nil {
+			logrus.Errorf("Failed, to enable %s: %s", features.ClusterAutoscaling, err)
+			os.Exit(1)
+		}
+	}
+
+	exitCode := m.Run()
+
+	if !configureAutoscaler {
+		err = features.UpdateFeatureFlag(client, features.ClusterAutoscaling, false)
+		if err != nil {
+			logrus.Errorf("Failed, to disable %s: %s", features.ClusterAutoscaling, err)
+			os.Exit(1)
+		}
+	}
+
+	os.Exit(exitCode)
+}

validation/prime/autoscaling/auto_scaling_test.go

@@ -359,5 +359,4 @@ func TestAutoScalingPause(t *testing.T) {
 			logrus.Warningf("Failed to upload schema parameters %s", err)
 		}
 	}
-
 }

validation/prime/autoscaling/defaults/defaults.yaml

@@ -24,6 +24,11 @@ awsMachineConfigs:
     vpcId: "<required>"
     retries: "5"
     zone: "a"
+
+#Required for non prime environments
+autoScalingConfig:
+  autoscalerChartRepository: "<required>" 
+  autoScalerImage: "<required>"
 
 logging:
    level: "info"