Add support to ListPipelines
This commit is contained in:
parent
1b811e6949
commit
0cef332840
@ -90,6 +90,9 @@ class DataPipelineBackend(BaseBackend):
|
|||||||
self.pipelines[pipeline.pipeline_id] = pipeline
|
self.pipelines[pipeline.pipeline_id] = pipeline
|
||||||
return pipeline
|
return pipeline
|
||||||
|
|
||||||
|
def list_pipelines(self):
|
||||||
|
return self.pipelines.values()
|
||||||
|
|
||||||
def describe_pipelines(self, pipeline_ids):
|
def describe_pipelines(self, pipeline_ids):
|
||||||
pipelines = [pipeline for pipeline in self.pipelines.values() if pipeline.pipeline_id in pipeline_ids]
|
pipelines = [pipeline for pipeline in self.pipelines.values() if pipeline.pipeline_id in pipeline_ids]
|
||||||
return pipelines
|
return pipelines
|
||||||
|
@ -28,6 +28,16 @@ class DataPipelineResponse(BaseResponse):
|
|||||||
"pipelineId": pipeline.pipeline_id,
|
"pipelineId": pipeline.pipeline_id,
|
||||||
})
|
})
|
||||||
|
|
||||||
|
def list_pipelines(self):
|
||||||
|
pipelines = self.datapipeline_backend.list_pipelines()
|
||||||
|
return json.dumps({
|
||||||
|
"HasMoreResults": False,
|
||||||
|
"Marker": None,
|
||||||
|
"PipelineIdList": [
|
||||||
|
{"Id": pipeline.pipeline_id, "Name": pipeline.name} for pipeline in pipelines
|
||||||
|
]
|
||||||
|
})
|
||||||
|
|
||||||
def describe_pipelines(self):
|
def describe_pipelines(self):
|
||||||
pipeline_ids = self.parameters["pipelineIds"]
|
pipeline_ids = self.parameters["pipelineIds"]
|
||||||
pipelines = self.datapipeline_backend.describe_pipelines(pipeline_ids)
|
pipelines = self.datapipeline_backend.describe_pipelines(pipeline_ids)
|
||||||
|
@ -130,3 +130,27 @@ def test_activate_pipeline():
|
|||||||
fields = pipeline_description['Fields']
|
fields = pipeline_description['Fields']
|
||||||
|
|
||||||
get_value_from_fields('@pipelineState', fields).should.equal("SCHEDULED")
|
get_value_from_fields('@pipelineState', fields).should.equal("SCHEDULED")
|
||||||
|
|
||||||
|
|
||||||
|
@mock_datapipeline
|
||||||
|
def test_listing_pipelines():
|
||||||
|
conn = boto.datapipeline.connect_to_region("us-west-2")
|
||||||
|
res1 = conn.create_pipeline("mypipeline1", "some-unique-id1")
|
||||||
|
res2 = conn.create_pipeline("mypipeline2", "some-unique-id2")
|
||||||
|
pipeline_id1 = res1["pipelineId"]
|
||||||
|
pipeline_id2 = res2["pipelineId"]
|
||||||
|
|
||||||
|
response = conn.list_pipelines()
|
||||||
|
|
||||||
|
response["HasMoreResults"].should.be(False)
|
||||||
|
response["Marker"].should.be.none
|
||||||
|
response["PipelineIdList"].should.equal([
|
||||||
|
{
|
||||||
|
"Id": res1["pipelineId"],
|
||||||
|
"Name": "mypipeline1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Id": res2["pipelineId"],
|
||||||
|
"Name": "mypipeline2"
|
||||||
|
}
|
||||||
|
])
|
||||||
|
Loading…
x
Reference in New Issue
Block a user