def test_many_instance_profiles(self): client = MockIAMClient() max_items = client.DEFAULT_MAX_ITEMS for i in range(2 * max_items): client.create_instance_profile(InstanceProfileName=('ip-%03d' % i)) paginator = client.get_paginator('list_instance_profiles') pages = list(paginator.paginate()) self.assertEqual(len(pages), 2) self.assertEqual(len(pages[0]['InstanceProfiles']), max_items) self.assertEqual(len(pages[1]['InstanceProfiles']), max_items)
def test_many_roles(self): client = MockIAMClient() max_items = client.DEFAULT_MAX_ITEMS for i in range(2 * max_items + 1): client.create_role( AssumeRolePolicyDocument=json.dumps(_MRJOB_SERVICE_ROLE), RoleName=('r-%03d' % i),) paginator = client.get_paginator('list_roles') pages = list(paginator.paginate()) self.assertEqual(len(pages), 3) self.assertEqual(len(pages[0]['Roles']), max_items) self.assertEqual(len(pages[1]['Roles']), max_items) self.assertEqual(len(pages[2]['Roles']), 1)
def test_many_roles(self): client = MockIAMClient() max_items = client.DEFAULT_MAX_ITEMS for i in range(2 * max_items + 1): client.create_role( AssumeRolePolicyDocument=json.dumps(_MRJOB_SERVICE_ROLE), RoleName=('r-%03d' % i), ) paginator = client.get_paginator('list_roles') pages = list(paginator.paginate()) self.assertEqual(len(pages), 3) self.assertEqual(len(pages[0]['Roles']), max_items) self.assertEqual(len(pages[1]['Roles']), max_items) self.assertEqual(len(pages[2]['Roles']), 1)