Function TestWriteWithSimpleProtocol

pkg/storage/postgres/postgres_test.go:121–164

TestWriteWithSimpleProtocol is a regression test for a bug where Write operations failed with "invalid input syntax for type integer: TUPLE_OPERATION_WRITE" (SQLSTATE 22P02) when the connection uses PostgreSQL's simple query protocol (e.g. behind PgBouncer in transaction pooling mode). Setting Defau

(t *testing.T)

// the same encoding path, reproducing the failure without a real PgBouncer instance.
// See: https://github.com/openfga/openfga/issues/3011
func TestWriteWithSimpleProtocol(t *testing.T) {
	testDatastore := storagefixtures.RunDatastoreTestContainer(t, "postgres")

	uri := testDatastore.GetConnectionURI(true)
	cfg := sqlcommon.NewConfig()

	// Build the pool config the same way New() does, then override the query
	// execution mode to SimpleProtocol to mimic PgBouncer transaction-pooling mode.
	poolCfg, err := parseConfig(uri, false, cfg)
	require.NoError(t, err)
	poolCfg.ConnConfig.DefaultQueryExecMode = pgx.QueryExecModeSimpleProtocol

	primaryDB, err := pgxpool.NewWithConfig(context.Background(), poolCfg)
	require.NoError(t, err)

	ds, err := NewWithDB(primaryDB, nil, cfg)
	require.NoError(t, err)
	defer ds.Close()

	ctx := context.Background()
	storeID := ulid.Make().String()

	writes := []*openfgav1.TupleKey{
		{Object: "document:1", Relation: "viewer", User: "user:alice"},
		{Object: "document:2", Relation: "editor", User: "user:bob"},
	}

	err = ds.Write(ctx, storeID, nil, writes)
	require.NoError(t, err, "Write must succeed with SimpleProtocol (PgBouncer regression)")

	deletes := []*openfgav1.TupleKeyWithoutCondition{
		{Object: "document:1", Relation: "viewer", User: "user:alice"},
	}
	err = ds.Write(ctx, storeID, deletes, nil)
	require.NoError(t, err, "Delete must succeed with SimpleProtocol (PgBouncer regression)")

	tk, err := ds.ReadUserTuple(ctx, storeID, storage.ReadUserTupleFilter{
		Object:   "document:2",
		Relation: "editor",
		User:     "user:bob",
	}, storage.ReadUserTupleOptions{})
	require.NoError(t, err)
	require.Equal(t, "document:2", tk.GetKey().GetObject())
}

func TestPostgresDatastoreStartup(t *testing.T) {
	primaryDatastore := storagefixtures.RunDatastoreTestContainer(t, "postgres")