Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added description and tags for create_pipeline

Browse Source
This commit is contained in:
Andrii Piasetskyi 2017-04-23 19:20:10 +03:00
parent d414ecd211
commit ee96c20034
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/datapipeline/models.py
View File

@ -26,7 +26,7 @@ class Pipeline(BaseModel):
def __init__(self, name, unique_id, **kwargs): def __init__(self, name, unique_id, **kwargs):
self.name = name self.name = name
self.unique_id = unique_id self.unique_id = unique_id
self.description = "" self.description = kwargs.get('description', '')
self.pipeline_id = get_random_pipeline_id() self.pipeline_id = get_random_pipeline_id()
self.creation_time = datetime.datetime.utcnow() self.creation_time = datetime.datetime.utcnow()
self.objects = [] self.objects = []

7
moto/datapipeline/responses.py
View File

@ -21,10 +21,11 @@ class DataPipelineResponse(BaseResponse):
return datapipeline_backends[self.region] return datapipeline_backends[self.region]
def create_pipeline(self): def create_pipeline(self):
name = self.parameters['name'] name = self.parameters.get('name')
unique_id = self.parameters['uniqueId'] unique_id = self.parameters.get('uniqueId')
description = self.parameters.get('description', '')
tags = self.parameters.get('tags', []) tags = self.parameters.get('tags', [])
pipeline = self.datapipeline_backend.create_pipeline(name, unique_id, tags=tags) pipeline = self.datapipeline_backend.create_pipeline(name, unique_id, description=description, tags=tags)
return json.dumps({ return json.dumps({
"pipelineId": pipeline.pipeline_id, "pipelineId": pipeline.pipeline_id,
}) })