def _one_pipeline_setup(pipelines_config=None, set_source_el=True):
if pipelines_config is None:
pipelines_config = _get_pipelines_config()
# override source op with a mock test class
if set_source_el:
log.info("set source=_TestSourceElement")
interpreter.Pipeline.PIPELINE_OPS['source'] = _TestSourceElement
return interpreter.get_pipelines(pipelines_config=pipelines_config)
def _one_pipeline_setup():
# override source op with a mock test class
interpreter.Pipeline.PIPELINE_OPS['source'] = _TestSourceElement
p = interpreter.get_pipelines(
pipelines_config={'pipeline_one': [{
'source': {
'uri': 'test'
}
}]})
return p
def test_get_pipelines_two():
# override source op with a mock test class
interpreter.Pipeline.PIPELINE_OPS["source"] = _TestSourceElement
p = interpreter.get_pipelines(
pipelines_config={
"pipeline_one": [{"source": {"uri": "test"}}],
"pipeline_two": [{"source": {"uri": "test2"}}],
}
)
assert isinstance(p[0], interpreter.Pipeline)
assert p[0].name == "pipeline_one"
assert isinstance(p[0]._pipe_elements[0], _TestSourceElement)
assert isinstance(p[1], interpreter.Pipeline)
assert p[0]._pipe_elements[0].config["uri"] == "test"
assert p[1].name == "pipeline_two"
assert isinstance(p[1]._pipe_elements[0], _TestSourceElement)
assert p[1]._pipe_elements[0].config["uri"] == "test2"
def test_get_pipelines_two():
# override source op with a mock test class
interpreter.Pipeline.PIPELINE_OPS['source'] = _TestSourceElement
p = interpreter.get_pipelines(
pipelines_config={
'pipeline_one': [
{'source': 'some_source'}
],
'pipeline_two': [
{'source': 'another_source'}
]
},
)
assert isinstance(p[0], interpreter.Pipeline)
assert p[0].name == 'pipeline_one'
assert isinstance(p[0].pipe_elements[0], _TestSourceElement)
assert isinstance(p[1], interpreter.Pipeline)
assert p[0].pipe_elements[0].config == 'some_source'
assert p[1].name == 'pipeline_two'
assert isinstance(p[1].pipe_elements[0], _TestSourceElement)
assert p[1].pipe_elements[0].config == 'another_source'
def test_get_pipelines_two():
# override source op with a mock test class
interpreter.Pipeline.PIPELINE_OPS['source'] = _TestSourceElement
p = interpreter.get_pipelines(
pipelines_config={
'pipeline_one': [
{'source': {'uri': 'test'}}
],
'pipeline_two': [
{'source': {'uri': 'test2'}}
]
}
)
assert isinstance(p[0], interpreter.Pipeline)
assert p[0].name == 'pipeline_one'
assert isinstance(p[0]._pipe_elements[0], _TestSourceElement)
assert isinstance(p[1], interpreter.Pipeline)
assert p[0]._pipe_elements[0].config['uri'] == 'test'
assert p[1].name == 'pipeline_two'
assert isinstance(p[1]._pipe_elements[0], _TestSourceElement)
assert p[1]._pipe_elements[0].config['uri'] == 'test2'
def test_get_pipelines_none():
p = interpreter.get_pipelines(pipelines_config=[])
assert not p
