ensure GetTransition does not panic

qmuntal · Mar 30, 2022 · ca30e11 · ca30e11
1 parent d977316
commit ca30e11
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/config.go b/config.go
@@ -14,7 +14,8 @@ func withTransition(ctx context.Context, transition Transition) context.Context
 // GetTransition returns the transition from the context.
 // If there is no transition the returned value is empty.
 func GetTransition(ctx context.Context) Transition {
-	return ctx.Value(transitionKey{}).(Transition)
+	tr, _ := ctx.Value(transitionKey{}).(Transition)
+	return tr
 }
 
 // ActionFunc describes a generic action function.

diff --git a/statemachine_test.go b/statemachine_test.go
@@ -1369,3 +1369,8 @@ func TestStateMachine_Firing_Concurrent(t *testing.T) {
 	wg.Wait()
 	assert.False(t, sm.Firing())
 }
+
+func TestGetTransition_ContextEmpty(t *testing.T) {
+	// It should not panic
+	GetTransition(context.Background())
+}