@@ -7,7 +7,7 @@ package api
7
7
import (
8
8
"context"
9
9
"fmt"
10
- "io/ioutil "
10
+ "io"
11
11
"net/http"
12
12
"net/http/httptest"
13
13
"strings"
@@ -83,7 +83,7 @@ func TestQueryCVSResultSingleTable(t *testing.T) {
83
83
)
84
84
85
85
reader := strings .NewReader (csvTable )
86
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
86
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
87
87
require .True (t , queryResult .Next (), queryResult .Err ())
88
88
require .Nil (t , queryResult .Err ())
89
89
@@ -307,7 +307,7 @@ func TestQueryCVSResultMultiTables(t *testing.T) {
307
307
)
308
308
309
309
reader := strings .NewReader (csvTable )
310
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
310
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
311
311
assert .Equal (t , - 1 , queryResult .TablePosition ())
312
312
require .True (t , queryResult .Next (), queryResult .Err ())
313
313
require .Nil (t , queryResult .Err ())
@@ -439,7 +439,7 @@ func TestQueryCVSResultSingleTableMultiColumnsNoValue(t *testing.T) {
439
439
)
440
440
441
441
reader := strings .NewReader (csvTable )
442
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
442
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
443
443
require .True (t , queryResult .Next (), queryResult .Err ())
444
444
require .Nil (t , queryResult .Err ())
445
445
@@ -462,7 +462,8 @@ func TestQueryCVSResultSingleTableMultiColumnsNoValue(t *testing.T) {
462
462
}
463
463
464
464
func TestQueryRawResult (t * testing.T ) {
465
- csvRows := []string {`#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,dateTime:RFC3339,double,string,string,string,string` ,
465
+ csvRows := []string {
466
+ `#datatype,string,long,dateTime:RFC3339,dateTime:RFC3339,dateTime:RFC3339,double,string,string,string,string` ,
466
467
`#group,false,false,true,true,false,false,true,true,true,true` ,
467
468
`#default,_result,,,,,,,,,` ,
468
469
`,result,table,_start,_stop,_time,_value,_field,_measurement,a,b` ,
@@ -482,12 +483,12 @@ func TestQueryRawResult(t *testing.T) {
482
483
server := httptest .NewServer (http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
483
484
<- time .After (100 * time .Millisecond )
484
485
if r .Method == http .MethodPost {
485
- rbody , _ := ioutil .ReadAll (r .Body )
486
+ rbody , _ := io .ReadAll (r .Body )
486
487
fmt .Printf ("Req: %s\n " , string (rbody ))
487
488
body , err := gzip .CompressWithGzip (strings .NewReader (csvTable ))
488
489
if err == nil {
489
490
var bytes []byte
490
- bytes , err = ioutil .ReadAll (body )
491
+ bytes , err = io .ReadAll (body )
491
492
if err == nil {
492
493
w .Header ().Set ("Content-Type" , "text/csv" )
493
494
w .Header ().Set ("Content-Encoding" , "gzip" )
@@ -510,7 +511,6 @@ func TestQueryRawResult(t *testing.T) {
510
511
require .Nil (t , err )
511
512
require .NotNil (t , result )
512
513
assert .Equal (t , csvTable , result )
513
-
514
514
}
515
515
516
516
func TestErrorInRow (t * testing.T ) {
@@ -519,10 +519,11 @@ func TestErrorInRow(t *testing.T) {
519
519
`#group,true,true` ,
520
520
`#default,,` ,
521
521
`,error,reference` ,
522
- `,failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time,897` }
522
+ `,failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time,897` ,
523
+ }
523
524
csvTable := makeCSVstring (csvRowsError )
524
525
reader := strings .NewReader (csvTable )
525
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
526
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
526
527
527
528
require .False (t , queryResult .Next ())
528
529
require .NotNil (t , queryResult .Err ())
@@ -533,10 +534,11 @@ func TestErrorInRow(t *testing.T) {
533
534
`#group,true,true` ,
534
535
`#default,,` ,
535
536
`,error,reference` ,
536
- `,failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time,` }
537
+ `,failed to create physical plan: invalid time bounds from procedure from: bounds contain zero time,` ,
538
+ }
537
539
csvTable = makeCSVstring (csvRowsErrorNoReference )
538
540
reader = strings .NewReader (csvTable )
539
- queryResult = NewQueryTableResult (ioutil .NopCloser (reader ))
541
+ queryResult = NewQueryTableResult (io .NopCloser (reader ))
540
542
541
543
require .False (t , queryResult .Next ())
542
544
require .NotNil (t , queryResult .Err ())
@@ -547,10 +549,11 @@ func TestErrorInRow(t *testing.T) {
547
549
`#group,true,true` ,
548
550
`#default,,` ,
549
551
`,error,reference` ,
550
- `,,` }
552
+ `,,` ,
553
+ }
551
554
csvTable = makeCSVstring (csvRowsErrorNoMessage )
552
555
reader = strings .NewReader (csvTable )
553
- queryResult = NewQueryTableResult (ioutil .NopCloser (reader ))
556
+ queryResult = NewQueryTableResult (io .NopCloser (reader ))
554
557
555
558
require .False (t , queryResult .Next ())
556
559
require .NotNil (t , queryResult .Err ())
@@ -567,7 +570,7 @@ func TestInvalidDataType(t *testing.T) {
567
570
`
568
571
569
572
reader := strings .NewReader (csvTable )
570
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
573
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
571
574
require .False (t , queryResult .Next ())
572
575
require .NotNil (t , queryResult .Err ())
573
576
assert .Equal (t , "deviceId has unknown data type int" , queryResult .Err ().Error ())
@@ -627,7 +630,7 @@ func TestReorderedAnnotations(t *testing.T) {
627
630
628
631
`
629
632
reader := strings .NewReader (csvTable1 )
630
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
633
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
631
634
require .True (t , queryResult .Next (), queryResult .Err ())
632
635
require .Nil (t , queryResult .Err ())
633
636
@@ -654,7 +657,7 @@ func TestReorderedAnnotations(t *testing.T) {
654
657
655
658
`
656
659
reader = strings .NewReader (csvTable2 )
657
- queryResult = NewQueryTableResult (ioutil .NopCloser (reader ))
660
+ queryResult = NewQueryTableResult (io .NopCloser (reader ))
658
661
require .True (t , queryResult .Next (), queryResult .Err ())
659
662
require .Nil (t , queryResult .Err ())
660
663
@@ -725,7 +728,7 @@ func TestDatatypeOnlyAnnotation(t *testing.T) {
725
728
726
729
`
727
730
reader := strings .NewReader (csvTable1 )
728
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
731
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
729
732
require .True (t , queryResult .Next (), queryResult .Err ())
730
733
require .Nil (t , queryResult .Err ())
731
734
@@ -754,7 +757,7 @@ func TestMissingDatatypeAnnotation(t *testing.T) {
754
757
`
755
758
756
759
reader := strings .NewReader (csvTable1 )
757
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
760
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
758
761
require .False (t , queryResult .Next ())
759
762
require .NotNil (t , queryResult .Err ())
760
763
assert .Equal (t , "parsing error, datatype annotation not found" , queryResult .Err ().Error ())
@@ -768,7 +771,7 @@ func TestMissingDatatypeAnnotation(t *testing.T) {
768
771
`
769
772
770
773
reader = strings .NewReader (csvTable2 )
771
- queryResult = NewQueryTableResult (ioutil .NopCloser (reader ))
774
+ queryResult = NewQueryTableResult (io .NopCloser (reader ))
772
775
require .False (t , queryResult .Next ())
773
776
require .NotNil (t , queryResult .Err ())
774
777
assert .Equal (t , "parsing error, datatype annotation not found" , queryResult .Err ().Error ())
@@ -782,7 +785,7 @@ func TestMissingAnnotations(t *testing.T) {
782
785
783
786
`
784
787
reader := strings .NewReader (csvTable3 )
785
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
788
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
786
789
require .False (t , queryResult .Next ())
787
790
require .NotNil (t , queryResult .Err ())
788
791
assert .Equal (t , "parsing error, annotations not found" , queryResult .Err ().Error ())
@@ -797,7 +800,7 @@ func TestDifferentNumberOfColumns(t *testing.T) {
797
800
`
798
801
799
802
reader := strings .NewReader (csvTable )
800
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
803
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
801
804
require .False (t , queryResult .Next ())
802
805
require .NotNil (t , queryResult .Err ())
803
806
assert .Equal (t , "parsing error, row has different number of columns than the table: 11 vs 10" , queryResult .Err ().Error ())
@@ -810,7 +813,7 @@ func TestDifferentNumberOfColumns(t *testing.T) {
810
813
`
811
814
812
815
reader = strings .NewReader (csvTable2 )
813
- queryResult = NewQueryTableResult (ioutil .NopCloser (reader ))
816
+ queryResult = NewQueryTableResult (io .NopCloser (reader ))
814
817
require .False (t , queryResult .Next ())
815
818
require .NotNil (t , queryResult .Err ())
816
819
assert .Equal (t , "parsing error, row has different number of columns than the table: 8 vs 10" , queryResult .Err ().Error ())
@@ -823,7 +826,7 @@ func TestDifferentNumberOfColumns(t *testing.T) {
823
826
`
824
827
825
828
reader = strings .NewReader (csvTable3 )
826
- queryResult = NewQueryTableResult (ioutil .NopCloser (reader ))
829
+ queryResult = NewQueryTableResult (io .NopCloser (reader ))
827
830
require .False (t , queryResult .Next ())
828
831
require .NotNil (t , queryResult .Err ())
829
832
assert .Equal (t , "parsing error, row has different number of columns than the table: 10 vs 8" , queryResult .Err ().Error ())
@@ -840,7 +843,7 @@ func TestEmptyValue(t *testing.T) {
840
843
`
841
844
842
845
reader := strings .NewReader (csvTable )
843
- queryResult := NewQueryTableResult (ioutil .NopCloser (reader ))
846
+ queryResult := NewQueryTableResult (io .NopCloser (reader ))
844
847
845
848
require .True (t , queryResult .Next (), queryResult .Err ())
846
849
require .Nil (t , queryResult .Err ())
@@ -864,7 +867,7 @@ func TestFluxError(t *testing.T) {
864
867
server := httptest .NewServer (http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
865
868
<- time .After (100 * time .Millisecond )
866
869
if r .Method == http .MethodPost {
867
- _ , _ = ioutil .ReadAll (r .Body )
870
+ _ , _ = io .ReadAll (r .Body )
868
871
w .Header ().Set ("Content-Type" , "application/json" )
869
872
w .WriteHeader (http .StatusBadRequest )
870
873
_ , _ = w .Write ([]byte (`{"code":"invalid","message":"compilation failed: loc 4:17-4:86: expected an operator between two expressions"}` ))
@@ -882,12 +885,11 @@ func TestFluxError(t *testing.T) {
882
885
assert .Nil (t , tableRes )
883
886
require .NotNil (t , err )
884
887
assert .Equal (t , "invalid: compilation failed: loc 4:17-4:86: expected an operator between two expressions" , err .Error ())
885
-
886
888
}
887
889
888
890
func TestQueryParamsTypes (t * testing.T ) {
889
891
var i int8 = 1
890
- var paramsTypeTests = []struct {
892
+ paramsTypeTests : = []struct {
891
893
testName string
892
894
params interface {}
893
895
expectError string
@@ -1020,7 +1022,7 @@ func TestQueryParamsSerialized(t *testing.T) {
1020
1022
expectedBody := `{"dialect":{"annotations":["datatype","group","default"],"delimiter":",","header":true},"query":"from(bucket: \"environment\") |\u003e range(start: time(v: params.start)) |\u003e filter(fn: (r) =\u003e r._measurement == \"air\") |\u003e filter(fn: (r) =\u003e r._field == params.field) |\u003e filter(fn: (r) =\u003e r._value \u003e params.value)","type":"flux","params":{"start":"2022-02-17T11:27:23+01:00","field":"field","value":24.4}}`
1021
1023
server := httptest .NewServer (http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
1022
1024
if r .Method == http .MethodPost {
1023
- body , err := ioutil .ReadAll (r .Body )
1025
+ body , err := io .ReadAll (r .Body )
1024
1026
if err != nil {
1025
1027
w .Header ().Set ("Content-Type" , "application/json" )
1026
1028
w .WriteHeader (http .StatusBadRequest )
@@ -1058,7 +1060,6 @@ func TestQueryParamsSerialized(t *testing.T) {
1058
1060
1059
1061
_ , err = queryAPI .QueryWithParams (context .Background (), query , condition )
1060
1062
require .NoError (t , err , err )
1061
-
1062
1063
}
1063
1064
1064
1065
func makeCSVstring (rows []string ) string {
0 commit comments