eipmap 0.0.1 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a2f1c0b232fc62fb36c85f964b4d3f94c15180e4
-  data.tar.gz: 97f4786d8f7e010c992f722c86f8c35bfd3bd649
+  metadata.gz: 1351fbe44fd9d8e912acd128d43ae4a121d8f9a3
+  data.tar.gz: a784a527ae399411c25cf5563784f9fc61c0adb9
 SHA512:
-  metadata.gz: 210bd9e16c09a217d9fe01b975c4a17587772a197a1a36d3b2354d45a28a6317cc16492a4047c2746e1e3535d41cc9e5422c3ed1828b92872788b6f5c5f095b1
-  data.tar.gz: 401a8d6b544f153daea65898a8d698ea6d090e2d25d2b5a3444fd6d539544aa525d0e0c2a154566f8305e5db46d4ae0a3c63310891fdff38d987661722557752
+  metadata.gz: bb29f18f62383b9dfb9157869324b0a8281c7bddf04cc6fbeebc8dd58a3a3560dff64208ca9d8cf14c48a3968c936ccbd8282d9d08233661f6d9c6dd35ca5893
+  data.tar.gz: 7caba6ba8454074e0c909b9c1ecee939d7898d1f2f1ce2a40c1929b1836896a5746ee65082d549586bc8cca1045630470be4e4b84de95cf7f187072c1444c5da

data/.rspec

@@ -0,0 +1,4 @@
+--require rspec/instafail
+--format RSpec::Instafail
+--colour
+--require spec_helper

data/.travis.yml

@@ -0,0 +1,11 @@
+language: ruby
+rvm:
+- 2.0.0
+script:
+- bundle install
+- bundle exec rake
+env:
+  global:
+  - secure: "aD9louz2hQ0gp47u2ScLLeQ7MiWQKlKNLR4Ux+U6KMO/sLMnktDcBGSi01lYt/+8yA6WyD3p/ceZv/e18enuRQU3ce5i3W8Xb11uAke0za44lTP5pllI13uTz0kHB9tmRS8MRSRtT2GV9HOcGJrW8td2YhImLwfc17pVOL2NnTc="
+  - secure: "ghKQZ8jbFpQnvRWnXO9eSNRjKOZJsGoq53Og5gH4KEdIEKIMBhnek7gOt34BfDFWSULJ7HRuuyIaftCy6r6rDJfV01zc88Pp+cRsf2LCfEpbEmXWBh8lBLYz0vneq6uoYpQ3y7FhedVkM89gQlbZ3snrykvr6j2FS6v/TUCgctc="
+  - EIPMAP_TEST_AWS_REGION=ap-southeast-1

data/README.md

