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_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, )