vominator 0.0.1

data/spec/lib/vominator/instances_spec.rb

@@ -0,0 +1,96 @@
+require 'spec_helper'
+require 'vominator/instances'
+
+describe Vominator::Instances do
+  describe 'get_instances' do
+    context 'when I pass a valid environment and product' do
+      let (:instances) {Vominator::Instances.get_instances('test', 'sample-api')}
+
+      subject { instances }
+
+      it 'should not raise an error' do
+        expect { instances }.to_not raise_error
+      end
+
+      it 'should contain 3 sample api instances' do
+        # noinspection RubyResolve
+        expect(instances).to contain_exactly({'sample-api-1' => nil,
+                                      'type' => {
+                                          'prod' => 'm3.medium',
+                                          'staging' => 'm3.medium'},
+                                     'ami'=>'ami-123123',
+                                     'family'=>'linux',
+                                     'ip'=>'10.OCTET.41.21',
+                                     'az'=>'us-east-1c',
+                                     'environment'=>['staging'],
+                                     'security_groups'=>['sample-api-server'],
+                                     'chef_recipes'=>['srv_sample_api']
+                                     },
+                                     {'sample-api-2' => nil,
+                                      'type' => {
+                                          'prod' => 'm3.medium',
+                                          'staging' => 'm3.medium'},
+                                      'family'=>'linux',
+                                      'ip'=>'10.OCTET.42.21',
+                                      'az'=>'us-east-1d',
+                                      'security_groups'=>['sample-api-server'],
+                                      'chef_recipes'=>['srv_sample_api']
+                                     },
+                                     {'sample-api-3' => nil,
+                                      'type' => {
+                                          'prod' => 'm3.medium',
+                                          'staging' => 'm3.medium'},
+                                      'family'=>'linux',
+                                      'ip'=>'10.OCTET.43.21',
+                                      'az'=>'us-east-1e',
+                                      'security_groups'=>['sample-api-server'],
+                                      'chef_recipes'=>['srv_sample_api']
+                                     })
+      end
+    end
+
+    context 'when I pass a filter' do
+      let (:instances) {Vominator::Instances.get_instances('test', 'sample-api', ['sample-api-1'])}
+
+      subject { instances }
+
+      it 'it should filter the results' do
+        expect(instances).to contain_exactly({'sample-api-1' => nil,
+                                              'type' => {
+                                                  'prod' => 'm3.medium',
+                                                  'staging' => 'm3.medium'},
+                                              'ami'=>'ami-123123',
+                                              'family'=>'linux',
+                                              'ip'=>'10.OCTET.41.21',
+                                              'az'=>'us-east-1c',
+                                              'environment'=>['staging'],
+                                              'security_groups'=>['sample-api-server'],
+                                              'chef_recipes'=>['srv_sample_api']
+                                             })
+
+      end
+    end
+    context 'when I pass an invalid environment or product' do
+      xit 'it should do something'
+    end
+
+  end
+
+  describe 'generate_cloud_config' do
+    context 'when I pass a valid hostname, environment, family, roles, and recipes' do
+      let (:cloud_config) { Vominator::Instances.generate_cloud_config('sample-api-1', 'test', 'linux', ['role1'], ['recipe1'] )}
+
+      subject { cloud_config }
+      it 'should return a rendered cloud_config' do
+        expect {cloud_config}.to_not raise_error
+        expect(cloud_config).to include('sample-api-1.test')
+        expect(cloud_config).to include('role1')
+        expect(cloud_config).to include('recipe1')
+      end
+    end
+
+    context 'when I pass invalid parameters' do
+      xit 'it should do something'
+    end
+  end
+end

data/spec/lib/vominator/route53_spec.rb

