def test_add_connection(self, model_new, mock_ddb_table):
request = {'body': json.dumps(model_new)}
response = aws.add(request)
assert response['statusCode'] == '200'
assert response['body']
body = json.loads(response['body'])
assert domain.validate_uuid(body['connection_id'])
def test_get_connection(self, model_valid, metadata_valid, mock_ddb_table):
model_valid.update({'metadata': metadata_valid})
boto3.resource('dynamodb').Table(
'connections').get_item.return_value = {
'Item': model_valid
}
response = aws.get({
'pathParameters': {
'connection_id': model_valid['connection_id']
}
})
assert response['statusCode'] == '200'
assert response['body']
body = json.loads(response['body'])
assert domain.validate_uuid(body['connection_id'])
assert domain.validate_date_format(body['metadata']['created'])
assert domain.validate_date_format(body['metadata']['updated'])
def test_list_connections(self, model_valid, metadata_valid,
mock_ddb_table):
model_valid.update({'metadata': metadata_valid})
boto3.resource('dynamodb').Table('connections').scan.return_value = {
'Items': [model_valid, model_valid]
}
response = aws.list({})
assert response['statusCode'] == '200'
assert response['body']
body = json.loads(response['body'])
assert body.get('items')
assert len(body['items']) == 2
assert body['count'] == 2
item = body['items'][0]
assert domain.validate_uuid(item['connection_id'])
assert domain.validate_date_format(item['metadata']['created'])
assert domain.validate_date_format(item['metadata']['updated'])
def test_add_connection_valid(self, mock_ddb_table, model_new):
response = ConnectionRestApi({'body': to_json(model_new)}).add()
assert response['status_code'] == '200'
assert validate_uuid(from_json(response['body'])['connection_id'])
def delete(self, connection_id):
validate_uuid(connection_id)
self.repo.delete(connection_id)
def get(self, connection_id) -> Connection:
validate_uuid(connection_id)
connection = self.repo.get(connection_id)
if not connection:
raise ResourceNotFoundException(connection_id)
return connection