vscripts 0.1.0 → 0.1.2

data/spec/unit/vscripts/aws/ec2_spec.rb

@@ -0,0 +1,98 @@
+require 'spec_helper'
+require 'aws_spec_helper'
+require 'vscripts/aws'
+
+describe VScripts::AWS::EC2 do
+  subject { Object.new.extend VScripts::AWS }
+
+  let(:tags) { {'Name' => 'test'} }
+  let(:fake_instances) {::AWS::Core::Data.new([
+    {id: '1111', status: :running},
+    {id: '2222', status: :running, tags: tags},
+    {id: '3333', status: :terminated}
+  ])}
+
+  before :each do
+    allow(subject).to receive(:check_instance).and_return(true)
+    allow(subject).to receive(:instance_id).and_return(fake_instances[0].id)
+    allow(subject).to receive(:region).and_return('us-east-1')
+  end
+
+  describe '#ec2' do
+    it 'returns AWS::EC2' do
+      expect(subject.ec2).to be_an_instance_of ::AWS::EC2
+    end
+  end
+
+  describe '#instance' do
+    it 'returns AWS::EC2' do
+      expect(subject.instance).to be_an_instance_of ::AWS::EC2::Instance
+      expect(subject.instance.id).to eq(fake_instances[0].id)
+    end
+  end
+
+  describe '#all_tags' do
+    it 'returns AWS::EC2::ResourceTagCollection' do
+      expect(subject.all_tags)
+        .to be_an_instance_of ::AWS::EC2::ResourceTagCollection
+    end
+  end
+
+  describe '#tag' do
+    it 'returns the tag value' do
+      resource = ::AWS.ec2.instances['i-12345678']
+      expect(subject.create_tag(resource, 'TestTag', value: 'TestValue'))
+        .to be_an_instance_of ::AWS::EC2::Tag
+    end
+  end
+
+  describe '#create_tag' do
+    it 'creates tag' do
+      allow(subject).to receive_message_chain('instance.tags')
+        .and_return(tags)
+      expect(subject.tag('Name'))
+        .to eq('test')
+    end
+  end
+
+  describe '#tags_without' do
+    it 'returns a filtered Hash' do
+      allow(subject).to receive('all_tags')
+        .and_return(tags)
+      expect(subject.tags_without).to be_an_instance_of Hash
+      expect(subject.tags_without(['Name'])).to be_empty
+    end
+  end
+
+  describe '#name' do
+    it 'returns Hash value' do
+      allow(subject).to receive(:all_tags_hash).and_return(tags)
+      expect(subject.name).to eq(tags['Name'])
+    end
+  end
+
+  describe '#named_instances' do
+    it 'returns AWS::EC2::InstanceCollection' do
+      expect(subject.named_instances)
+        .to be_an_instance_of ::AWS::EC2::InstanceCollection
+    end
+  end
+
+  describe '#functional_instances' do
+    it 'returns an Array' do
+      allow(subject).to receive(:named_instances)
+        .and_return(fake_instances)
+      expect(subject.functional_instances).to be_an Array
+      expect(subject.functional_instances)
+        .to include(fake_instances[0], fake_instances[1])
+    end
+  end
+
+  describe '#similar_instances' do
+    it 'returns an Array' do
+      allow(subject).to receive(:functional_instances)
+        .and_return([fake_instances[0], fake_instances[1]])
+      expect(subject.similar_instances).to include(tags['Name'])
+    end
+  end
+end

data/spec/unit/vscripts/aws/metadata_spec.rb

@@ -0,0 +1,62 @@
+require 'spec_helper'
+require 'vscripts/aws'
+
+describe VScripts::AWS::Metadata do
+  subject { Class.new { extend VScripts::AWS } }
+
+  before(:each) do
+    allow(subject).to receive_message_chain('open.read')
+      .and_return('Remote server response')
+  end
+
+  describe '#metadata_url' do
+    it 'returns metadata url string' do
+      expect(subject.metadata_url).to be_a String
+    end
+  end
+
+  describe '#zone' do
+    it 'returns zone string' do
+      expect(subject.zone).to eq('Remote server response')
+    end
+  end
+
+  describe '#region' do
+    it 'returns region string' do
+      allow(subject).to receive(:zone).and_return('us-test-1a')
+      expect(subject.region).to eq('us-test-1')
+    end
+  end
+
+  describe '#instance_id' do
+    it 'returns instance id string' do
+      expect(subject.instance_id).to eq('Remote server response')
+    end
+  end
+
+  describe '#public_hostname' do
+    it 'returns public hostname string' do
+      expect(subject.public_hostname).to eq('Remote server response')
+    end
+  end
+
+  describe '#ec2_instance?' do
+    it 'returns true if metadata accessible' do
+      allow(Net::HTTP).to receive(:get_response).and_return(true)
+      expect(subject.ec2_instance?).to be true
+    end
+    it 'returns false if metadata not accessible' do
+      allow(Net::HTTP).to receive(:get_response).and_raise(StandardError)
+      expect(subject.ec2_instance?).to be false
+    end
+  end
+
+  describe '#check_instance' do
+    it 'exits if not ec2 instance' do
+      $stderr = StringIO.new
+      allow(subject).to receive(:ec2_instance?).and_return(false)
+      expect { subject.check_instance }.to raise_error(SystemExit)
+      $stderr = STDERR
+    end
+  end
+end

