ec2ssh 2.0.8 → 3.0.0.beta1

data/spec/lib/ec2ssh/command/init_spec.rb

@@ -0,0 +1,41 @@
+require 'ec2ssh/command/init'
+
+describe Ec2ssh::Command::Init do
+  describe '#run' do
+    let(:command) do
+      described_class.new(cli).tap do |cmd|
+        allow(cmd).to receive(:ssh_config_path).and_return('/path/to/ssh/config')
+        allow(cmd).to receive(:dotfile_path).and_return('/path/to/dotfile')
+      end
+    end
+    let(:cli) do
+      double(:cli, red: nil, yellow: nil, green: nil)
+    end
+    let(:ssh_config) do
+      double(:ssh_config, mark_exist?: nil, append_mark!: nil)
+    end
+
+    before do
+      expect(ssh_config).to receive(:mark_exist?).and_return(mark_exist)
+      allow(command).to receive(:ssh_config).and_return(ssh_config)
+    end
+
+    context 'when the marker already exists' do
+      let(:mark_exist) { true }
+
+      it do
+        expect { command.run }.to raise_error(Ec2ssh::MarkAlreadyExists)
+        expect(ssh_config).to_not have_received(:append_mark!)
+      end
+    end
+
+    context 'when the marker does not exists' do
+      let(:mark_exist) { false }
+
+      it do
+        expect { command.run }.not_to raise_error
+        expect(ssh_config).to have_received(:append_mark!).once
+      end
+    end
+  end
+end

data/spec/lib/ec2ssh/command/migrate_spec.rb

@@ -0,0 +1,108 @@
+require 'spec_helper'
+require 'ec2ssh/command/migrate'
+
+describe Ec2ssh::Command::Migrate do
+  describe '#run' do
+    let(:cli) do
+      double(:cli,
+             options: options, yes?: nil,
+             red: nil, yellow: nil, green: nil)
+    end
+    let(:command) do
+      described_class.new cli
+    end
+    let(:options) do
+      double(:options, dotfile: '/dotfile')
+    end
+
+    before do
+      File.open('/dotfile', 'w') {|f| f.write dotfile_str }
+    end
+
+    around do |example|
+      silence_stdout { example.run }
+    end
+
+    context 'version 2' do
+      let(:dotfile_str) { <<-END }
+---
+path: /path/to/ssh/config
+aws_keys:
+  key1:
+    access_key_id: ACCESS_KEY1
+    secret_access_key: SECRET1
+      END
+
+      context 'yes' do
+        before do
+          expect(cli).to receive(:yes?).and_return(true)
+          Timecop.freeze(Time.utc(2014, 1, 1)) do
+            command.run
+          end
+        end
+
+        it do
+          expect(File.read('/dotfile')).to eq <<-END
+path '/path/to/ssh/config'
+aws_keys(
+  key1: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' }
+)
+
+# You can use methods of AWS::EC2::Instance.
+# See http://docs.aws.amazon.com/AWSRubySDK/latest/AWS/EC2/Instance.html
+host_line <<EOS
+Host <%= tags['Name'] %>.<%= availability_zone %>
+  HostName <%= dns_name || private_ip_address %>
+EOS
+
+# ---
+# path: /path/to/ssh/config
+# aws_keys:
+#   key1:
+#     access_key_id: ACCESS_KEY1
+#     secret_access_key: SECRET1
+          END
+        end
+
+        it do
+          expect(File.read('/dotfile.20140101000000')).to eq(dotfile_str)
+        end
+      end
+
+      context 'no' do
+        before do
+          expect(cli).to receive(:yes?).and_return(false)
+          command.run
+        end
+
+        it do
+          expect(File.read('/dotfile')).to eq(dotfile_str)
+        end
+      end
+    end
+
+    context 'version 3' do
+      let(:dotfile_str) { <<-END }
+path '/path/to/ssh/config'
+aws_keys(
+  key1: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' }
+)
+
+# You can use methods of AWS::EC2::Instance.
+# See http://docs.aws.amazon.com/AWSRubySDK/latest/AWS/EC2/Instance.html
+host_line <<EOS
+Host <%= tags['Name'] %>.<%= availability_zone %>
+  HostName <%= dns_name || private_ip_address %>
+EOS
+      END
+
+      before do
+        command.run
+      end
+
+      it do
+        expect(File.read('/dotfile')).to eq(dotfile_str)
+      end
+    end
+  end
+end

data/spec/lib/ec2ssh/command/remove_spec.rb

