def run(self): grants = keymanager.get_grants() for grant in grants: kms_client.revoke_grant(KeyId=keymanager.get_key_id( app.config['AUTH_KEY']), GrantId=grant['GrantId']) app.logger.info('Finished revoking grants.')
def run(self): grants = keymanager.get_grants() for grant in grants: kms.revoke_grant( KeyId=keymanager.get_key_id(app.config['AUTH_KEY']), GrantId=grant['GrantId'] ) logging.info('Finished revoking grants.')
def test_get_key_id_cached(self, kms_mock): self.assertEqual( keymanager.get_key_id('mockalias'), 'mockid' )
def test_get_key_id(self, kms_mock): kms_mock.return_value = {'KeyMetadata': {'KeyId': 'mockid'}} self.assertEqual( keymanager.get_key_id('mockalias'), 'mockid' )
def test_get_key_id_cached(self, kms_mock): self.assertEqual(keymanager.get_key_id('mockalias'), 'mockid')
def test_get_key_id(self, kms_mock): kms_mock.return_value = {'KeyMetadata': {'KeyId': 'mockid'}} self.assertEqual(keymanager.get_key_id('mockalias'), 'mockid')