(t *testing.T)
| 999 | } |
| 1000 | |
| 1001 | func TestSendBatchSimpleProtocol(t *testing.T) { |
| 1002 | t.Parallel() |
| 1003 | |
| 1004 | ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second) |
| 1005 | defer cancel() |
| 1006 | |
| 1007 | config := mustParseConfig(t, os.Getenv("PGX_TEST_DATABASE")) |
| 1008 | config.DefaultQueryExecMode = pgx.QueryExecModeSimpleProtocol |
| 1009 | |
| 1010 | conn := mustConnect(t, config) |
| 1011 | defer closeConn(t, conn) |
| 1012 | |
| 1013 | var batch pgx.Batch |
| 1014 | batch.Queue("SELECT 1::int") |
| 1015 | batch.Queue("SELECT 2::int; SELECT $1::int", 3) |
| 1016 | results := conn.SendBatch(ctx, &batch) |
| 1017 | rows, err := results.Query() |
| 1018 | assert.NoError(t, err) |
| 1019 | assert.True(t, rows.Next()) |
| 1020 | values, err := rows.Values() |
| 1021 | assert.NoError(t, err) |
| 1022 | assert.EqualValues(t, 1, values[0]) |
| 1023 | assert.False(t, rows.Next()) |
| 1024 | |
| 1025 | rows, err = results.Query() |
| 1026 | assert.NoError(t, err) |
| 1027 | assert.True(t, rows.Next()) |
| 1028 | values, err = rows.Values() |
| 1029 | assert.NoError(t, err) |
| 1030 | assert.EqualValues(t, 2, values[0]) |
| 1031 | assert.False(t, rows.Next()) |
| 1032 | |
| 1033 | rows, err = results.Query() |
| 1034 | assert.NoError(t, err) |
| 1035 | assert.True(t, rows.Next()) |
| 1036 | values, err = rows.Values() |
| 1037 | assert.NoError(t, err) |
| 1038 | assert.EqualValues(t, 3, values[0]) |
| 1039 | assert.False(t, rows.Next()) |
| 1040 | } |
| 1041 | |
| 1042 | // https://github.com/jackc/pgx/issues/1847#issuecomment-2347858887 |
| 1043 | func TestConnSendBatchErrorDoesNotLeaveOrphanedPreparedStatement(t *testing.T) { |
nothing calls this directly
no test coverage detected