def get_aws_credentials(federation_client, account, role): try: aws_credentials = federation_client.get_aws_credentials(account, role) except Exception as exc: error("Failed to get credentials from AWS: %s" % exc) aws_credentials['AWS_VALID_SECONDS'] = cli.get_valid_seconds(aws_credentials['AWS_EXPIRATION_DATE'], datetime.utcnow()) aws_credentials['AWS_ACCOUNT_NAME'] = account aws_credentials['AWS_ASSUMED_ROLE'] = role return aws_credentials
def test_get_valid_seconds_catches(self): future_date = 'NO_SUCH_DATE' utc_now = datetime(1970, 1, 1) self.assertEqual(get_valid_seconds(future_date, utc_now), 3600)
def test_get_valid_seconds(self): future_date = '1970-01-01T00:30:00Z' utc_now = datetime(1970, 1, 1) self.assertEqual(get_valid_seconds(future_date, utc_now), 30 * 60)
def test_get_valid_seconds(self): self.assertEqual(cli.get_valid_seconds('2016-08-16T07:45:00Z', datetime(2016, 8, 16, hour=7, minute=15)), 30*60)