has_tag for ALB

doc/_resource_types/alb.md

@@ -37,3 +37,11 @@ describe alb('my-alb') do
   it { should belong_to_vpc('my-vpc') }
 end
 ```
+
+### have_tag
+
+```ruby
+describe alb('my-alb') do
+  it { should have_tag('environment').value('dev') }
+end
+```

doc/resource_types.md

@@ -144,6 +144,22 @@ end
 ```
 
 
+### have_tag
+
+```ruby
+describe alb('my-alb') do
+  it { should have_tag('environment').value('dev') }
+end
+```
+
+### have_tag
+
+```ruby
+describe alb('my-alb') do
+  it { should have_tag('environment').value('dev') }
+end
+```
+
 ### belong_to_vpc
 
 ```ruby
@@ -152,6 +168,7 @@ describe alb('my-alb') do
 end
 ```
 
+
 ### its(:load_balancer_arn), its(:dns_name), its(:canonical_hosted_zone_id), its(:created_time), its(:load_balancer_name), its(:scheme), its(:vpc_id), its(:type), its(:security_groups), its(:ip_address_type)
 ## <a name="alb_listener">alb_listener</a>
 
@@ -274,7 +291,7 @@ describe apigateway('my-apigateway') do
 end
 ```
 
-### its(:id), its(:name), its(:description), its(:created_date), its(:version), its(:warnings), its(:binary_media_types), its(:minimum_compression_size), its(:api_key_source), its(:policy)
+### its(:id), its(:name), its(:description), its(:created_date), its(:version), its(:warnings), its(:binary_media_types), its(:minimum_compression_size), its(:api_key_source), its(:policy), its(:tags)
 ## <a name="autoscaling_group">autoscaling_group</a>
 
 AutoscalingGroup resource type.

lib/awspec/generator/doc/type/alb.rb

@@ -9,7 +9,8 @@ def initialize
           @ret = @type.resource_via_client
           @matchers = [
             Awspec::Type::Alb::STATES.map { |state| 'be_' + state }.join(', '),
-            'belong_to_vpc'
+            'belong_to_vpc',
+            'have_tag'
           ]
           @ignore_matchers = Awspec::Type::Alb::STATES.map { |state| 'be_' + state }
           @describes = []

lib/awspec/helper/finder/alb.rb

@@ -54,6 +54,15 @@ def select_rule_by_alb_listener_id(id)
         end
         selected
       end
+
+      def select_all_alb_tags(id)
+        res = elbv2_client.describe_tags({ resource_arns: [id] })
+        res.tag_descriptions.select do |resource|
+          resource.resource_arn == id
+        end.first.tags
+      rescue
+        return nil
+      end
     end
   end
 end

lib/awspec/stub/alb.rb

@@ -81,6 +81,15 @@
           protocol: 'HTTP'
         }
       ]
+    },
+    describe_tags: {
+      tag_descriptions: [
+        resource_arn: 'arn:aws:elasticloadbalancing:ap-northeast-1:1234567890:loadbalancer/app/my-alb/1aa1bb1cc1ddee11',
+        tags: [
+          key: 'environment',
+          value: 'dev'
+        ]
+      ]
     }
   }
 }

lib/awspec/type/alb.rb

@@ -1,5 +1,6 @@
 module Awspec::Type
   class Alb < ResourceBase
+    tags_allowed
     def resource_via_client
       @resource_via_client ||= find_alb(@display_name)
     end
@@ -38,5 +39,13 @@ def has_subnet?(subnet_id)
       subnet2 = find_subnet(subnet_id)
       subnet2.subnet_id = subnet_id
     end
+
+    def has_tag?(tag_key, tag_value)
+      alb_arn = resource_via_client.load_balancer_arn
+      tag_set = select_all_alb_tags(alb_arn)
+      tag_set.find do |tag|
+        tag.key == tag_key && tag.value == tag_value
+      end
+    end
   end
 end

spec/type/alb_spec.rb

@@ -10,4 +10,5 @@
   it { should have_subnet('subnet-1234a567') }
   it { should have_subnet('my-subnet') }
   it { should belong_to_vpc('my-vpc') }
+  it { should have_tag('environment').value('dev') }
 end