Test for setting SpotMarketOptions

openshift · Mar 23, 2020 · 1d73d6d · 1d73d6d
1 parent cd210b3
commit 1d73d6d
Showing 1 changed file with 65 additions and 0 deletions.
diff --git a/pkg/actuators/machine/instances_test.go b/pkg/actuators/machine/instances_test.go
@@ -428,3 +428,68 @@ func TestSortInstances(t *testing.T) {
 	}
 	sortInstances(instances)
 }
+
+func TestGetInstanceMarketOptionsRequest(t *testing.T) {
+	testCases := []struct {
+		name              string
+		spotMarketOptions *awsproviderv1.SpotMarketOptions
+		expectedRequest   *ec2.InstanceMarketOptionsRequest
+	}{
+		{
+			name:              "with no Spot options specified",
+			spotMarketOptions: nil,
+			expectedRequest:   nil,
+		},
+		{
+			name:              "with an empty Spot options specified",
+			spotMarketOptions: &awsproviderv1.SpotMarketOptions{},
+			expectedRequest: &ec2.InstanceMarketOptionsRequest{
+				MarketType: aws.String(ec2.MarketTypeSpot),
+				SpotOptions: &ec2.SpotMarketOptions{
+					InstanceInterruptionBehavior: aws.String(ec2.InstanceInterruptionBehaviorTerminate),
+					SpotInstanceType:             aws.String(ec2.SpotInstanceTypeOneTime),
+				},
+			},
+		},
+		{
+			name: "with an empty MaxPrice specified",
+			spotMarketOptions: &awsproviderv1.SpotMarketOptions{
+				MaxPrice: aws.String(""),
+			},
+			expectedRequest: &ec2.InstanceMarketOptionsRequest{
+				MarketType: aws.String(ec2.MarketTypeSpot),
+				SpotOptions: &ec2.SpotMarketOptions{
+					InstanceInterruptionBehavior: aws.String(ec2.InstanceInterruptionBehaviorTerminate),
+					SpotInstanceType:             aws.String(ec2.SpotInstanceTypeOneTime),
+				},
+			},
+		},
+		{
+			name: "with a valid MaxPrice specified",
+			spotMarketOptions: &awsproviderv1.SpotMarketOptions{
+				MaxPrice: aws.String("0.01"),
+			},
+			expectedRequest: &ec2.InstanceMarketOptionsRequest{
+				MarketType: aws.String(ec2.MarketTypeSpot),
+				SpotOptions: &ec2.SpotMarketOptions{
+					InstanceInterruptionBehavior: aws.String(ec2.InstanceInterruptionBehaviorTerminate),
+					SpotInstanceType:             aws.String(ec2.SpotInstanceTypeOneTime),
+					MaxPrice:                     aws.String("0.01"),
+				},
+			},
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			providerConfig := &awsproviderv1.AWSMachineProviderConfig{
+				SpotMarketOptions: tc.spotMarketOptions,
+			}
+
+			request := getInstanceMarketOptionsRequest(providerConfig)
+			if !reflect.DeepEqual(request, tc.expectedRequest) {
+				t.Errorf("Case: %s. Got: %v, expected: %v", tc.name, request, tc.expectedRequest)
+			}
+		})
+	}
+}