data/spec/{vscripts → unit/vscripts}/aws_spec.rb

@@ -1,7 +1,9 @@
+require 'spec_helper'
+require 'aws_spec_helper'
 require 'vscripts/aws'
 
 describe VScripts::AWS do
   it 'Should load default configuration for AWS SDK' do
-    expect(::AWS.config.stub_requests).to be_true
+    expect(::AWS.config.stub_requests).to be true
   end
 end

data/spec/unit/vscripts/command_line_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+require 'vscripts/command_line'
+
+describe VScripts::CommandLine do
+  subject { VScripts::CommandLine.new([cmd, 'extra_args']) }
+  let(:cmd) { VScripts::Commands.list.first.to_s }
+
+  describe '#new' do
+    it 'returns the command line arguments' do
+      expect(subject).to be_an_instance_of VScripts::CommandLine
+    end
+  end
+
+  describe '#global' do
+    it 'returns the global options as a Hash' do
+      expect(subject.global).to be_an_instance_of Hash
+    end
+  end
+
+  describe '#command' do
+    it 'returns the command name' do
+      expect(subject.command).to be_a Symbol
+      expect(subject.command).to eql cmd.capitalize.to_sym
+    end
+  end
+
+  describe '#extra' do
+    it 'returns the rest of the arguments as an Array' do
+      expect(subject.arguments).to be_an_instance_of Array
+      expect(subject.arguments).to eql ['extra_args']
+    end
+  end
+
+  describe '#verify_command' do
+    it 'fails if command not present' do
+      $stderr = StringIO.new
+      expect {
+        VScripts::CommandLine.new(['TestCommand', 'TestArguments']).command
+      }.to raise_error(SystemExit)
+      $stderr = STDERR
+    end
+  end
+end

data/spec/unit/vscripts/commands/identify_spec.rb

