def test_get_connection(self, m_boto):
CONNECTION = "JJG"
REGION = "La bas, tout est neuf et tout est sauvage."
KEY_ID = "foo"
KEY_VALUE = "bar"
m_boto.connect_dynamodb.return_value = "JJG"
borg1 = ConnectionBorg()
borg2 = ConnectionBorg()
borg1._region = REGION
borg1._aws_access_key_id = KEY_ID
borg1._aws_secret_access_key = KEY_VALUE
self.assertIsNone(borg1._connection)
self.assertEqual(CONNECTION, borg2._get_connection())
self.assertEqual(CONNECTION, borg1._get_connection())
self.assertEqual(CONNECTION, borg1._connection)
m_boto.connect_dynamodb.assert_called_once_with(
aws_access_key_id=KEY_ID,
aws_secret_access_key=KEY_VALUE,
region=REGION,
)
def test_get_table_default(self, m_boto):
m_get_table = m_boto.connect_dynamodb.return_value.get_table
aws_access_key_id = "foo"
aws_secret_access_key = "bar"
table_name = "foo"
borg1 = ConnectionBorg()
borg1._aws_access_key_id = aws_access_key_id
borg1._aws_secret_access_key = aws_secret_access_key
ConnectionBorg().get_table(table_name)
m_get_table.assert_called_with(table_name)
m_boto.connect_dynamodb.assert_called_with(
aws_access_key_id=aws_access_key_id,
aws_secret_access_key=aws_secret_access_key,
region=None,
)
def test_get_table_eu_region(self, m_boto):
m_get_table = m_boto.connect_dynamodb.return_value.get_table
m_region = mocked_region('eu-west-1')
aws_access_key_id = "foo"
aws_secret_access_key = "bar"
table_name = "foo"
borg1 = ConnectionBorg()
borg1._region = m_region
borg1._aws_access_key_id = aws_access_key_id
borg1._aws_secret_access_key = aws_secret_access_key
ConnectionBorg().get_table(table_name)
m_get_table.assert_called_with(table_name)
m_boto.connect_dynamodb.assert_called_with(
aws_access_key_id=aws_access_key_id,
aws_secret_access_key=aws_secret_access_key,
region=m_region,
)
