kotovalexarian-likes-github/fog--fog-aws
1
0
Fork 0
mirror of https://github.com/fog/fog-aws.git synced 2022-11-09 13:50:52 -05:00
Code Releases Activity

Merge pull request #374 from v-yarotsky/vy-fix-mocking

Browse source 
Fix AWS credential mocking
This commit is contained in:
Wesley Beary 2017-07-06 11:06:44 -05:00 • committed by GitHub
commit e91cd8e906
2 changed files with 16 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/fog/aws/credential_fetcher.rb
View file

@ -11,7 +11,7 @@ module Fog
module ServiceMethods
def fetch_credentials(options)
if options[:use_iam_profile] && Fog.mocking?
Fog::Compute::AWS::Mock.data[:iam_role_based_creds]
return Fog::Compute::AWS::Mock.data[:iam_role_based_creds]
end
if options[:use_iam_profile]
begin

19
tests/credentials_tests.rb
View file

@ -2,10 +2,9 @@ Shindo.tests('AWS | credentials', ['aws']) do
old_mock_value = Excon.defaults[:mock]
fog_was_mocked = Fog.mocking?
Excon.stubs.clear
Fog.mock!
Fog.unmock!
begin
Excon.defaults[:mock] = true
default_credentials = Fog::Compute::AWS.fetch_credentials({})
Excon.stub({:method => :get, :path => "/latest/meta-data/iam/security-credentials/"}, {:status => 200, :body => 'arole'})
Excon.stub({:method => :get, :path => "/latest/meta-data/placement/availability-zone/"}, {:status => 200, :body => 'us-west-1a'})
@ -17,7 +16,6 @@ Shindo.tests('AWS | credentials', ['aws']) do
'Expiration' => expires_at.xmlschema
}
Fog::Compute::AWS::Mock.data[:iam_role_based_creds] = credentials
Excon.stub({:method => :get, :path => "/latest/meta-data/iam/security-credentials/arole"}, {:status => 200, :body => Fog::JSON.encode(credentials)})
tests("#fetch_credentials") do
@ -60,16 +58,29 @@ Shindo.tests('AWS | credentials', ['aws']) do
end
Fog::Time.now = Time.now
default_credentials = Fog::Compute::AWS.fetch_credentials({})
tests("#fetch_credentials when the url 404s") do
Excon.stub({:method => :get, :path => "/latest/meta-data/iam/security-credentials/"}, {:status => 404, :body => 'not bound'})
Excon.stub({:method => :get, :path => "/latest/meta-data/placement/availability-zone/"}, {:status => 400, :body => 'not found'})
returns(default_credentials) {Fog::Compute::AWS.fetch_credentials(:use_iam_profile => true)}
end
mocked_credentials = {
:aws_access_key_id => "access-key-id",
:aws_secret_access_key => "secret-access-key",
:aws_session_token => "session-token",
:aws_credentials_expire_at => Time.at(Time.now.to_i + 500).xmlschema
}
tests("#fetch_credentials when mocking") do
Fog.mock!
Fog::Compute::AWS::Mock.data[:iam_role_based_creds] = mocked_credentials
returns(mocked_credentials) {Fog::Compute::AWS.fetch_credentials(:use_iam_profile => true)}
end
ensure
ENV["AWS_CONTAINER_CREDENTIALS_RELATIVE_URI"] = nil
Excon.stubs.clear
Excon.defaults[:mock] = old_mock_value
Fog.unmock! if !fog_was_mocked
Fog.mock! if fog_was_mocked
end
end