github.com/kikitux/packer@v0.10.1-0.20160322154024-6237df566f9f/builder/azure/arm/step_create_resource_group_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 TestStepCreateResourceGroupShouldFailIfCreateFails(t *testing.T) { 15 var testSubject = &StepCreateResourceGroup{ 16 create: func(string, string) error { return fmt.Errorf("!! Unit Test FAIL !!") }, 17 say: func(message string) {}, 18 error: func(e error) {}, 19 } 20 21 stateBag := createTestStateBagStepCreateResourceGroup() 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 TestStepCreateResourceGroupShouldPassIfCreatePasses(t *testing.T) { 34 var testSubject = &StepCreateResourceGroup{ 35 create: func(string, string) error { return nil }, 36 say: func(message string) {}, 37 error: func(e error) {}, 38 } 39 40 stateBag := createTestStateBagStepCreateResourceGroup() 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 TestStepCreateResourceGroupShouldTakeStepArgumentsFromStateBag(t *testing.T) { 53 var actualResourceGroupName string 54 var actualLocation string 55 56 var testSubject = &StepCreateResourceGroup{ 57 create: func(resourceGroupName string, location string) error { 58 actualResourceGroupName = resourceGroupName 59 actualLocation = location 60 return nil 61 }, 62 say: func(message string) {}, 63 error: func(e error) {}, 64 } 65 66 stateBag := createTestStateBagStepCreateResourceGroup() 67 var result = testSubject.Run(stateBag) 68 69 if result != multistep.ActionContinue { 70 t.Fatalf("Expected the step to return 'ActionContinue', but got '%d'.", result) 71 } 72 73 var expectedLocation = stateBag.Get(constants.ArmLocation).(string) 74 var expectedResourceGroupName = stateBag.Get(constants.ArmResourceGroupName).(string) 75 76 if actualResourceGroupName != expectedResourceGroupName { 77 t.Fatalf("Expected the step to source 'constants.ArmResourceGroupName' from the state bag, but it did not.") 78 } 79 80 if actualLocation != expectedLocation { 81 t.Fatalf("Expected the step to source 'constants.ArmResourceGroupName' from the state bag, but it did not.") 82 } 83 } 84 85 func createTestStateBagStepCreateResourceGroup() multistep.StateBag { 86 stateBag := new(multistep.BasicStateBag) 87 88 stateBag.Put(constants.ArmLocation, "Unit Test: Location") 89 stateBag.Put(constants.ArmResourceGroupName, "Unit Test: ResourceGroupName") 90 91 return stateBag 92 }