@@ -0,0 +1,64 @@
+require 'spec_helper'
+require 'vominator/route53'
+
+describe Vominator::Route53 do
+  before(:all) do
+    @puke_variables = Vominator.get_puke_variables('test')
+    Aws.config[:stub_responses] = true
+    @r53 = Aws::Route53::Client.new
+    @r53.stub_responses(:list_resource_record_sets, { :is_truncated => false, :max_items => 50, :resource_record_sets => [
+                                                      { :name => 'sample-api-1.test.example.com.', :type => 'A', :resource_records => [{ :value => '10.203.41.21'}] },
+                                                      { :name => 'sample-api-2.test.example.com.', :type => 'A', :resource_records => [{ :value => '10.203.42.21'}] },
+                                                      { :name => 'sample-api-3.test.example.com.', :type => 'A', :resource_records => [{ :value => '10.203.43.21'}] }
+                                                  ]}
+    )
+  end
+
+  describe 'get_records' do
+    context 'when I pass a valid route53 client and an invalid zone file' do
+      xit 'should do something'
+    end
+
+    context 'when I pass an invalid route53 client and a valid zone file' do
+      xit 'should do something'
+    end
+
+    context 'when I pass an invalid route53 client and an invalid zone file' do
+      xit 'should do something'
+    end
+
+    context 'when I pass a valid route53 client and zone file' do
+      let(:records) { Vominator::Route53.get_records(@r53,"/hostedzone/#{@puke_variables['zone']}", 2)}
+
+      subject { records }
+
+      it { is_expected.to include('sample-api-1.test.example.com.') }
+      it { is_expected.to include('sample-api-2.test.example.com.') }
+      it { is_expected.to include('sample-api-3.test.example.com.') }
+    end
+  end
+
+  describe 'create_record' do
+    context 'when I pass a valid route53 client, zone, fqdn, and ip' do
+      let (:response) { Vominator::Route53.create_record(@r53,"#{@puke_variables['zone']}", 'sample-api-1.test.example.com', '10.203.41.21')}
+
+      subject { response }
+
+      it 'should return true' do
+        @r53.stub_responses(:change_resource_record_sets, :change_info => { :status => 'PENDING', :id => '12345', :submitted_at => Time.now})
+      end
+    end
+  end
+
+  describe 'delete_record' do
+    context 'When I pass a valid route53 client, zone, fqdn, and ip' do
+      let (:response) { Vominator::Route53.delete_record(@r53, "#{@puke_variables['zone']}",'sample-api-1.test.example.com', '10.203.41.21')}
+
+      subject { response }
+
+      it 'should return true' do
+        @r53.stub_responses(:change_resource_record_sets, :change_info => { :status => 'PENDING', :id => '12345', :submitted_at => Time.now})
+      end
+    end
+  end
+end

data/spec/lib/vominator/ssm_spec.rb

