Merge pull request #427 from htnosm/pr-add-autoscaling_group-generate

Add have_nlb_target_group to generator for autoscaling_group
k1LoW · Dec 15, 2018 · 113552a · 113552a
2 parents a0dc72a + 4570625
commit 113552a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/awspec/generator/spec/autoscaling_group.rb b/lib/awspec/generator/spec/autoscaling_group.rb
@@ -40,8 +40,12 @@ def autoscaling_group_spec_template
 <% autoscaling_group[:load_balancer_names].each do |desc| %>
   it { should have_elb('<%= desc %>') }
 <% end %>
-<% autoscaling_group[:target_group_arns].each do |desc| %>
-  it { should have_alb_target_group('<%= desc.sub(/^.*targetgroup\/(.[^\/]*).*$/, '\1') %>') }
+<% autoscaling_group[:target_group_arns].join(",").gsub(/arn:.+?:targetgroup\/(.[^\/]*)\/.[^,]*/, '\1').split(",").each do |desc| %>
+  <%- if not find_alb_target_group(desc).nil? -%>
+  it { should have_alb_target_group('<%= desc %>') }
+  <%- elsif not find_nlb_target_group(desc).nil? -%>
+  it { should have_nlb_target_group('<%= desc %>') }
+  <%- end -%>
 <% end %>
 end
 EOF