def test_missing_credentials(capsys):
func = MagicMock(side_effect=boto.exception.NoAuthHandlerFound())
try:
handle_exceptions(func)()
except SystemExit:
pass
out, err = capsys.readouterr()
assert 'No AWS credentials found.' in err
def test_missing_credentials(capsys):
func = MagicMock(side_effect=boto.exception.NoAuthHandlerFound())
try:
handle_exceptions(func)()
except SystemExit:
pass
out, err = capsys.readouterr()
assert 'No AWS credentials found.' in err
def test_missing_credentials(capsys):
func = MagicMock(side_effect=botocore.exceptions.NoCredentialsError())
try:
handle_exceptions(func)()
except SystemExit:
pass
out, err = capsys.readouterr()
assert "No AWS credentials found." in err
def test_expired_credentials(capsys):
func = MagicMock(side_effect=boto.exception.BotoServerError(403, 'X', {'message': '**Security Token included in the Request is expired**'}))
try:
handle_exceptions(func)()
except SystemExit:
pass
out, err = capsys.readouterr()
assert 'AWS credentials have expired.' in err
def test_expired_credentials(capsys):
func = MagicMock(side_effect=boto.exception.BotoServerError(403, 'X',
{'message': '**Security Token included in the Request is expired**'}))
try:
handle_exceptions(func)()
except SystemExit:
pass
out, err = capsys.readouterr()
assert 'AWS credentials have expired.' in err
def test_expired_credentials(capsys):
func = MagicMock(
side_effect=botocore.exceptions.ClientError(
{"Error": {"Code": "ExpiredToken", "Message": "Token expired"}}, "foobar"
)
)
try:
handle_exceptions(func)()
except SystemExit:
pass
out, err = capsys.readouterr()
assert "AWS credentials have expired." in err
# # Wait for the tunnel to be available
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
conn_test = '127.0.0.1', tunnels['postgres']
result = sock.connect_ex(conn_test)
timeout = 5
epoch_time = time.time()
threshold_time = time.time() + timeout
# # Loop until connection is established
while result != 0:
time.sleep(0.1)
result = sock.connect_ex(conn_test)
if time.time() > threshold_time:
raise Exception('Tunnel was not established within timeout of {} seconds'.format(timeout))
break
sock.close()
logging.debug('Established connectivity on tunnel after {} seconds'.format(time.time() - epoch_time))
if not options.get('background', False):
managed_processes['tunnel'] = tunnel
return tunnel.pid
if __name__ == '__main__':
handle_exceptions(cli)()
conn_test = '127.0.0.1', tunnels['postgres']
result = sock.connect_ex(conn_test)
timeout = 5
epoch_time = time.time()
threshold_time = time.time() + timeout
# # Loop until connection is established
while result != 0:
time.sleep(0.1)
result = sock.connect_ex(conn_test)
if time.time() > threshold_time:
raise Exception(
'Tunnel was not established within timeout of {} seconds'.
format(timeout))
break
sock.close()
logging.debug('Established connectivity on tunnel after {} seconds'.format(
time.time() - epoch_time))
if not options.get('background', False):
managed_processes['tunnel'] = tunnel
return tunnel.pid
if __name__ == '__main__':
handle_exceptions(cli)()
