Function SetupState

acceptance/testenv/testenv.go:148–181

SetupState initializes the given state S and stores it in Context under S.Key(), if invoked twice for the same S.Key() the state will be set to the existing value from Context. If the test environment is being restored, values from the persisted file will be loaded.

(ctx context.Context, state **S)

// twice for the same S.Key() the state will be set to the existing value from Context. If the
// test environment is being restored, values from the persisted file will be loaded.
func SetupState[S WithState](ctx context.Context, state **S) (context.Context, error) {
	p := ctx.Value(persistedEnv)

	if p == nil {
		p = &map[string]any{}
	}

	// we need to new() to be able to invoke S.Key()
	newS := new(S)
	key := persistedKey(*newS)

	store := *p.(*map[string]any)

	existing := store[key]

	if existing == nil {
		*state = newS
		if i, ok := any(*state).(Initializing); ok {
			i.Initialize()
		}
	} else {
		*state = existing.(*S)
	}

	store[key] = *state

	if Restored(ctx) {
		if err := restoreInto(key, &state); err != nil {
			return ctx, err
		}
	}

	return context.WithValue(ctx, persistedEnv, p), nil
}

// FetchState fetches the state from the Context stored under S.Key()
func FetchState[S WithState](ctx context.Context) *S {