@@ -0,0 +1,68 @@
+require 'spec_helper'
+require 'ec2ssh/command/remove'
+
+describe Ec2ssh::Command::Remove do
+  describe '#run' do
+    let(:command) do
+      described_class.new(cli).tap do |cmd|
+        allow(cmd).to receive(:options).and_return(options)
+      end
+    end
+    let(:options) do
+      double(:options, path: '/ssh_config', dotfile: '/dotfile', aws_key: 'default')
+    end
+    let(:cli) do
+      double(:cli, options: options, red: nil, yellow: nil, green: nil)
+    end
+
+    let(:dotfile_str) { <<-END }
+path '/dotfile'
+aws_keys(
+  default: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' }
+)
+host_line <<EOS
+Host <%= tags['Name'] %>
+  HostName <%= private_ip_address %>
+EOS
+    END
+
+    before do
+      File.open('/ssh_config', 'w') {|f| f.write ssh_config_str }
+      File.open('/dotfile', 'w') {|f| f.write dotfile_str }
+    end
+
+    context 'with unmarked ssh_config' do
+      let(:ssh_config_str) { '' }
+
+      it do
+        expect { command.run }.to raise_error(Ec2ssh::MarkNotFound)
+      end
+    end
+
+    context 'with marked ssh_config' do
+      let(:ssh_config_str) { <<-END }
+# before lines...
+
+### EC2SSH BEGIN ###
+### EC2SSH END ###
+
+# after lines...
+      END
+
+      before do
+        Timecop.freeze(Time.utc(2014,1,1)) do
+          command.run
+        end
+      end
+
+      it do
+        expect(File.read('/ssh_config')).to eq(<<-END)
+# before lines...
+
+
+# after lines...
+        END
+      end
+    end
+  end
+end

data/spec/lib/ec2ssh/command/update_spec.rb

@@ -0,0 +1,97 @@
+require 'spec_helper'
+require 'ec2ssh/command/update'
+require 'ec2ssh/exceptions'
+
+describe Ec2ssh::Command::Update do
+  describe '#run' do
+    let(:command) do
+      described_class.new(cli).tap do |cmd|
+        allow(cmd).to receive(:options).and_return(options)
+        allow(cmd.builder.ec2s).to receive(:instances) { instances }
+      end
+    end
+    let(:options) do
+      double(:options, path: '/ssh_config', dotfile: '/dotfile', aws_key: 'default')
+    end
+    let(:cli) do
+      double(:cli, options: options, red: nil, yellow: nil, green: nil)
+    end
+    let(:instances) do
+      [
+        double('instance', tags: {'Name' => 'srv1'}, private_ip_address: '10.0.0.1'),
+        double('instance', tags: {'Name' => 'srv2'}, private_ip_address: '10.0.0.2')
+      ]
+    end
+
+    before do
+      File.open('/ssh_config', 'w') {|f| f.write ssh_config_str }
+      File.open('/dotfile', 'w') {|f| f.write dotfile_str }
+    end
+
+    context 'with unmarked ssh_config' do
+      let(:ssh_config_str) { '' }
+      let(:dotfile_str) { <<-END }
+path '/dotfile'
+aws_keys(
+  default: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' }
+)
+host_line <<EOS
+Host <%= tags['Name'] %>
+  HostName <%= private_ip_address %>
+EOS
+      END
+
+      it do
+        expect { command.run }.to raise_error(Ec2ssh::MarkNotFound)
+      end
+    end
+
+    context 'with marked ssh_config' do
+      let(:ssh_config_str) { <<-END }
+# before lines...
+
+### EC2SSH BEGIN ###
+### EC2SSH END ###
+
+# after lines...
+      END
+
+      let(:dotfile_str) { <<-END }
+path '/dotfile'
+aws_keys(
+  default: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' }
+)
+host_line <<EOS
+Host <%= tags['Name'] %>
+  HostName <%= private_ip_address %>
+EOS
+      END
+
+      before do
+        Timecop.freeze(Time.utc(2014,1,1)) do
+          command.run
+        end
+      end
+
+      it do
+        expect(File.read('/ssh_config')).to eq(<<-END)
+# before lines...
+
+### EC2SSH BEGIN ###
+# Generated by ec2ssh http://github.com/mirakui/ec2ssh
+# DO NOT edit this block!
+# Updated 2014-01-01T00:00:00Z
+# section: default
+Host srv1
+  HostName 10.0.0.1
+Host srv2
+  HostName 10.0.0.2
+
+### EC2SSH END ###
+
+# after lines...
+        END
+      end
+    end
+  end
+end

