Function TestImplicitInputCachePerClient

dagql/dagql_test.go:2791–2820

(t *testing.T)

}

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

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

	callForClient := func(clientID string) int {
		ctx := engine.ContextWithClientMetadata(context.Background(), &engine.ClientMetadata{
			ClientID:  clientID,
			SessionID: "dagql-test-session",
		})
		ctx = dagql.ContextWithCache(ctx, cache)
		var res int
		err := srv.Select(ctx, srv.Root(), &res, dagql.Selector{
			Field: "perClientCounter",
		})
		require.NoError(t, err)
		return res
	}

	assert.Equal(t, callForClient("client-a"), 1)
	assert.Equal(t, callForClient("client-a"), 1)
	assert.Equal(t, callForClient("client-b"), 2)
	assert.Equal(t, callForClient("client-b"), 2)
}

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