| 308 | } |
| 309 | |
| 310 | func testBatch(ctx context.Context, r *reporter) { |
| 311 | fmt.Println("\n=== Batch queries ===") |
| 312 | suite := "batch" |
| 313 | |
| 314 | conn, err := connect(ctx) |
| 315 | if err != nil { |
| 316 | r.report(suite, "connect", "fail", err.Error()) |
| 317 | return |
| 318 | } |
| 319 | defer conn.Close(ctx) |
| 320 | |
| 321 | batch := &pgx.Batch{} |
| 322 | batch.Queue("SELECT 1 AS a") |
| 323 | batch.Queue("SELECT 'hello' AS b") |
| 324 | batch.Queue("SELECT 42 AS c") |
| 325 | |
| 326 | br := conn.SendBatch(ctx, batch) |
| 327 | |
| 328 | // Result 1 |
| 329 | rows1, err := br.Query() |
| 330 | if err != nil { |
| 331 | r.report(suite, "batch_query_1", "fail", err.Error()) |
| 332 | } else { |
| 333 | if rows1.Next() { |
| 334 | var a int |
| 335 | if err := rows1.Scan(&a); err != nil { |
| 336 | r.report(suite, "batch_query_1", "fail", err.Error()) |
| 337 | } else if a == 1 { |
| 338 | r.report(suite, "batch_query_1", "pass", "") |
| 339 | } else { |
| 340 | r.report(suite, "batch_query_1", "fail", fmt.Sprintf("expected 1, got %d", a)) |
| 341 | } |
| 342 | } |
| 343 | rows1.Close() |
| 344 | } |
| 345 | |
| 346 | // Result 2 |
| 347 | rows2, err := br.Query() |
| 348 | if err != nil { |
| 349 | r.report(suite, "batch_query_2", "fail", err.Error()) |
| 350 | } else { |
| 351 | if rows2.Next() { |
| 352 | var b string |
| 353 | if err := rows2.Scan(&b); err != nil { |
| 354 | r.report(suite, "batch_query_2", "fail", err.Error()) |
| 355 | } else if b == "hello" { |
| 356 | r.report(suite, "batch_query_2", "pass", "") |
| 357 | } else { |
| 358 | r.report(suite, "batch_query_2", "fail", fmt.Sprintf("expected hello, got %s", b)) |
| 359 | } |
| 360 | } |
| 361 | rows2.Close() |
| 362 | } |
| 363 | |
| 364 | // Result 3 |
| 365 | rows3, err := br.Query() |
| 366 | if err != nil { |
| 367 | r.report(suite, "batch_query_3", "fail", err.Error()) |