@@ -27,7 +27,7 @@ const (
2727)
2828
2929func TestWithNil (t * testing.T ) {
30- r := csvdata .NewRows ((* csvdata .Reader )(nil ).WithComma (',' ).WithFieldsPerRecord (1 ), true )
30+ r := csvdata .NewRows ((* csvdata .Reader )(nil ).WithComma (',' ).WithLazyQuotes ( true ). WithTrimLeadingSpace ( true ). WithFieldsPerRecord (1 ), true )
3131 defer r .Close () //dummy
3232 if err := r .Next (); ! errors .Is (err , csvdata .ErrNullPointer ) {
3333 t .Errorf ("Next() is \" %+v\" , want \" %+v\" ." , err , csvdata .ErrNullPointer )
@@ -65,7 +65,7 @@ func TestWithNil(t *testing.T) {
6565
6666func TestErrReader (t * testing.T ) {
6767 errtest := errors .New ("test" )
68- r := csvdata .NewRows (csvdata .New (iotest .ErrReader (errtest )).WithComma (',' ).WithFieldsPerRecord (1 ), true )
68+ r := csvdata .NewRows (csvdata .New (iotest .ErrReader (errtest )).WithComma (',' ).WithLazyQuotes ( true ). WithTrimLeadingSpace ( true ). WithFieldsPerRecord (1 ), true )
6969 defer r .Close () //dummy
7070 if err := r .Next (); ! errors .Is (err , errtest ) {
7171 t .Errorf ("Next() is \" %+v\" , want \" %+v\" ." , err , errtest )
@@ -95,7 +95,7 @@ func TestNormal(t *testing.T) {
9595 flagBool sql.NullBool
9696 mass float64
9797 massFloat64 sql.NullFloat64
98- order int64
98+ order int8
9999 orderByte sql.NullByte
100100 orderInt16 sql.NullInt16
101101 orderInt32 sql.NullInt32
@@ -107,7 +107,7 @@ func TestNormal(t *testing.T) {
107107 }
108108
109109 for _ , tc := range testCases {
110- rc := csvdata .NewRows (csvdata .New (tc .inp ).WithComma (tc .sep ).WithFieldsPerRecord (tc .size ), tc .headerFlag )
110+ rc := csvdata .NewRows (csvdata .New (tc .inp ).WithComma (tc .sep ).WithLazyQuotes ( true ). WithTrimLeadingSpace ( true ). WithFieldsPerRecord (tc .size ), tc .headerFlag )
111111 if err := rc .Next (); err != nil {
112112 t .Errorf ("Next() is \" %+v\" , want nil." , err )
113113 } else {
@@ -140,7 +140,7 @@ func TestNormal(t *testing.T) {
140140 t .Errorf ("Column() is \" %v\" , want \" %v\" ." , name , tc .name2 )
141141 }
142142 //bool
143- if _ , err = rc .GetBool (5 ); ! errors .Is (err , csvdata .ErrNullPointer ) {
143+ if _ , err = rc .GetBool (5 ); ! errors .Is (err , csvdata .ErrNullValue ) {
144144 t .Errorf ("GetBool() is \" %+v\" , want \" %+v\" ." , err , strconv .ErrSyntax )
145145 }
146146 if _ , err = rc .ColumnBool ("name" ); ! errors .Is (err , strconv .ErrSyntax ) && ! errors .Is (err , tc .err ) {
@@ -161,7 +161,7 @@ func TestNormal(t *testing.T) {
161161 t .Errorf ("ColumnBool() is \" %+v\" , want \" %+v\" ." , flagBool , tc .flagBool )
162162 }
163163 //float
164- if _ , err = rc .GetFloat64 (5 ); ! errors .Is (err , csvdata .ErrNullPointer ) {
164+ if _ , err = rc .GetFloat64 (5 ); ! errors .Is (err , csvdata .ErrNullValue ) {
165165 t .Errorf ("GetFloat() is \" %+v\" , want \" %+v\" ." , err , strconv .ErrSyntax )
166166 }
167167 if _ , err = rc .ColumnFloat64 ("name" ); ! errors .Is (err , strconv .ErrSyntax ) && ! errors .Is (err , tc .err ) {
@@ -176,52 +176,80 @@ func TestNormal(t *testing.T) {
176176 }
177177 massFloat64 , err := rc .ColumnNullFloat64 ("mass" )
178178 if ! errors .Is (err , tc .err ) {
179- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , err , tc .err )
179+ t .Errorf ("ColumnNullFloat64 () is \" %+v\" , want \" %+v\" ." , err , tc .err )
180180 }
181181 if err == nil && massFloat64 != tc .massFloat64 {
182- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , massFloat64 , tc .massFloat64 )
182+ t .Errorf ("ColumnNullFloat64 () is \" %+v\" , want \" %+v\" ." , massFloat64 , tc .massFloat64 )
183183 }
184184 //int
185- if _ , err = rc .GetInt64 (5 , 10 ); ! errors .Is (err , csvdata .ErrNullPointer ) {
186- t .Errorf ("GetFloat () is \" %+v\" , want \" %+v\" ." , err , strconv .ErrSyntax )
185+ if _ , err = rc .GetInt64 (5 , 10 ); ! errors .Is (err , csvdata .ErrNullValue ) {
186+ t .Errorf ("GetInt64 () is \" %+v\" , want \" %+v\" ." , err , strconv .ErrSyntax )
187187 }
188188 if _ , err = rc .ColumnInt64 ("name" , 10 ); ! errors .Is (err , strconv .ErrSyntax ) && ! errors .Is (err , tc .err ) {
189- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , err , strconv .ErrSyntax )
189+ t .Errorf ("ColumnInt64 () is \" %+v\" , want \" %+v\" ." , err , strconv .ErrSyntax )
190190 }
191- order , err := rc .ColumnInt64 ("order" , 10 )
191+ order64 , err := rc .ColumnInt64 ("order" , 10 )
192192 if ! errors .Is (err , tc .err ) {
193- t .Errorf ("ColumnFloat() is \" %+v\" , want \" %+v\" ." , err , tc .err )
193+ t .Errorf ("ColumnInt64() is \" %+v\" , want \" %+v\" ." , err , tc .err )
194+ }
195+ if err == nil && order64 != int64 (tc .order ) {
196+ t .Errorf ("ColumnInt64() is \" %+v\" , want \" %+v\" ." , order64 , tc .order )
197+ }
198+ order32 , err := rc .ColumnInt32 ("order" , 10 )
199+ if ! errors .Is (err , tc .err ) {
200+ t .Errorf ("ColumnInt32() is \" %+v\" , want \" %+v\" ." , err , tc .err )
201+ }
202+ if err == nil && order32 != int32 (tc .order ) {
203+ t .Errorf ("ColumnInt32() is \" %+v\" , want \" %+v\" ." , order32 , tc .order )
204+ }
205+ order16 , err := rc .ColumnInt16 ("order" , 10 )
206+ if ! errors .Is (err , tc .err ) {
207+ t .Errorf ("ColumnInt16() is \" %+v\" , want \" %+v\" ." , err , tc .err )
194208 }
195- if err == nil && order != tc .order {
196- t .Errorf ("ColumnFloat() is \" %+v\" , want \" %+v\" ." , order , tc .order )
209+ if err == nil && order16 != int16 (tc .order ) {
210+ t .Errorf ("ColumnInt16() is \" %+v\" , want \" %+v\" ." , order16 , tc .order )
211+ }
212+ order8 , err := rc .ColumnInt8 ("order" , 10 )
213+ if ! errors .Is (err , tc .err ) {
214+ t .Errorf ("ColumnInt8() is \" %+v\" , want \" %+v\" ." , err , tc .err )
215+ }
216+ if err == nil && order8 != tc .order {
217+ t .Errorf ("ColumnInt8() is \" %+v\" , want \" %+v\" ." , order8 , tc .order )
218+ }
219+ orderB , err := rc .ColumnByte ("order" , 10 )
220+ if ! errors .Is (err , tc .err ) {
221+ t .Errorf ("ColumnByte() is \" %+v\" , want \" %+v\" ." , err , tc .err )
222+ }
223+ if err == nil && orderB != byte (tc .order ) {
224+ t .Errorf ("ColumnByte() is \" %+v\" , want \" %+v\" ." , orderB , tc .order )
197225 }
198226 orderByte , err := rc .ColumnNullByte ("order" , 10 )
199227 if ! errors .Is (err , tc .err ) {
200- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , err , tc .err )
228+ t .Errorf ("ColumnNullByte () is \" %+v\" , want \" %+v\" ." , err , tc .err )
201229 }
202230 if err == nil && orderByte != tc .orderByte {
203- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , orderByte , tc .orderByte )
231+ t .Errorf ("ColumnNullByte () is \" %+v\" , want \" %+v\" ." , orderByte , tc .orderByte )
204232 }
205233 orderInt16 , err := rc .ColumnNullInt16 ("order" , 10 )
206234 if ! errors .Is (err , tc .err ) {
207- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , err , tc .err )
235+ t .Errorf ("ColumnNullInt16 () is \" %+v\" , want \" %+v\" ." , err , tc .err )
208236 }
209237 if err == nil && orderInt16 != tc .orderInt16 {
210- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , orderInt16 , tc .orderInt16 )
238+ t .Errorf ("ColumnNullInt16 () is \" %+v\" , want \" %+v\" ." , orderInt16 , tc .orderInt16 )
211239 }
212240 orderInt32 , err := rc .ColumnNullInt32 ("order" , 10 )
213241 if ! errors .Is (err , tc .err ) {
214- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , err , tc .err )
242+ t .Errorf ("ColumnNullInt32 () is \" %+v\" , want \" %+v\" ." , err , tc .err )
215243 }
216244 if err == nil && orderInt32 != tc .orderInt32 {
217- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , orderInt32 , tc .orderInt32 )
245+ t .Errorf ("ColumnNullInt32 () is \" %+v\" , want \" %+v\" ." , orderInt32 , tc .orderInt32 )
218246 }
219247 orderInt64 , err := rc .ColumnNullInt64 ("order" , 10 )
220248 if ! errors .Is (err , tc .err ) {
221- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , err , tc .err )
249+ t .Errorf ("ColumnNullInt64 () is \" %+v\" , want \" %+v\" ." , err , tc .err )
222250 }
223251 if err == nil && orderInt64 != tc .orderInt64 {
224- t .Errorf ("ColumnFloat () is \" %+v\" , want \" %+v\" ." , orderInt64 , tc .orderInt64 )
252+ t .Errorf ("ColumnNullInt64 () is \" %+v\" , want \" %+v\" ." , orderInt64 , tc .orderInt64 )
225253 }
226254 }
227255 }
0 commit comments