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