@@ -0,0 +1,95 @@
+require 'spec_helper'
+require 'vominator/ssm'
+
+describe Vominator::SSM do
+  before(:each) do
+    Aws.config[:stub_responses] = true
+    @ssm = Aws::SSM::Client.new
+  end
+
+  describe 'get_documents' do
+    context 'when I pass a valid client' do
+      let (:documents) { Vominator::SSM.get_documents(@ssm,2)}
+
+      subject { documents }
+
+      it 'returns an array of document names' do
+        @ssm.stub_responses(:list_documents, {:next_token => 'abcd', :document_identifiers => [{:name => 'document1'},{:name => 'document2'}]},
+                            {:next_token => nil, :document_identifiers => [{:name => 'document3'},{:name => 'document4'}]})
+
+        expect {documents}.to_not raise_error
+        expect(documents.count).to eq 4
+        expect(documents).to include('document3')
+      end
+    end
+  end
+
+  describe 'describe_document' do
+    context 'when I pass a valid client and name' do
+      let (:document) { Vominator::SSM.describe_document(@ssm,'document-3')}
+
+      subject { document }
+
+      it 'returns details about a document' do
+        @ssm.stub_responses(:describe_document, :document => {:created_date => Time.now, :name => 'document-3', :status => 'active', :sha_1 => 'abcd'})
+        expect { document}.to_not raise_error
+        expect(document.name).to match 'document-3'
+      end
+    end
+  end
+
+  describe 'get_document' do
+    context 'when I pass a valid client and name' do
+      let (:document) { Vominator::SSM.get_document(@ssm, 'document-3')}
+
+      subject { document }
+
+      it 'returns the document' do
+        @ssm.stub_responses(:get_document, {:content => '{"schemaVersion":"1.0","description":"test policy"}', :name => 'document-3'})
+        expect { document }.to_not raise_error
+        expect(document.name).to match 'document-3'
+        expect(document.content).to match '{"schemaVersion":"1.0","description":"test policy"}'
+      end
+    end
+  end
+  describe 'put_document' do
+    context 'when I pass a valid client, name, and data' do
+      let (:document) { Vominator::SSM.put_document(@ssm, 'document-5','{"schemaVersion":"1.0","description":"test policy"}') }
+
+      subject { document }
+
+      it 'creates the document' do
+        @ssm.stub_responses(:describe_document, :document => {:created_date => Time.now, :name => 'document-5', :status => 'active', :sha_1 => 'abcd'})
+        expect { document }.to_not raise_error
+      end
+    end
+  end
+
+  describe 'associated' do
+    context 'when I pass a valid client, name and instance_id and is associated to the instance' do
+      let (:association) { Vominator::SSM.associated?(@ssm, 'document-1','i-123456')}
+
+      subject { association }
+
+      it 'returns true' do
+        @ssm.stub_responses(:describe_association, :association_description => {:date => Time.now, :instance_id => 'i-123456', :name => 'document-1', :status => {:name => 'Success', :date => Time.now, :message => 'It worked'}})
+        expect { association }.to_not raise_error
+        expect(association).to be true
+      end
+    end
+  end
+
+  describe 'create_association' do
+    context 'when I pass a valid client, name, and instance_id' do
+      let (:association) { Vominator::SSM.create_association(@ssm, 'document-1','i-123456')}
+
+      subject { association }
+
+      it 'returns true' do
+        @ssm.stub_responses(:describe_association, :association_description => {:date => Time.now, :instance_id => 'i-123456', :name => 'document-1', :status => {:name => 'Success', :date => Time.now, :message => 'It worked'}})
+        expect { association }.to_not raise_error
+        expect(association).to be true
+      end
+    end
+  end
+end

data/spec/lib/vominator/vominator_spec.rb

