Skip to content

Commit

Permalink
Adding more auth options
Browse files Browse the repository at this point in the history
  • Loading branch information
JamieCressey committed Dec 6, 2015
1 parent 1ee9081 commit ebcf1d7
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions beaver/tests/test_sqs_transport.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,10 +39,24 @@ def setUpClass(cls):
output_file = Fixture.download_official_distribution()
Fixture.extract_distribution(output_file)

@mock_sqs
def test_sqs_default_auth_profile(cls):
cls._create_queues()
cls.beaver_config.set('sqs_aws_profile_name', None)
cls.beaver_config.set('sqs_aws_access_key', None)
cls.beaver_config.set('sqs_aws_secret_key', None)
cls.beaver_config.set('sqs_aws_queue', 'queue1')

transport = create_transport(cls.beaver_config, logger=cls.logger)

cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport)

@mock_sqs
def test_sqs_auth_profile(cls):
cls._create_queues()
cls.beaver_config.set('sqs_aws_profile_name', 'beaver_queue')
cls.beaver_config.set('sqs_aws_access_key', None)
cls.beaver_config.set('sqs_aws_secret_key', None)
cls.beaver_config.set('sqs_aws_queue', 'queue1')

transport = create_transport(cls.beaver_config, logger=cls.logger)
Expand All @@ -61,6 +75,19 @@ def test_sqs_auth_key(cls):

cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport)

@mock_sqs
def test_sqs_auth_account_id(cls):
cls._create_queues()
cls.beaver_config.set('sqs_aws_queue_owner_acct_id', 'abc123')
cls.beaver_config.set('sqs_aws_profile_name', None)
cls.beaver_config.set('sqs_aws_access_key', 'beaver_test_key')
cls.beaver_config.set('sqs_aws_secret_key', 'beaver_test_secret')
cls.beaver_config.set('sqs_aws_queue', 'queue1')

transport = create_transport(cls.beaver_config, logger=cls.logger)

cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport)

@mock_sqs
def test_sqs_single_queue(cls):
cls._create_queues()
Expand Down

0 comments on commit ebcf1d7

Please sign in to comment.