data/spec/lib/ec2ssh/dsl_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+require 'ec2ssh/dsl'
+
+describe Ec2ssh::Dsl do
+  shared_examples 'a filled dsl container' do
+  end
+
+  let(:dsl_str) do
+<<-END
+aws_keys(
+  key1: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' },
+  key2: { access_key_id: 'ACCESS_KEY2', secret_access_key: 'SECRET2' }
+)
+regions 'ap-northeast-1', 'us-east-1'
+host_line 'host lines'
+reject {|instance| instance }
+path 'path'
+END
+  end
+
+  subject(:result) { Ec2ssh::Dsl::Parser.parse dsl_str }
+
+  its(:aws_keys) do
+    should == {
+      key1: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' },
+      key2: { access_key_id: 'ACCESS_KEY2', secret_access_key: 'SECRET2' }
+    }
+  end
+  its(:regions) { should == ['ap-northeast-1', 'us-east-1'] }
+  its(:host_line) { should == 'host lines' }
+  it { expect(result.reject.call(123)).to eq(123) }
+  its(:path) { should == 'path' }
+end

data/spec/lib/ec2ssh/migrator_spec.rb

@@ -0,0 +1,59 @@
+require 'spec_helper'
+require 'ec2ssh/migrator'
+
+describe Ec2ssh::Migrator do
+  subject(:migrator) { described_class.new '/dotfile' }
+
+  before do
+    File.open('/dotfile', 'w') {|f| f.write dotfile_str }
+  end
+
+  context 'from version 2' do
+    let(:dotfile_str) { <<-END }
+---
+path: /path/to/ssh/config
+aws_keys:
+  key1:
+    access_key_id: ACCESS_KEY1
+    secret_access_key: SECRET1
+  key2:
+    access_key_id: ACCESS_KEY2
+    secret_access_key: SECRET2
+regions:
+- ap-northeast-1
+- us-east-1
+    END
+
+    let(:new_dotfile_str) { <<-END }
+path '/path/to/ssh/config'
+aws_keys(
+  key1: { access_key_id: 'ACCESS_KEY1', secret_access_key: 'SECRET1' },
+  key2: { access_key_id: 'ACCESS_KEY2', secret_access_key: 'SECRET2' }
+)
+regions 'ap-northeast-1', 'us-east-1'
+
+# You can use methods of AWS::EC2::Instance.
+# See http://docs.aws.amazon.com/AWSRubySDK/latest/AWS/EC2/Instance.html
+host_line <<EOS
+Host <%= tags['Name'] %>.<%= availability_zone %>
+  HostName <%= dns_name || private_ip_address %>
+EOS
+
+# ---
+# path: /path/to/ssh/config
+# aws_keys:
+#   key1:
+#     access_key_id: ACCESS_KEY1
+#     secret_access_key: SECRET1
+#   key2:
+#     access_key_id: ACCESS_KEY2
+#     secret_access_key: SECRET2
+# regions:
+# - ap-northeast-1
+# - us-east-1
+    END
+
+    it { expect(migrator.check_version).to eq('2') }
+    it { expect(migrator.migrate_from_2).to eq(new_dotfile_str) }
+  end
+end

data/spec/lib/ec2ssh/ssh_config_spec.rb

@@ -2,13 +2,45 @@ require 'spec_helper'
 require 'ec2ssh/ssh_config'
 
 describe Ec2ssh::SshConfig do
+  let(:path) { Pathname('/ssh_config') }
+
+  subject(:ssh_config) do
+    described_class.new(path).tap(&:parse!)
+  end
+
+  before do
+    File.open(path, 'w') {|f| f.write config_str }
+  end
+
+  context 'expect be false' do
+    let(:config_str) { <<-END }
+Host host1
+HostName 0.0.0.0
+END
+
+    it { expect(ssh_config.mark_exist?).to be_falsey }
+  end
+
+  describe '#mark_exist?' do
+    context 'expect be true' do
+      let(:config_str) { <<-END }
+### EC2SSH BEGIN ###
+# Generated by ec2ssh http://github.com/mirakui/ec2ssh
+# DO NOT edit this block!
+# Updated 2013-01-01T00:00:00+00:00
+Host db-01.ap-northeast-1
+  HostName ec2-1-1-1-1.ap-northeast-1.ec2.amazonaws.com
+
+### EC2SSH END ###
+END
+
+      it { expect(ssh_config.mark_exist?).to be_truthy }
+    end
+  end
+
   describe '#parse!' do
     context 'when no section exists' do
-      let(:path) { path = tmp_dir.join('ssh_config') }
-      subject    { described_class.new(path)         }
-
-      before do
-        path.open('w') {|f| f.write <<-END }
+      let(:config_str) { ; <<-END }
 ### EC2SSH BEGIN ###
 # Generated by ec2ssh http://github.com/mirakui/ec2ssh
 # DO NOT edit this block!