@@ -0,0 +1,209 @@
+require 'spec_helper'
+require 'vominator/vominator'
+
+describe Vominator do
+
+  describe 'get_config' do
+    context 'when I pass an invalid config file' do
+      let(:vominator_config) { Vominator.get_config('/badpath') }
+
+      subject { vominator_config }
+
+      it 'should exit' do
+        cached_vominator_config = ENV.delete('VOMINATOR_CONFIG')
+	expect{ subject }.to exit_with_code(1)
+        ENV['VOMINATOR_CONFIG'] = cached_vominator_config
+      end
+
+    end
+
+    context 'when I pass a valid config file' do
+      let(:vominator_config) { Vominator.get_config('test/vominator.yaml') }
+
+      subject { vominator_config }
+
+      it { is_expected.not_to be false }
+      it { is_expected.to include('access_key_id' => 'DUMMY_ACCESS_KEY') }
+      it { is_expected.to include('secret_access_key' => 'DUMMY_SECRET_KEY') }
+      it { is_expected.to include('configuration_path' => 'test/puke') }
+      it { is_expected.to include('key_pair_name' => 'ci@example.com') }
+      it { is_expected.to include('chef_client_key' => 'ci.pem') }
+      it { is_expected.to include('chef_client_name' => 'ci') }
+    end
+  end
+
+  describe 'get_puke_config' do
+
+    context 'when I pass an invalid directory' do
+      let (:puke_config) { Vominator.get_puke_config('/badpath')}
+
+      subject { puke_config }
+
+      it 'should raise an error' do
+        expect { puke_config }.to raise_error
+      end
+    end
+
+    context 'when I pass a valid directory' do
+      let (:puke_config) { Vominator.get_puke_config(VOMINATOR_CONFIG['configuration_path'])}
+
+      subject { puke_config }
+
+      it 'should not raise an error' do
+        expect { puke_config }.to_not raise_error
+      end
+    end
+  end
+
+  describe 'get_key_pair' do
+    context 'when I pass a valid Vominator config' do
+      vominator_config = Vominator.get_config('test/vominator.yaml')
+      let (:key_pair) { Vominator.get_key_pair(vominator_config) }
+
+      subject { key_pair }
+
+      it 'should return a key_pair name' do
+        expect { key_pair }.to_not raise_error
+        expect(key_pair).to match 'ci@example.com'
+      end
+    end
+  end
+  describe 'get_puke_variables' do
+
+    context 'when I pass an invalid environment' do
+      let (:puke_variables) { Vominator.get_puke_variables('invalid_environment')}
+
+      subject { puke_variables }
+
+      xit 'should raise an error' do
+        expect { puke_variables }.to raise_error
+      end
+    end
+
+    context 'when I pass a valid environment' do
+      let (:puke_variables) { Vominator.get_puke_variables('test')}
+
+      subject { puke_variables }
+
+      it 'should not raise an error' do
+        expect { puke_variables }.to_not raise_error
+      end
+    end
+  end
+
+  describe 'yesno' do
+    context 'default is yes' do
+      let (:yesno) { Vominator.yesno? }
+      context 'when I say yes' do
+        it 'should prompt the user and return true' do
+          expect($terminal).to receive(:ask).with('Continue? [Y/n] ').and_return('y')
+          expect(yesno).to be_truthy
+        end
+      end
+
+      context 'when I say no' do
+        it 'should prompt the user and return false' do
+          expect($terminal).to receive(:ask).with('Continue? [Y/n] ').and_return('n')
+          expect(yesno).to be_falsey
+        end
+      end
+    end
+
+    context 'default is no' do
+      let (:yesno) { Vominator.yesno?(default: false) }
+      context 'when I say yes' do
+        it 'should prompt the user and return true' do
+          expect($terminal).to receive(:ask).with('Continue? [y/N] ').and_return('y')
+          expect(yesno).to be_truthy
+        end
+      end
+
+      context 'when I say no' do
+        it 'should prompt the user and return false' do
+          expect($terminal).to receive(:ask).with('Continue? [y/N] ').and_return('n')
+          expect(yesno).to be_falsey
+        end
+      end
+    end
+  end
+end
+
+describe Vominator::Logger do
+  describe 'info' do
+    context 'when I pass a log message' do
+      message = 'This is a test message'
+      let (:info_log) { Vominator::Logger.info(message) }
+
+      subject { info_log }
+
+      it 'should print the log message' do
+        expect { info_log }.to output("#{message}\n").to_stdout
+      end
+    end
+  end
+
+  describe 'test' do
+    context 'when I pass a log message' do
+      message = 'This is a test message'
+      let (:test_log) { Vominator::Logger.test(message) }
+
+      subject { test_log }
+
+      it 'should print the log message' do
+        expect { test_log }.to output("\e[36m#{message}\e[0m\n").to_stdout
+      end
+    end
+  end
+
+  describe 'error' do
+    context 'when I pass a log message' do
+      message = 'This is a test message'
+      let (:error_log) { Vominator::Logger.error(message) }
+
+      subject { error_log }
+
+      it 'should print the log message' do
+        expect { error_log }.to output("\e[31m#{message}\e[0m\n").to_stdout
+      end
+    end
+  end
+
+  describe 'fatal' do
+    context 'when I pass a log message' do
+      message = 'This is a test message'
+      let (:fatal_log) { Vominator::Logger.fatal(message) }
+
+      subject { fatal_log }
+
+      it 'should print the log message and exit' do
+        expect { fatal_log }.to exit_with_code(1).and output("\e[31m#{message}\e[0m\n").to_stdout 
+      end
+    end
+  end
+
+  describe 'success' do
+    context 'when I pass a log message' do
+      message = 'This is a test message'
+      let (:success_log) { Vominator::Logger.success(message) }
+
+      subject { success_log }
+
+      it 'should print the log message' do
+        expect { success_log }.to output("\e[32m#{message}\e[0m\n").to_stdout
+      end
+    end
+  end
+
+  describe 'warning' do
+    context 'when I pass a log message' do
+      message = 'This is a test message'
+      let (:warning_log) { Vominator::Logger.warning(message) }
+
+      subject { warning_log }
+
+      it 'should print the log message' do
+        expect { warning_log }.to output("\e[33m#{message}\e[0m\n").to_stdout
+      end
+    end
+  end
+end