def test_rollback_no_pipeline():
pth = PipelineTransactionHandler()
with pytest.raises(TransactionException):
pth.rollback()
def test_execute_no_pipeline():
pth = PipelineTransactionHandler()
with pytest.raises(TransactionException):
pth.execute()
def test_rollback():
pipeline = ActionsPipelineMock()
pth = PipelineTransactionHandler(pipeline)
pth.rollback()
assert not pipeline.do.called
assert pipeline.undo.called
def test_prepare_commit():
pth = PipelineTransactionHandler()
assert pth.prepare_commit()
def test_pipeline_name_setter():
pth = PipelineTransactionHandler()
with pytest.raises(TransactionException):
pth.pipeline_name = "pipeline_name"
def test_pipeline_name():
pth = PipelineTransactionHandler(ActionsPipelineMock("pipeline_name"))
assert pth.pipeline_name == "pipeline_name"
def test_actions_pipeline():
pth = PipelineTransactionHandler()
pipeline = ActionsPipelineMock()
pth.actions_pipeline = pipeline
assert pth.actions_pipeline == pipeline
def test_init():
assert PipelineTransactionHandler(ActionsPipelineMock())