@@ -21,54 +21,55 @@ func TestRegister_mode(t *testing.T) {
21
21
if err != nil {
22
22
t .Fatal (err )
23
23
}
24
- if stmt .Step () {
25
- if got := stmt .ColumnInt ( 0 ); got != 3 {
26
- t . Errorf ( " got %v, want 3" , got )
27
- }
24
+ if ! stmt .Step () {
25
+ t . Fatal ( stmt .Err ())
26
+ } else if got := stmt . ColumnInt ( 0 ); got != 3 {
27
+ t . Errorf ( "got %v, want 3" , got )
28
28
}
29
29
stmt .Close ()
30
30
31
31
stmt , _ , err = db .Prepare (`SELECT mode(column1) FROM (VALUES (1), (1), (2), (2), (3))` )
32
32
if err != nil {
33
33
t .Fatal (err )
34
34
}
35
- if stmt .Step () {
36
- if got := stmt .ColumnInt ( 0 ); got != 1 {
37
- t . Errorf ( " got %v, want 1" , got )
38
- }
35
+ if ! stmt .Step () {
36
+ t . Fatal ( stmt .Err ())
37
+ } else if got := stmt . ColumnInt ( 0 ); got != 1 {
38
+ t . Errorf ( "got %v, want 1" , got )
39
39
}
40
40
stmt .Close ()
41
41
42
42
stmt , _ , err = db .Prepare (`SELECT mode(column1) FROM (VALUES (0.5), (1), (2.5), (2), (2.5))` )
43
43
if err != nil {
44
44
t .Fatal (err )
45
45
}
46
- if stmt .Step () {
47
- if got := stmt .ColumnFloat ( 0 ); got != 2.5 {
48
- t . Errorf ( " got %v, want 2.5" , got )
49
- }
46
+ if ! stmt .Step () {
47
+ t . Fatal ( stmt .Err ())
48
+ } else if got := stmt . ColumnFloat ( 0 ); got != 2.5 {
49
+ t . Errorf ( "got %v, want 2.5" , got )
50
50
}
51
51
stmt .Close ()
52
52
53
53
stmt , _ , err = db .Prepare (`SELECT mode(column1) FROM (VALUES ('red'), ('green'), ('blue'), ('red'))` )
54
54
if err != nil {
55
55
t .Fatal (err )
56
56
}
57
- if stmt .Step () {
58
- if got := stmt .ColumnText ( 0 ); got != "red" {
59
- t . Errorf ( " got %q, want red", got )
60
- }
57
+ if ! stmt .Step () {
58
+ t . Fatal ( stmt .Err ())
59
+ } else if got := stmt . ColumnText ( 0 ); got != " red" {
60
+ t . Errorf ( "got %q, want red" , got )
61
61
}
62
+
62
63
stmt .Close ()
63
64
64
65
stmt , _ , err = db .Prepare (`SELECT mode(column1) FROM (VALUES (X'cafebabe'), ('green'), ('blue'), (X'cafebabe'))` )
65
66
if err != nil {
66
67
t .Fatal (err )
67
68
}
68
- if stmt .Step () {
69
- if got := stmt .ColumnText ( 0 ); got != " \xca \xfe \xba \xbe " {
70
- t . Errorf ( " got %q, want cafebabe" , got )
71
- }
69
+ if ! stmt .Step () {
70
+ t . Fatal ( stmt .Err ())
71
+ } else if got := stmt . ColumnText ( 0 ); got != " \xca \xfe \xba \xbe " {
72
+ t . Errorf ( "got %q, want cafebabe" , got )
72
73
}
73
74
stmt .Close ()
74
75
@@ -92,10 +93,10 @@ func TestRegister_mode(t *testing.T) {
92
93
stmt .BindInt (3 , 2 )
93
94
stmt .BindFloat (4 , 2 )
94
95
stmt .BindFloat (5 , 2 )
95
- if stmt .Step () {
96
- if got := stmt .ColumnInt ( 0 ); got != 2 {
97
- t . Errorf ( " got %v, want 2" , got )
98
- }
96
+ if ! stmt .Step () {
97
+ t . Fatal ( stmt .Err ())
98
+ } else if got := stmt . ColumnInt ( 0 ); got != 2 {
99
+ t . Errorf ( "got %v, want 2" , got )
99
100
}
100
101
stmt .Close ()
101
102
}
0 commit comments