2
0

Add driver.RowsColumnTypeDatabaseTypeName support to stdlib.Rows

This commit is contained in:
Jack Christensen
2017-05-19 16:05:30 -05:00
parent 5ee76a26c8
commit f9cb22e4b8
2 changed files with 32 additions and 0 deletions
+27
View File
@@ -1073,3 +1073,30 @@ func TestConnQueryContextCancel(t *testing.T) {
t.Errorf("mock server err: %v", err)
}
}
func TestRowsColumnTypeDatabaseTypeName(t *testing.T) {
db := openDB(t)
defer closeDB(t, db)
rows, err := db.Query("select * from generate_series(1,10) n")
if err != nil {
t.Fatalf("db.Query failed: %v", err)
}
columnTypes, err := rows.ColumnTypes()
if err != nil {
t.Fatalf("rows.ColumnTypes failed: %v", err)
}
if len(columnTypes) != 1 {
t.Fatalf("len(columnTypes) => %v, want %v", len(columnTypes), 1)
}
if columnTypes[0].DatabaseTypeName() != "INT4" {
t.Errorf("columnTypes[0].DatabaseTypeName() => %v, want %v", columnTypes[0].DatabaseTypeName(), "INT4")
}
rows.Close()
ensureConnValid(t, db)
}