Function TestImplicitInputCachePerSession

dagql/dagql_test.go:2851–2880

(t *testing.T)

}

func TestImplicitInputCachePerSession(t *testing.T) {
	srv := newExternalDagqlServerForTest(t, Query{})
	cache := newCache(t)

	var calls atomic.Int64
	dagql.Fields[Query]{
		dagql.NodeFunc("perSessionCounter", func(ctx context.Context, _ dagql.ObjectResult[Query], _ struct{}) (int, error) {
			return int(calls.Add(1)), nil
		}).WithInput(dagql.PerSessionInput),
	}.Install(srv)

	callForSession := func(sessionID string) int {
		ctx := engine.ContextWithClientMetadata(context.Background(), &engine.ClientMetadata{
			ClientID:  "same-client",
			SessionID: sessionID,
		})
		ctx = dagql.ContextWithCache(ctx, cache)
		var res int
		err := srv.Select(ctx, srv.Root(), &res, dagql.Selector{
			Field: "perSessionCounter",
		})
		require.NoError(t, err)
		return res
	}

	assert.Equal(t, callForSession("session-a"), 1)
	assert.Equal(t, callForSession("session-a"), 1)
	assert.Equal(t, callForSession("session-b"), 2)
	assert.Equal(t, callForSession("session-b"), 2)
}

func TestImplicitInputCachePerCall(t *testing.T) {
	srv := newExternalDagqlServerForTest(t, Query{})