@@ -5,6 +5,8 @@ Eipmap is a tool to manage Elastic IP Addresses (EIP).
 It defines the state of EIP using DSL, and updates EIP according to DSL.
 
 [![Gem Version](https://badge.fury.io/rb/eipmap.svg)](http://badge.fury.io/rb/eipmap)
+[![Build Status](https://travis-ci.org/winebarrel/eipmap.svg?branch=master)](https://travis-ci.org/winebarrel/eipmap)
+[![Coverage Status](https://coveralls.io/repos/winebarrel/eipmap/badge.png?branch=master)](https://coveralls.io/r/winebarrel/eipmap?branch=master)
 
 ## Installation
 
@@ -28,10 +30,10 @@ Or install it yourself as:
 export AWS_ACCESS_KEY_ID='...'
 export AWS_SECRET_ACCESS_KEY='...'
 export AWS_REGION='us-east-1'
-eipmap -e -o EIPfile  # export EIP
+eipmap -e -o EIPfile  # export EIP status
 vi EIPfile
 eipmap -a --dry-run
-eipmap -a             # apply `EIPfile` to EIP
+eipmap -a             # apply `EIPfile`
 ```
 
 ## Help

data/Rakefile

@@ -1,2 +1,5 @@
 require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
 
+RSpec::Core::RakeTask.new('spec')
+task :default => :spec

data/eipmap.gemspec

@@ -22,4 +22,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "term-ansicolor"
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rspec', '>= 3.0.0'
+  spec.add_development_dependency 'rspec-instafail'
+  spec.add_development_dependency 'coveralls'
 end

data/lib/eipmap/client.rb

@@ -3,7 +3,8 @@ class Eipmap::Client
 
   def initialize(options = {})
     @options = options
-    @ec2 = Aws::EC2::Client.new
+    aws_config = options.delete(:aws_config) || {}
+    @ec2 = Aws::EC2::Client.new(aws_config)
     @driver = Eipmap::Driver.new(@ec2, options)
   end
 

data/lib/eipmap/version.rb

@@ -1,3 +1,3 @@
 module Eipmap
-  VERSION = '0.0.1'
+  VERSION = '0.1.0'
 end

data/spec/eipmap_spec.rb

@@ -0,0 +1,119 @@
+describe Eipmap do
+  let(:eips) do
+    eips = describe_addresses["vpc"].keys.zip(
+             describe_network_interface.map {|interface_id, private_ips|
+               { network_interface_id: interface_id,
+                 private_ip_address: private_ips.first}})
+    Hash[*eips.flatten]
+  end
+
+  context "when no association" do
+    it do
+      dsl = <<-EOS
+domain "vpc" do
+<%- eips.each do |public_ip, attrs| -%>
+  ip "<%= public_ip %>"
+<%- end -%>
+end
+      EOS
+
+      result = apply { dsl }
+      expect(result).to be_falsey
+      expect(describe_addresses["vpc"].values).to match_array [{}, {}, {}]
+    end
+  end
+
+  context "when associate" do
+    it do
+      dsl = <<-EOS
+domain "vpc" do
+<%- eips.each do |public_ip, attrs| -%>
+  ip "<%= public_ip %>", <%= attrs.inspect %>
+<%- end -%>
+end
+      EOS
+
+      result = apply { dsl }
+      expect(result).to be_truthy
+      expect(describe_addresses["vpc"]).to eq eips
+    end
+  end
+
+  context "when disassociate" do
+    before do
+      apply {
+        <<-EOS
+domain "vpc" do
+<%- eips.each do |public_ip, attrs| -%>
+  ip "<%= public_ip %>", <%= attrs.inspect %>
+<%- end -%>
+end
+        EOS
+      }
+    end
+
+    it do
+      dsl = <<-EOS
+domain "vpc" do
+<%- eips.each do |public_ip, attrs| -%>
+  ip "<%= public_ip %>"
+<%- end -%>
+end
+      EOS
+
+      result = apply { dsl }
+      expect(result).to be_truthy
+      expect(describe_addresses["vpc"].values).to match_array [{}, {}, {}]
+    end
+  end
+
+  context "when swap association" do
+    let(:eips_with_one_assoc) do
+      eips.each_with_index do |(public_ip, attrs), i|
+        attrs.clear unless i.zero?
+      end
+
+      eips
+    end
+
+    let(:swapped_eips_with_one_assoc) do
+      public_ips = eips_with_one_assoc.keys
+      swapped = {}
+      mapping = {0 => 1, 1 => 0}
+
+      eips_with_one_assoc.each_with_index do |(public_ip, attrs), i|
+        ip_idx = mapping.fetch(i, i)
+        ip = public_ips[ip_idx]
+        swapped[public_ip] = eips_with_one_assoc[ip]
+      end
+
+      swapped
+    end
+
+    before do
+      apply {
+        <<-EOS
+domain "vpc" do
+<%- eips_with_one_assoc.each do |public_ip, attrs| -%>
+  ip "<%= public_ip %>", <%= attrs.inspect %>
+<%- end -%>
+end
+        EOS
+      }
+    end
+
+    it do
+      dsl = <<-EOS
+domain "vpc" do
+<%- swapped_eips_with_one_assoc.each do |public_ip, attrs| -%>
+  ip "<%= public_ip %>", <%= attrs.inspect %>
+<%- end -%>
+end
+      EOS
+
+      result = apply { dsl }
+      expect(result).to be_truthy
+      expect(describe_addresses["vpc"]).to eq swapped_eips_with_one_assoc
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,176 @@
+if ENV['TRAVIS']
+  require 'simplecov'
+  require 'coveralls'
+
+  SimpleCov.formatter = Coveralls::SimpleCov::Formatter
+  SimpleCov.start do
+    add_filter "spec/"
+  end
+end
+
+require 'erb'
+require 'tempfile'
+require 'eipmap'
+
+# Amazon Linux AMI 2014.09 (HVM)
+TEST_IMAGE_ID = 'ami-d6e1c584'
+
+Aws.config.update(
+  access_key_id: ENV['EIPMAP_TEST_ACCESS_KEY_ID'],
+  secret_access_key: ENV['EIPMAP_TEST_AWS_SECRET_ACCESS_KEY'],
+  region: ENV['EIPMAP_TEST_AWS_REGION'])
+
+$ec2 = Aws::EC2::Client.new
+
+def run_instances(n)
+  resp = $ec2.run_instances(
+    image_id: TEST_IMAGE_ID,
+    min_count: n,
+    max_count: n,
+    instance_type: 't2.micro',
+    network_interfaces: [{
+      device_index: 0,
+      associate_public_ip_address: false}])
+
+  instance_ids = resp.instances.map(&:instance_id)
+  $ec2.wait_until(:instance_running, instance_ids: instance_ids)
+  $test_instances = resp.instances
+end
+
+def terminate_instances
+  instance_ids = $test_instances.map(&:instance_id)
+  $ec2.terminate_instances(instance_ids: instance_ids)
+  $ec2.wait_until(:instance_terminated, instance_ids: instance_ids)
+end
+
+def allocate_addresses(n)
+  $test_addresses = (1..n).map do
+    $ec2.allocate_address(domain: 'vpc')
+  end
+end
+
+def release_addresses
+  allocation_ids = $test_addresses.map(&:allocation_id)
+
+  allocation_ids.each do |allocation_id|
+    $ec2.release_address(allocation_id: allocation_id)
+  end
+end
+
+def disassociate_addresses
+  allocation_ids = $test_addresses.map(&:allocation_id)
+  resp = $ec2.describe_addresses(allocation_ids: allocation_ids)
+  association_ids = resp.addresses.map(&:association_id)
+
+  association_ids.each do |association_id|
+    next unless association_id
+    $ec2.disassociate_address(association_id: association_id)
+  end
+end
+
+def describe_network_interface
+  result = {}
+
+  $test_instances.each do |instance|
+    interface = instance.network_interfaces.first
+    interface_id = interface.network_interface_id
+    result[interface_id] = []
+
+    interface.private_ip_addresses.sort_by {|address|
+      address.primary ? '' : address.private_ip_address
+    }.each {|address|
+      result[interface_id] << address.private_ip_address
+    }
+  end
+
+  result
+end
+
+def describe_addresses
+  result = {}
+
+  $ec2.describe_addresses.each do |resp|
+    resp.addresses.each do |address|
+      domain = address.domain
+      public_ip = address.public_ip
+      result[domain] ||= {}
+      result[domain][public_ip] = {}
+
+      [:network_interface_id, :private_ip_address].each do |key|
+        value = address[key] || ''
+        result[domain][public_ip][key] = value unless value.empty?
+      end
+    end
+  end
+
+  result
+end
+
+def client(user_options = {})
+  options = {logger: Logger.new('/dev/null')}
+
+  if_debug do
+    logger = Eipmap::Logger.instance
+    logger.set_debug(true)
+    options.update(
+      debug: true,
+      logger: logger,
+      aws_config: {
+        http_wire_trace: true,
+        logger: logger})
+  end
+
+  options = options.merge(user_options)
+  Eipmap::Client.new(options)
+end
+
+def tempfile(content, options = {})
+  basename = "#{File.basename __FILE__}.#{$$}"
+  basename = [basename, options[:ext]] if options[:ext]
+
+  Tempfile.open(basename) do |f|
+    f.puts(content)
+    f.flush
+    f.rewind
+    yield(f)
+  end
+end
+
+def apply(cli = client)
+  elbfile = yield
+  elbfile = ERB.new(elbfile, nil, '-').result(binding)
+
+  if_debug do
+    puts <<-EOS
+--- ELBfile ---
+#{elbfile.strip}
+---------------
+    EOS
+  end
+
+  tempfile(elbfile) do |f|
+    cli.apply(f.path)
+  end
+end
+
+def if_debug
+  if ENV['DEBUG'] == '1'
+    yield
+  end
+end
+
+RSpec.configure do |config|
+  config.before(:all) do
+    run_instances(3)
+    allocate_addresses(3)
+  end
+
+  config.before(:each) do
+    disassociate_addresses
+  end
+
+  config.after(:all) do
+    terminate_instances
+    release_addresses
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: eipmap
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
 platform: ruby
 authors:
 - Genki Sugawara
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-12 00:00:00.000000000 Z
+date: 2014-10-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -66,6 +66,48 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+- !ruby/object:Gem::Dependency
+  name: rspec-instafail
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: coveralls
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Eipmap is a tool to manage Elastic IP Addresses (EIP). It defines the
   state of EIP using DSL, and updates EIP according to DSL.
 email:
@@ -76,6 +118,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - .gitignore
+- .rspec
+- .travis.yml
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -93,6 +137,8 @@ files:
 - lib/eipmap/ext/string_ext.rb
 - lib/eipmap/logger.rb
 - lib/eipmap/version.rb
+- spec/eipmap_spec.rb
+- spec/spec_helper.rb
 homepage: https://github.com/winebarrel/eipmap
 licenses:
 - MIT
@@ -117,4 +163,6 @@ rubygems_version: 2.4.1
 signing_key: 
 specification_version: 4
 summary: Eipmap is a tool to manage Elastic IP Addresses (EIP).
-test_files: []
+test_files:
+- spec/eipmap_spec.rb
+- spec/spec_helper.rb