Function TestSharedState_ContextAwareVariants

shared_state_test.go:222–258

(t *testing.T)

}

func TestSharedState_ContextAwareVariants(t *testing.T) {
	t.Parallel()

	type testContextKey string

	ctxKey := testContextKey("tenant")
	store := &contextCheckingStorage{ctxKey: ctxKey, base: newSharedStateMemoryStorage(t)}
	app := New(Config{SharedStorage: store})

	t.Run("missing context", func(t *testing.T) {
		t.Parallel()

		err := app.SharedState().SetJSONWithContext(context.Background(), "key", Map{"ok": true}, time.Second)
		require.Error(t, err)
	})

	t.Run("context propagation", func(t *testing.T) {
		t.Parallel()

		ctx := context.WithValue(context.Background(), ctxKey, "value")
		err := app.SharedState().SetJSONWithContext(ctx, "key", Map{"ok": true}, time.Second)
		require.NoError(t, err)

		var out map[string]bool
		_, found, err := app.SharedState().GetJSONWithContext(ctx, "key", &out)
		require.NoError(t, err)
		require.True(t, found)
		require.True(t, out["ok"])

		has, err := app.SharedState().HasWithContext(ctx, "key")
		require.NoError(t, err)
		require.True(t, has)

		err = app.SharedState().DeleteWithContext(ctx, "key")
		require.NoError(t, err)
	})
}

func TestSharedState_KeyNamespacing(t *testing.T) {
	t.Parallel()