spec/aerosol/env_spec.rb in aerosol-1.2.0 vs spec/aerosol/env_spec.rb in aerosol-1.3.0

@@ -3,14 +3,47 @@
 describe Aerosol::Env do
   describe '#deploy' do
     let(:name) { "unique_name_#{Time.now.to_i}".to_sym }
     let!(:deploy) { Aerosol.deploy(name) { } }
 
-
     it 'adds a deploy to the list of deploys' do
       expect { subject.deploy(name) }
         .to change { subject.deploy }
         .from(nil)
         .to([deploy])
+    end
+  end
+
+  describe '#perform_role_assumption' do
+    context 'when assume_role is nil' do
+      it 'does not change the aws config' do
+        expect(Aws).to_not receive(:config)
+      end
+    end
+
+    context 'when assume_role exists' do
+      let(:assume_role) { 'arn:aws:sts::123456789123:role/role-aerosol' }
+
+      before do
+        Aerosol::AWS.sts.stub_responses(
+          :assume_role,
+          credentials: {
+            access_key_id: '123',
+            secret_access_key: '456',
+            session_token: '789',
+            expiration: Time.new + 60
+          }
+        )
+      end
+
+      after do
+        Aws.config.update(credentials: nil)
+      end
+
+      it 'should set the Aws.config[:credentials]' do
+        subject.assume_role(assume_role)
+        expect { subject.perform_role_assumption }
+          .to change { Aws.config[:credentials] }
+      end
     end
   end
 end