99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
# Copyright 2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"). You
|
|
# may not use this file except in compliance with the License. A copy of
|
|
# the License is located at
|
|
#
|
|
# https://aws.amazon.com/apache2.0/
|
|
#
|
|
# or in the "license" file accompanying this file. This file is
|
|
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
|
|
# ANY KIND, either express or implied. See the License for the specific
|
|
# language governing permissions and limitations under the License.
|
|
|
|
import boto3
|
|
|
|
from tests import mock, unittest
|
|
|
|
|
|
class TestBoto3(unittest.TestCase):
|
|
def setUp(self):
|
|
self.session_patch = mock.patch('boto3.Session', autospec=True)
|
|
self.Session = self.session_patch.start()
|
|
|
|
def tearDown(self):
|
|
boto3.DEFAULT_SESSION = None
|
|
self.session_patch.stop()
|
|
|
|
def test_create_default_session(self):
|
|
session = self.Session.return_value
|
|
|
|
boto3.setup_default_session()
|
|
|
|
assert boto3.DEFAULT_SESSION == session
|
|
|
|
def test_create_default_session_with_args(self):
|
|
boto3.setup_default_session(
|
|
aws_access_key_id='key',
|
|
aws_secret_access_key='secret')
|
|
|
|
self.Session.assert_called_with(
|
|
aws_access_key_id='key',
|
|
aws_secret_access_key='secret')
|
|
|
|
@mock.patch('boto3.setup_default_session',
|
|
wraps=boto3.setup_default_session)
|
|
def test_client_creates_default_session(self, setup_session):
|
|
boto3.DEFAULT_SESSION = None
|
|
|
|
boto3.client('sqs')
|
|
|
|
assert setup_session.called
|
|
assert boto3.DEFAULT_SESSION.client.called
|
|
|
|
@mock.patch('boto3.setup_default_session',
|
|
wraps=boto3.setup_default_session)
|
|
def test_client_uses_existing_session(self, setup_session):
|
|
boto3.DEFAULT_SESSION = self.Session()
|
|
|
|
boto3.client('sqs')
|
|
|
|
assert not setup_session.called
|
|
assert boto3.DEFAULT_SESSION.client.called
|
|
|
|
def test_client_passes_through_arguments(self):
|
|
boto3.DEFAULT_SESSION = self.Session()
|
|
|
|
boto3.client('sqs', region_name='us-west-2', verify=False)
|
|
|
|
boto3.DEFAULT_SESSION.client.assert_called_with(
|
|
'sqs', region_name='us-west-2', verify=False)
|
|
|
|
@mock.patch('boto3.setup_default_session',
|
|
wraps=boto3.setup_default_session)
|
|
def test_resource_creates_default_session(self, setup_session):
|
|
boto3.DEFAULT_SESSION = None
|
|
|
|
boto3.resource('sqs')
|
|
|
|
assert setup_session.called
|
|
assert boto3.DEFAULT_SESSION.resource.called
|
|
|
|
@mock.patch('boto3.setup_default_session',
|
|
wraps=boto3.setup_default_session)
|
|
def test_resource_uses_existing_session(self, setup_session):
|
|
boto3.DEFAULT_SESSION = self.Session()
|
|
|
|
boto3.resource('sqs')
|
|
|
|
assert not setup_session.called
|
|
assert boto3.DEFAULT_SESSION.resource.called
|
|
|
|
def test_resource_passes_through_arguments(self):
|
|
boto3.DEFAULT_SESSION = self.Session()
|
|
|
|
boto3.resource('sqs', region_name='us-west-2', verify=False)
|
|
|
|
boto3.DEFAULT_SESSION.resource.assert_called_with(
|
|
'sqs', region_name='us-west-2', verify=False)
|