github.com/kikitux/packer@v0.10.1-0.20160322154024-6237df566f9f/builder/azure/arm/step_power_off_compute_test.go (about) 1 // Copyright (c) Microsoft Corporation. All rights reserved. 2 // Licensed under the MIT License. See the LICENSE file in builder/azure for license information. 3 4 package arm 5 6 import ( 7 "fmt" 8 "testing" 9 10 "github.com/mitchellh/multistep" 11 "github.com/mitchellh/packer/builder/azure/common/constants" 12 ) 13 14 func TestStepPowerOffComputeShouldFailIfPowerOffFails(t *testing.T) { 15 var testSubject = &StepPowerOffCompute{ 16 powerOff: func(string, string) error { return fmt.Errorf("!! Unit Test FAIL !!") }, 17 say: func(message string) {}, 18 error: func(e error) {}, 19 } 20 21 stateBag := createTestStateBagStepPowerOffCompute() 22 23 var result = testSubject.Run(stateBag) 24 if result != multistep.ActionHalt { 25 t.Fatalf("Expected the step to return 'ActionHalt', but got '%d'.", result) 26 } 27 28 if _, ok := stateBag.GetOk(constants.Error); ok == false { 29 t.Fatalf("Expected the step to set stateBag['%s'], but it was not.", constants.Error) 30 } 31 } 32 33 func TestStepPowerOffComputeShouldPassIfPowerOffPasses(t *testing.T) { 34 var testSubject = &StepPowerOffCompute{ 35 powerOff: func(string, string) error { return nil }, 36 say: func(message string) {}, 37 error: func(e error) {}, 38 } 39 40 stateBag := createTestStateBagStepPowerOffCompute() 41 42 var result = testSubject.Run(stateBag) 43 if result != multistep.ActionContinue { 44 t.Fatalf("Expected the step to return 'ActionContinue', but got '%d'.", result) 45 } 46 47 if _, ok := stateBag.GetOk(constants.Error); ok == true { 48 t.Fatalf("Expected the step to not set stateBag['%s'], but it was.", constants.Error) 49 } 50 } 51 52 func TestStepPowerOffComputeShouldTakeStepArgumentsFromStateBag(t *testing.T) { 53 var actualResourceGroupName string 54 var actualComputeName string 55 56 var testSubject = &StepPowerOffCompute{ 57 powerOff: func(resourceGroupName string, computeName string) error { 58 actualResourceGroupName = resourceGroupName 59 actualComputeName = computeName 60 61 return nil 62 }, 63 say: func(message string) {}, 64 error: func(e error) {}, 65 } 66 67 stateBag := createTestStateBagStepPowerOffCompute() 68 var result = testSubject.Run(stateBag) 69 70 if result != multistep.ActionContinue { 71 t.Fatalf("Expected the step to return 'ActionContinue', but got '%d'.", result) 72 } 73 74 var expectedComputeName = stateBag.Get(constants.ArmComputeName).(string) 75 var expectedResourceGroupName = stateBag.Get(constants.ArmResourceGroupName).(string) 76 77 if actualComputeName != expectedComputeName { 78 t.Fatalf("Expected the step to source 'constants.ArmResourceGroupName' from the state bag, but it did not.") 79 } 80 81 if actualResourceGroupName != expectedResourceGroupName { 82 t.Fatalf("Expected the step to source 'constants.ArmResourceGroupName' from the state bag, but it did not.") 83 } 84 } 85 86 func createTestStateBagStepPowerOffCompute() multistep.StateBag { 87 stateBag := new(multistep.BasicStateBag) 88 89 stateBag.Put(constants.ArmComputeName, "Unit Test: ComputeName") 90 stateBag.Put(constants.ArmResourceGroupName, "Unit Test: ResourceGroupName") 91 92 return stateBag 93 }