@@ -0,0 +1,178 @@
+require 'spec_helper'
+require 'vscripts/commands/identify'
+
+describe VScripts::Commands::Identify do
+  before :each do
+    allow(subject).to receive(:tag).and_return('TestTag')
+    allow(subject).to receive(:similar_instances)
+      .and_return(['TestTag-TestTag-1.'])
+  end
+
+  describe 'USAGE' do
+    it 'returns a String' do
+      expect(VScripts::Commands::Identify::USAGE).to be_a String
+    end
+  end
+
+  describe '#new' do
+    context 'when theme is not passed' do
+      it 'returns the default theme' do
+        expect(subject.theme).to eq('Group-Role-#')
+      end
+    end
+
+    context 'when theme is passed' do
+      it 'returns the theme' do
+        subject1 = VScripts::Commands::Identify.new(['--ec2-tag-theme=test'])
+        expect(subject1.theme).to eq('test')
+      end
+    end
+
+    context 'when host is not passed' do
+      it 'host is nil' do
+        expect(subject.host).to be_nil
+      end
+    end
+
+    context 'when host is passed' do
+      it 'returns the host' do
+        subject2 = VScripts::Commands::Identify.new(['--host=test'])
+        expect(subject2.host).to eq('test')
+      end
+    end
+
+    context 'when domain is not passed' do
+      it 'domain is nil' do
+        expect(subject.domain).to be_nil
+      end
+    end
+
+    context 'when domain is passed' do
+      it 'returns the domain' do
+        subject3 = VScripts::Commands::Identify.new(['--domain=test'])
+        expect(subject3.domain).to eq('test')
+      end
+    end
+  end
+
+  describe '#parser' do
+    it 'returns Trollop' do
+      expect(subject.parser).to be_a Trollop::Parser
+    end
+  end
+
+  describe '#cli' do
+    it 'returns cli arguments as a Hash' do
+      expect(subject.cli).to be_a Hash
+    end
+  end
+
+  describe '#theme_elements' do
+    it 'returns an array' do
+      expect(subject.theme_elements).to eq(['Group', 'Role', '#'])
+    end
+  end
+
+  describe '#map2tags' do
+    it 'returns an array' do
+      expect(subject.map2tags).to eq(['TestTag', 'TestTag', '#'])
+    end
+  end
+
+  describe '#themed_host_name' do
+    it 'returns a string' do
+      expect(subject.themed_host_name).to eq('TestTag-TestTag-#')
+    end
+  end
+
+  describe '#incremented_hostname' do
+    it 'returns a string' do
+      expect(subject.incremented_hostname).to eq('TestTag-TestTag-2')
+    end
+  end
+
+  describe '#new_hostname' do
+    context 'when host is not passed' do
+      it 'hostname is incremented' do
+        expect(subject.new_hostname).to eq('TestTag-TestTag-2')
+      end
+    end
+
+    context 'when host is passed' do
+      it 'returns a string' do
+        subject4 = VScripts::Commands::Identify.new(['--host=test'])
+        expect(subject4.new_hostname).to eq('test')
+      end
+    end
+  end
+
+  describe '#new_domain' do
+    context 'when domain is not passed' do
+      context 'when tag is not present' do
+        it 'returns a String' do
+          allow(subject).to receive(:tag)
+          allow(subject).to receive(:local_domain_name).and_return('local')
+          expect(subject.new_domain).to eq('local')
+        end
+      end
+
+      context 'when tag is present' do
+        it 'returns a String' do
+          allow(subject).to receive(:tag)
+            .and_return('Test')
+          expect(subject.new_domain).to eq('Test')
+        end
+      end
+    end
+
+    context 'when domain is passed' do
+      it 'returns a string' do
+        subject5 = VScripts::Commands::Identify.new(['--domain=test'])
+        expect(subject5.new_domain).to eq('test')
+      end
+    end
+  end
+
+  describe '#new_fqdn' do
+    it 'returns a string' do
+      expect(subject.new_fqdn).to eq('TestTag-TestTag-2.TestTag')
+    end
+  end
+
+  describe '#set_name_tag' do
+    it 'returns a string' do
+      allow(subject).to receive(:create_tag).and_return(true)
+      allow(subject).to receive(:instance).and_return(true)
+      allow(subject).to receive(:puts)
+      expect(subject.set_name_tag).to be true
+    end
+  end
+
+  describe '#set_hostname' do
+    it 'returns nil' do
+      allow(subject).to receive('`')
+      allow(subject).to receive(:write_file)
+      allow(subject).to receive(:puts)
+      allow(File).to receive('read').and_return('test')
+      expect(subject.set_hostname).to be_nil
+    end
+  end
+
+  describe '#update_hosts' do
+    it 'returns nil' do
+      allow(subject).to receive(:write_file)
+      allow(subject).to receive(:puts)
+      expect(subject.update_hosts).to be_nil
+    end
+  end
+
+  describe '#execute' do
+    it 'returns nil' do
+      allow(subject).to receive(:set_name_tag)
+      allow(subject).to receive(:set_hostname)
+      allow(subject).to receive(:update_hosts)
+      expect(STDOUT).to receive(:puts).with('Done.')
+      subject.execute
+    end
+  end
+end

data/spec/unit/vscripts/commands/tags2facts_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+require 'vscripts/commands/tags2facts'
+
+describe VScripts::Commands::Tags2facts do
+  subject { VScripts::Commands::Tags2facts.new(['extra_args']) }
+  before :each do
+    allow_any_instance_of(VScripts::Commands::Tags2facts).to receive(:check_instance)
+      .and_return(true)
+    allow_any_instance_of(VScripts::Commands::Tags2facts).to receive(:region)
+      .and_return('us-east-1')
+  end
+
+  describe '#new' do
+    it 'returns cli arguments' do
+      expect(subject.cli).to be_a Hash
+    end
+  end
+
+  describe '#parser' do
+    it 'returns parser' do
+      expect(subject.parser).to be_an_instance_of Trollop::Parser
+    end
+  end
+
+  describe '#exclude_list' do
+    it 'returns exclude list' do
+      expect(subject.exclude_list).to eq(['Name', 'Domain'])
+    end
+  end
+
+  describe '#tags_json' do
+    it 'returns JSON formatted string' do
+      $stderr = StringIO.new
+      allow(subject).to receive(:tags_without)
+        .and_return({}, { key: 'value' })
+      expect{subject.tags_json}.to raise_error(SystemExit)
+      expect(subject.tags_json).to eq("{\n  \"key\": \"value\"\n}")
+      $stderr = STDERR
+    end
+  end
+
+  describe '#execute' do
+    it 'executes the command' do
+      $stdout = StringIO.new
+      allow(subject).to receive(:tags_json)
+      allow(subject).to receive(:ensure_file_content)
+      subject.execute
+      expect($stdout.string).to match(/Writing tags to/)
+      $stdout = STDOUT
+    end
+  end
+end