change to r

Azure · Mar 18, 2024 · d9387b3 · d9387b3
1 parent 7cb3aca
commit d9387b3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/e2e/suite_test.go b/e2e/suite_test.go
@@ -113,7 +113,7 @@ func runScenario(ctx context.Context, t *testing.T, r *mrand.Rand, opts *scenari
 
 	vmssSucceeded := true
 
-	vmssModel, cleanupVMSS, err := bootstrapVMSS(ctx, t, vmssName, opts, publicKeyBytes)
+	vmssModel, cleanupVMSS, err := bootstrapVMSS(ctx, t, r, vmssName, opts, publicKeyBytes)
 	if !opts.suiteConfig.KeepVMSS && cleanupVMSS != nil {
 		defer cleanupVMSS()
 	}

diff --git a/e2e/vmss.go b/e2e/vmss.go
@@ -30,7 +30,7 @@ type VMSSOperationRetrier struct {
 	maxRetries int
 }
 
-func bootstrapVMSS(ctx context.Context, t *testing.T, vmssName string, opts *scenarioRunOpts, publicKeyBytes []byte) (*armcompute.VirtualMachineScaleSet, func(), error) {
+func bootstrapVMSS(ctx context.Context, t *testing.T, r *mrand.Rand, vmssName string, opts *scenarioRunOpts, publicKeyBytes []byte) (*armcompute.VirtualMachineScaleSet, func(), error) {
 	nodeBootstrapping, err := getNodeBootstrapping(ctx, opts.nbc)
 	if err != nil {
 		return nil, nil, fmt.Errorf("unable to get node bootstrapping: %w", err)
@@ -49,8 +49,8 @@ func bootstrapVMSS(ctx context.Context, t *testing.T, vmssName string, opts *sce
 		log.Printf("finished deleting vmss %q", vmssName)
 	}
 
-	r := VMSSOperationRetrier{maxRetries: maxRetries}
-	vmssModel, err := r.createVMSSWithPayload(ctx, nodeBootstrapping.CustomData, nodeBootstrapping.CSE, vmssName, publicKeyBytes, opts)
+	rOpts := VMSSOperationRetrier{maxRetries: maxRetries}
+	vmssModel, err := rOpts.createVMSSWithPayload(ctx, nodeBootstrapping.CustomData, nodeBootstrapping.CSE, vmssName, publicKeyBytes, opts)
 	if err != nil {
 		return nil, cleanupVMSS, fmt.Errorf("unable to create VMSS with payload: %w", err)
 	}