@@ -0,0 +1,880 @@
+from tibanna.stepfunction_cost_updater import StepFunctionCostUpdater%0A%0Adef test_StepFunctionCostUpdater():%0A    sf = StepFunctionCostUpdater()%0A    assert sf.sfn_start_lambda == 'Wait'%0A    assert sf.definition%5B'States'%5D == sf.sfn_state_defs%0A    assert sf.definition%5B'StartAt'%5D == 'Wait'%0A    assert 'Wait' in sf.sfn_state_defs%0A    assert 'UpdateCostAwsem' in sf.sfn_state_defs%0A    assert 'UpdateCostDone' in sf.sfn_state_defs%0A    assert 'Done' in sf.sfn_state_defs%0A    correct_lambda_name = 'arn:aws:lambda:us-east-1:%25s:function:update_cost_awsem' %25 sf.aws_acc%0A    assert sf.sfn_state_defs%5B'UpdateCostAwsem'%5D%5B'Resource'%5D == correct_lambda_name%0A    # try changing dev suffix%0A    sf.dev_suffix = 'dev'%0A    correct_lambda_name = 'arn:aws:lambda:us-east-1:%25s:function:update_cost_awsem_dev' %25 sf.aws_acc%0A    assert sf.sfn_state_defs%5B'UpdateCostAwsem'%5D%5B'Resource'%5D == correct_lambda_name%0A

test for stepfunction_cost_updater