@@ -43,31 +43,32 @@ func (m *MockEC2) CreateTags(ctx context.Context, request *ec2.CreateTagsInput,
4343
4444func (m * MockEC2 ) addTags (resourceId string , tags ... ec2types.Tag ) {
4545 var resourceType ec2types.ResourceType
46- if strings .HasPrefix (resourceId , "subnet-" ) {
46+ switch {
47+ case strings .HasPrefix (resourceId , "subnet-" ):
4748 resourceType = ec2types .ResourceTypeSubnet
48- } else if strings .HasPrefix (resourceId , "vpc-" ) {
49+ case strings .HasPrefix (resourceId , "vpc-" ):
4950 resourceType = ec2types .ResourceTypeVpc
50- } else if strings .HasPrefix (resourceId , "sg-" ) {
51+ case strings .HasPrefix (resourceId , "sg-" ):
5152 resourceType = ec2types .ResourceTypeSecurityGroup
52- } else if strings .HasPrefix (resourceId , "vol-" ) {
53+ case strings .HasPrefix (resourceId , "vol-" ):
5354 resourceType = ec2types .ResourceTypeVolume
54- } else if strings .HasPrefix (resourceId , "igw-" ) {
55+ case strings .HasPrefix (resourceId , "igw-" ):
5556 resourceType = ec2types .ResourceTypeInternetGateway
56- } else if strings .HasPrefix (resourceId , "eigw-" ) {
57+ case strings .HasPrefix (resourceId , "eigw-" ):
5758 resourceType = ec2types .ResourceTypeEgressOnlyInternetGateway
58- } else if strings .HasPrefix (resourceId , "nat-" ) {
59+ case strings .HasPrefix (resourceId , "nat-" ):
5960 resourceType = ec2types .ResourceTypeNatgateway
60- } else if strings .HasPrefix (resourceId , "dopt-" ) {
61+ case strings .HasPrefix (resourceId , "dopt-" ):
6162 resourceType = ec2types .ResourceTypeDhcpOptions
62- } else if strings .HasPrefix (resourceId , "rtb-" ) {
63+ case strings .HasPrefix (resourceId , "rtb-" ):
6364 resourceType = ec2types .ResourceTypeRouteTable
64- } else if strings .HasPrefix (resourceId , "eipalloc-" ) {
65+ case strings .HasPrefix (resourceId , "eipalloc-" ):
6566 resourceType = ec2types .ResourceTypeElasticIp
66- } else if strings .HasPrefix (resourceId , "lt-" ) {
67+ case strings .HasPrefix (resourceId , "lt-" ):
6768 resourceType = ec2types .ResourceTypeLaunchTemplate
68- } else if strings .HasPrefix (resourceId , "key-" ) {
69+ case strings .HasPrefix (resourceId , "key-" ):
6970 resourceType = ec2types .ResourceTypeKeyPair
70- } else {
71+ default :
7172 klog .Fatalf ("Unknown resource-type in create tags: %v" , resourceId )
7273 }
7374 for _ , tag := range tags {
@@ -83,7 +84,8 @@ func (m *MockEC2) addTags(resourceId string, tags ...ec2types.Tag) {
8384
8485func (m * MockEC2 ) hasTag (resourceType ec2types.ResourceType , resourceId string , filter ec2types.Filter ) bool {
8586 name := * filter .Name
86- if strings .HasPrefix (name , "tag:" ) {
87+ switch {
88+ case strings .HasPrefix (name , "tag:" ):
8789 tagKey := name [4 :]
8890
8991 for _ , tag := range m .Tags {
@@ -103,7 +105,7 @@ func (m *MockEC2) hasTag(resourceType ec2types.ResourceType, resourceId string,
103105 }
104106 }
105107 }
106- } else if name == "tag-key" {
108+ case name == "tag-key" :
107109 for _ , tag := range m .Tags {
108110 if * tag .ResourceId != resourceId {
109111 continue
@@ -117,7 +119,7 @@ func (m *MockEC2) hasTag(resourceType ec2types.ResourceType, resourceId string,
117119 }
118120 }
119121 }
120- } else {
122+ default :
121123 klog .Fatalf ("Unsupported filter: %v" , filter )
122124 }
123125 return false
0 commit comments