Test BasicRunner halting ability

basic_runner_test.go

@@ -15,8 +15,8 @@ func TestBasicRunner_ImplRunner(t *testing.T) {
 
 func TestBasicRunner_Run(t *testing.T) {
 	data := make(map[string]interface{})
-	stepA := &TestStepAcc{"a"}
-	stepB := &TestStepAcc{"b"}
+	stepA := &TestStepAcc{Data: "a"}
+	stepB := &TestStepAcc{Data: "b"}
 
 	r := &BasicRunner{[]Step{stepA, stepB}}
 	r.Run(data)
@@ -27,3 +27,19 @@ func TestBasicRunner_Run(t *testing.T) {
 		t.Errorf("unexpected result: %#v", results)
 	}
 }
+
+func TestBasicRunner_Run_Halt(t *testing.T) {
+	data := make(map[string]interface{})
+	stepA := &TestStepAcc{Data: "a"}
+	stepB := &TestStepAcc{Data: "b", Halt: true}
+	stepC := &TestStepAcc{Data: "c"}
+
+	r := &BasicRunner{[]Step{stepA, stepB, stepC}}
+	r.Run(data)
+
+	expected := []string{"a", "b"}
+	results := data["data"].([]string)
+	if !reflect.DeepEqual(results, expected) {
+		t.Errorf("unexpected result: %#v", results)
+	}
+}

multistep_test.go

@@ -6,6 +6,9 @@ package multistep
 type TestStepAcc struct {
 	// The data inserted into the state bag.
 	Data string
+
+	// If true, it will halt at the step when it is run
+	Halt bool
 }
 
 func (s TestStepAcc) Run(state map[string]interface{}) StepAction {
@@ -17,6 +20,10 @@ func (s TestStepAcc) Run(state map[string]interface{}) StepAction {
 	data = append(data, s.Data)
 	state["data"] = data
 
+	if s.Halt {
+		return ActionHalt
+	}
+
 	return ActionContinue
 }