@@ -18,20 +50,13 @@ Host db-01.ap-northeast-1
 
 ### EC2SSH END ###
 END
-        subject.parse!
-      end
 
-      it { expect(subject.sections.size).to be == 1 }
-      it { expect(subject.sections['default']).to be_an_instance_of Ec2ssh::SshConfig::Section }
+      it { expect(ssh_config.sections.size).to be == 1 }
+      it { expect(ssh_config.sections['default']).to be_an_instance_of Ec2ssh::SshConfig::Section }
     end
 
     context 'when a section exists' do
-      let(:path) { path = tmp_dir.join('ssh_config') }
-      subject    { described_class.new(path)         }
-
-      before do
-        path = tmp_dir.join('ssh_config')
-        path.open('w') {|f| f.write <<-END }
+      let(:config_str) { <<-END }
 Host foo.bar.com
   HostName 1.2.3.4
 ### EC2SSH BEGIN ###
@@ -44,20 +69,13 @@ Host db-01.ap-northeast-1
 
 ### EC2SSH END ###
 END
-        subject.parse!
-      end
 
       it { expect(subject.sections.size).to be == 2 }
       it { expect(subject.sections['foo']).to be_an_instance_of Ec2ssh::SshConfig::Section }
     end
 
     context 'when multiple sections exist' do
-      let(:path) { path = tmp_dir.join('ssh_config') }
-      subject    { described_class.new(path)         }
-
-      before do
-        path = tmp_dir.join('ssh_config')
-        path.open('w') {|f| f.write <<-END }
+      let(:config_str) { <<-END }
 Host foo.bar.com
   HostName 1.2.3.4
 ### EC2SSH BEGIN ###
@@ -73,53 +91,51 @@ Host db-02.ap-northeast-1
 
 ### EC2SSH END ###
 END
-        subject.parse!
-      end
 
-      it { expect(subject.sections.size).to be == 3 }
-      it { expect(subject.sections['foo']).to be_an_instance_of Ec2ssh::SshConfig::Section }
-      it { expect(subject.sections['bar']).to be_an_instance_of Ec2ssh::SshConfig::Section }
+      it { expect(ssh_config.sections.size).to be == 3 }
+      it { expect(ssh_config.sections['foo']).to be_an_instance_of Ec2ssh::SshConfig::Section }
+      it { expect(ssh_config.sections['bar']).to be_an_instance_of Ec2ssh::SshConfig::Section }
     end
   end
+end
 
-  describe Ec2ssh::SshConfig::Section do
-    describe '#append' do
-      let(:section) { Ec2ssh::SshConfig::Section.new('test') }
-      before { section.append('foo') }
+describe Ec2ssh::SshConfig::Section do
+  describe '#append' do
+    let(:section) { Ec2ssh::SshConfig::Section.new('test') }
+    before { section.append('foo') }
 
-      it { expect(section.text).to eq 'foo' }
-    end
+    it { expect(section.text).to eq 'foo' }
+  end
 
-    describe '#replace' do
-      let(:section) { Ec2ssh::SshConfig::Section.new('test', 'foo') }
-      before { section.replace!('bar') }
+  describe '#replace' do
+    let(:section) { Ec2ssh::SshConfig::Section.new('test', 'foo') }
+    before { section.replace!('bar') }
 
-      it { expect(section.text).to eq 'bar' }
-    end
+    it { expect(section.text).to eq 'bar' }
+  end
 
-    describe '#to_s' do
-      context 'when no text given' do
-        let(:section) { Ec2ssh::SshConfig::Section.new('test') }
+  describe '#to_s' do
+    context 'when no text given' do
+      let(:section) { Ec2ssh::SshConfig::Section.new('test') }
 
-        it { expect(section.to_s).to eq '' }
-      end
+      it { expect(section.to_s).to eq '' }
+    end
 
-      context 'when empty text given' do
-        let(:section) { Ec2ssh::SshConfig::Section.new('test', '') }
+    context 'when empty text given' do
+      let(:section) { Ec2ssh::SshConfig::Section.new('test', '') }
 
-        it { expect(section.to_s).to eq '' }
-      end
+      it { expect(section.to_s).to eq '' }
+    end
 
-      context 'when some text given' do
-        let(:section) { Ec2ssh::SshConfig::Section.new('test', 'foo') }
+    context 'when some text given' do
+      let(:section) { Ec2ssh::SshConfig::Section.new('test', 'foo') }
 
-        it {
-          expect(section.to_s).to eq <<EOS
+      it {
+        expect(section.to_s).to eq <<EOS
 # section: test
 foo
 EOS
-        }
-      end
+      }
     end
   end
 end