from tests import unittest
import botocore.session
class TestCanChangeParsing(unittest.TestCase):
def setUp(self):
self.session = botocore.session.get_session()
def test_can_change_timestamp_with_clients(self):
factory = self.session.get_component('response_parser_factory')
factory.set_parser_defaults(timestamp_parser=lambda x: str(x))
s3 = self.session.create_client('s3', 'us-west-2')
parsed = s3.list_buckets()
dates = [bucket['CreationDate'] for bucket in parsed['Buckets']]
self.assertTrue(all(isinstance(date, str) for date in dates),
"Expected all str types but instead got: %s" % dates)
def test_maps_service_name_when_overriden(self):
ses = self.session.get_service_model('ses')
self.assertEqual(ses.endpoint_prefix, 'email')
self.assertEqual(ses.service_name, 'ses')
def test_maps_service_name_from_client(self):
client = self.session.create_client('ses', region_name='us-east-1')
self.assertEqual(client.meta.service_model.service_name, 'ses')