@@ -1063,36 +1063,44 @@ func TestQueryer(t *testing.T) {
1063
1063
defer db .Close ()
1064
1064
1065
1065
_ , err = db .Exec (`
1066
- create table foo (id integer);
1066
+ create table foo (id integer);
1067
1067
` )
1068
1068
if err != nil {
1069
1069
t .Error ("Failed to call db.Query:" , err )
1070
1070
}
1071
1071
1072
- rows , err := db .Query (`
1073
- insert into foo(id) values(?);
1074
- insert into foo(id) values(?);
1075
- insert into foo(id) values(?);
1076
- select id from foo order by id;
1072
+ _ , err = db .Exec (`
1073
+ insert into foo(id) values(?);
1074
+ insert into foo(id) values(?);
1075
+ insert into foo(id) values(?);
1077
1076
` , 3 , 2 , 1 )
1077
+ if err != nil {
1078
+ t .Error ("Failed to call db.Exec:" , err )
1079
+ }
1080
+ rows , err := db .Query (`
1081
+ select id from foo order by id;
1082
+ ` )
1078
1083
if err != nil {
1079
1084
t .Error ("Failed to call db.Query:" , err )
1080
1085
}
1081
1086
defer rows .Close ()
1082
- n := 1
1087
+ n := 0
1083
1088
for rows .Next () {
1084
1089
var id int
1085
1090
err = rows .Scan (& id )
1086
1091
if err != nil {
1087
1092
t .Error ("Failed to db.Query:" , err )
1088
1093
}
1089
- if id != n {
1094
+ if id != n + 1 {
1090
1095
t .Error ("Failed to db.Query: not matched results" )
1091
1096
}
1092
1097
n = n + 1
1093
1098
}
1094
- if n != 4 {
1095
- t .Errorf ("Expected 3 rows but retrieved %v" , n - 1 )
1099
+ if err := rows .Err (); err != nil {
1100
+ t .Errorf ("Post-scan failed: %v\n " , err )
1101
+ }
1102
+ if n != 3 {
1103
+ t .Errorf ("Expected 3 rows but retrieved %v" , n )
1096
1104
}
1097
1105
}
1098
1106
0 commit comments