@@ -150,98 +150,6 @@ func (suite *QueryTestSuite) TestBasicQuery() {
150
150
suite .executeQuery (stmt , nil , expNumRows , 0 , true , expValues )
151
151
}
152
152
153
- func (suite * QueryTestSuite ) TestQueryWithSmallLimit () {
154
- numMajor , numPerMajor , recordKB := 1 , 5 , 2
155
- minRead := suite .minRead ()
156
- prepCost := suite .minQueryCost ()
157
-
158
- suite .loadRowsToTable (numMajor , numPerMajor , recordKB , suite .table )
159
-
160
- // Update with number based limit of 1.
161
- newRecordKB := 1
162
- longStr := test .GenString ((newRecordKB - 1 ) * 1024 )
163
- stmt := fmt .Sprintf (`update %s set longString = "%s" where sid = 0 and id = 0` ,
164
- suite .table , longStr )
165
- req := & nosqldb.QueryRequest {
166
- Statement : stmt ,
167
- Limit : 1 ,
168
- }
169
- res , err := suite .Client .Query (req )
170
-
171
- expectReadKB := minRead + recordKB
172
- expectWriteKB := recordKB + newRecordKB
173
-
174
- if suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err ) {
175
- if test .IsCloud () {
176
- suite .AssertReadWriteKB (res , expectReadKB , expectWriteKB , prepCost , true )
177
- } else {
178
- suite .AssertZeroReadWriteKB (res )
179
- }
180
- }
181
-
182
- if suite .IsCloud () || (suite .IsOnPrem () && suite .Version > "20.3" ) {
183
- // Update with maxReadKB of 1, expect an IllegalArgument error.
184
- stmt = fmt .Sprintf (`update %s set longString = "%s" where sid = 0 and id = 1` ,
185
- suite .table , longStr )
186
- for kb := 1 ; kb <= expectReadKB ; kb ++ {
187
- req = & nosqldb.QueryRequest {
188
- Statement : stmt ,
189
- MaxReadKB : uint (kb ),
190
- }
191
- res , err = suite .Client .Query (req )
192
- if kb < expectReadKB {
193
- suite .Truef (nosqlerr .IsIllegalArgument (err ),
194
- "Update with MaxReadKB=%d (expect ReadKB=%d) should have failed with an " +
195
- "IllegalArgument error, got error %v" , kb , expectReadKB , err )
196
-
197
- } else if suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err ) {
198
- if test .IsCloud () {
199
- suite .AssertReadWriteKB (res , expectReadKB , expectWriteKB , prepCost , true )
200
- } else {
201
- suite .AssertZeroReadWriteKB (res )
202
- }
203
- }
204
- }
205
- }
206
-
207
- // Update 0 rows with maxReadKB of 1.
208
- stmt = fmt .Sprintf ("update %s set longString = \" %s\" where sid = 10000 and id = 1" ,
209
- suite .table , longStr )
210
- req = & nosqldb.QueryRequest {
211
- Statement : stmt ,
212
- MaxReadKB : 1 ,
213
- }
214
- res , err = suite .Client .Query (req )
215
- if suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err ) {
216
- if test .IsCloud () {
217
- suite .AssertReadWriteKB (res , minRead , 0 , prepCost , true )
218
- } else {
219
- suite .AssertZeroReadWriteKB (res )
220
- }
221
- }
222
-
223
- // Query with number limit of 1
224
- stmt = fmt .Sprintf ("select * from %s where sid = 0 and id > 1" , suite .table )
225
- numRows := numMajor * (numPerMajor - 2 )
226
- expectReadKB = suite .getExpectReadKB (false , recordKB , numRows , numRows )
227
- suite .executeQueryTest (stmt ,
228
- false , // keyOnly
229
- false , // indexScan
230
- numRows , // expected number of rows returned
231
- expectReadKB , // expected readKB
232
- 1 , // limit
233
- 0 , // maxReadKB
234
- recordKB )
235
-
236
- // Query with MaxReadKB of 1, expect an IllegalArgument error
237
- req = & nosqldb.QueryRequest {
238
- Statement : stmt ,
239
- MaxReadKB : 1 ,
240
- }
241
- _ , err = suite .ExecuteQueryRequest (req )
242
- suite .Truef (nosqlerr .IsIllegalArgument (err ), "Query(MaxReadKB=1, stmt=%s) " +
243
- "should have failed with an IllegalArgument error, got error %v" , err )
244
- }
245
153
246
154
func (suite * QueryTestSuite ) TestIllegalQuery () {
247
155
@@ -838,8 +746,6 @@ func (suite *QueryTestSuite) TestGroupByWithLimits() {
838
746
func (suite * QueryTestSuite ) TestUpdateQuery () {
839
747
numMajor , numPerMajor := 1 , 10
840
748
recordKB := 2
841
- minRead := suite .minRead ()
842
- prepCost := suite .minQueryCost ()
843
749
844
750
suite .loadRowsToTable (numMajor , numPerMajor , recordKB , suite .table )
845
751
@@ -849,26 +755,14 @@ func (suite *QueryTestSuite) TestUpdateQuery() {
849
755
stmt := fmt .Sprintf (`update %s set longString = "%s" where sid = 0 and id = 0` ,
850
756
suite .table , longStr )
851
757
req := & nosqldb.QueryRequest {Statement : stmt }
852
- res , err := suite .Client .Query (req )
853
- if suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err ) {
854
- if test .IsCloud () {
855
- suite .AssertReadWriteKB (res , minRead + recordKB , recordKB + newRecordKB , prepCost , true )
856
- } else {
857
- suite .AssertZeroReadWriteKB (res )
858
- }
859
- }
758
+ _ , err := suite .Client .Query (req )
759
+ suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err )
860
760
861
761
// Update non-exsiting row.
862
762
stmt = fmt .Sprintf (`update %s set longString = "test" where sid = 1000 and id = 0` , suite .table )
863
763
req = & nosqldb.QueryRequest {Statement : stmt }
864
- res , err = suite .Client .Query (req )
865
- if suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err ) {
866
- if test .IsCloud () {
867
- suite .AssertReadWriteKB (res , minRead , 0 , prepCost , true )
868
- } else {
869
- suite .AssertZeroReadWriteKB (res )
870
- }
871
- }
764
+ _ , err = suite .Client .Query (req )
765
+ suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err )
872
766
873
767
// Update using prepared statement
874
768
stmt = fmt .Sprintf ("declare $sval string; $sid integer; $id integer;" +
@@ -879,25 +773,13 @@ func (suite *QueryTestSuite) TestUpdateQuery() {
879
773
//TODO
880
774
// suite.NotNil(prepRes.PreparedStatement.Statement)
881
775
882
- if test .IsCloud () {
883
- suite .AssertReadWriteKB (prepRes , prepCost , 0 , 0 , false )
884
- } else {
885
- suite .AssertZeroReadWriteKB (prepRes )
886
- }
887
-
888
776
prepRes .PreparedStatement .SetVariable ("$sval" , longStr )
889
777
prepRes .PreparedStatement .SetVariable ("$sid" , 0 )
890
778
prepRes .PreparedStatement .SetVariable ("$id" , 1 )
891
779
892
780
req = & nosqldb.QueryRequest {PreparedStatement : & prepRes .PreparedStatement }
893
- res , err = suite .Client .Query (req )
894
- if suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err ) {
895
- if test .IsCloud () {
896
- suite .AssertReadWriteKB (res , minRead + recordKB , recordKB + newRecordKB , 0 , true )
897
- } else {
898
- suite .AssertZeroReadWriteKB (res )
899
- }
900
- }
781
+ _ , err = suite .Client .Query (req )
782
+ suite .NoErrorf (err , "failed to execute %q, got error: %v" , stmt , err )
901
783
}
902
784
903
785
}
0 commit comments