Function TestQueryMultipleResults

driver_test.go:2645–2679

(t *testing.T)

}

func TestQueryMultipleResults(t *testing.T) {
	ctx := context.Background()
	runTestsWithMultiStatement(t, dsn, func(dbt *DBTest) {
		dbt.mustExec(`
		CREATE TABLE test (
			id INT NOT NULL AUTO_INCREMENT,
			value VARCHAR(255),
			PRIMARY KEY (id)
		)`)
		conn, err := dbt.db.Conn(ctx)
		if err != nil {
			t.Fatalf("failed to connect: %v", err)
		}
		conn.Raw(func(conn any) error {
			//lint:ignore SA1019 this is a test
			qr := conn.(driver.Queryer)
			c := conn.(*mysqlConn)

			// Demonstrate that repeated queries reset the affectedRows
			for range 2 {
				_, err := qr.Query(`
				INSERT INTO test (value) VALUES ('a'), ('b');
				INSERT INTO test (value) VALUES ('c'), ('d'), ('e');
			`, nil)
				if err != nil {
					t.Fatalf("insert statements failed: %v", err)
				}
				if got, want := c.result.affectedRows, []int64{2, 3}; !reflect.DeepEqual(got, want) {
					t.Errorf("bad affectedRows: got %v, want=%v", got, want)
				}
			}
			return nil
		})
	})
}

func TestPingContext(t *testing.T) {
	runTestsParallel(t, dsn, func(dbt *DBTest, _ string) {