awspec 0.36.0 → 0.36.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d6df1c727f0aeffc393ecf79feba39041cd14b5c
-  data.tar.gz: 247f84a0ffef96e9b51903bb58049821541b3b0b
+  metadata.gz: d3fddd77e3ed0bec352d8a53cfaa7f239b1fb896
+  data.tar.gz: f259c5bb2c4b5b90ab582c808d718d9f362a990d
 SHA512:
-  metadata.gz: abcc9b2a6623cc2c012bfb4bb606c48455fecf0e999e3ffdb356061998f1a6e5d8a136e8e77780f98510cb2f5a63896dfa23447883ee55c1f38f094e59912754
-  data.tar.gz: eaf78213317dd30bd9c39eb91c54b9f18814c519b809e685f1b85ff080386b49d960602a088c6fbcfb18be85e6d68f7e1aea9b96dd4d2bc44e13faebab5991d4
+  metadata.gz: 45c6af40e0ccbc9fd2d766dcd6df9dbf66168079e12893156ec5b7ece8190fe554b849f88c5eac54e80759ebe5c9bd47460790099356288da483123ba614cfbf
+  data.tar.gz: 32e845dd34c5dc984cf5959bef2bb9b9d2e59f86d65cda1d45b6b6de16d51bb61c24afb12d8aa2514e9ddbea530a8c181e166e78f192ccf90682b97e34e859ac

data/lib/awspec/command/generate.rb

@@ -8,7 +8,7 @@ module Awspec
     class_option :secrets_path, default: 'secrets.yml'
 
     types = %w(
-      vpc ec2 rds security_group elb network_acl route_table subnet nat_gateway
+      vpc ec2 rds security_group elb network_acl route_table subnet nat_gateway network_interface
     )
 
     types.each do |type|

data/lib/awspec/generator.rb

@@ -15,6 +15,7 @@ require 'awspec/generator/spec/ebs'
 require 'awspec/generator/spec/s3_bucket'
 require 'awspec/generator/spec/nat_gateway'
 require 'awspec/generator/spec/lambda'
+require 'awspec/generator/spec/network_interface'
 
 # Doc
 require 'awspec/generator/doc/type'

data/lib/awspec/generator/spec/network_interface.rb

@@ -0,0 +1,80 @@
+module Awspec::Generator
+  module Spec
+    class NetworkInterface
+      include Awspec::Helper::Finder
+      def generate_by_vpc_id(vpc_id)
+        describes = %w(
+        )
+        vpc = find_vpc(vpc_id)
+        raise 'Not Found VPC' unless vpc
+        @vpc_id = vpc[:vpc_id]
+        @vpc_tag_name = vpc.tag_name
+        network_interfaces = select_network_interface_by_vpc_id(@vpc_id)
+        specs = network_interfaces.map do |interface|
+          instance_spec = generate_instance_spec(interface)
+          subnet_spec = generate_subnet_spec(interface)
+          network_interface_id = interface[:network_interface_id]
+          linespecs = generate_linespecs(interface)
+          private_ip_addresses_count = interface.private_ip_addresses.count
+          content = ERB.new(network_interface_spec_template, nil, '-').result(binding).gsub(/^\n/, '')
+        end
+        specs.join("\n")
+      end
+
+      def generate_instance_spec(interface)
+        return unless interface.attachment.instance_id
+        instance = find_ec2(interface.attachment.instance_id)
+        instance_spec = if instance.tag_name
+                          "it { should be_attached_to('#{instance.tag_name}')"
+                        else
+                          "it { should be_attached_to('#{instance.instance_id}')"
+                        end
+        instance_spec += ".as_eth#{interface.attachment.device_index} }"
+        instance_spec
+      end
+
+      def generate_subnet_spec(interface)
+        return unless interface.subnet_id
+        subnet = find_subnet(interface.subnet_id)
+        subnet_spec = if subnet.tag_name
+                        "it { should belong_to_subnet('#{subnet.tag_name}') }"
+                      else
+                        "it { should belong_to_subnet('#{subnet.subnet_id}') }"
+                      end
+        subnet_spec
+      end
+
+      def generate_linespecs(interface)
+        linespecs = []
+        interface.private_ip_addresses.each do |ip_address|
+          line = "it { should have_private_ip_address('#{ip_address.private_ip_address}')"
+          line += '.primary' if ip_address.primary
+          line += ' }'
+          linespecs.push(line)
+        end
+        linespecs
+      end
+
+      def network_interface_spec_template
+        template = <<-'EOF'
+describe network_interface('<%= network_interface_id %>') do
+  it { should exist }
+  it { should be_<%= interface.status.tr('-', '_') %> }
+<%- if instance_spec -%>
+  <%= instance_spec %>
+<%- end -%>
+  it { should belong_to_vpc('<%= @vpc_tag_name %>') }
+<%- if subnet_spec -%>
+  <%= subnet_spec %>
+<%- end -%>
+<% linespecs.each do |line| %>
+  <%= line %>
+<% end %>
+  its(:private_ip_addresses_count) { should eq <%= interface.private_ip_addresses.count %> }
+end
+EOF
+        template
+      end
+    end
+  end
+end

data/lib/awspec/helper/finder/ec2.rb

@@ -94,6 +94,13 @@ module Awspec::Helper
                                                })
         res.nat_gateways
       end
+
+      def select_network_interface_by_vpc_id(vpc_id)
+        res = ec2_client.describe_network_interfaces({
+                                                       filters: [{ name: 'vpc-id', values: [vpc_id] }]
+                                                     })
+        res.network_interfaces
+      end
     end
   end
 end

data/lib/awspec/version.rb

@@ -1,3 +1,3 @@
 module Awspec
-  VERSION = '0.36.0'
+  VERSION = '0.36.1'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: awspec
 version: !ruby/object:Gem::Version
-  version: 0.36.0
+  version: 0.36.1
 platform: ruby
 authors:
 - k1LoW
@@ -276,6 +276,7 @@ files:
 - lib/awspec/generator/spec/lambda.rb
 - lib/awspec/generator/spec/nat_gateway.rb
 - lib/awspec/generator/spec/network_acl.rb
+- lib/awspec/generator/spec/network_interface.rb
 - lib/awspec/generator/spec/rds.rb
 - lib/awspec/generator/spec/route53_hosted_zone.rb
 - lib/awspec/generator/spec/route_table.rb