@@ -150,98 +150,6 @@ func (suite *QueryTestSuite) TestBasicQuery() {
150150 suite .executeQuery (stmt , nil , expNumRows , 0 , true , expValues )
151151}
152152
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- }
245153
246154func (suite * QueryTestSuite ) TestIllegalQuery () {
247155
@@ -838,8 +746,6 @@ func (suite *QueryTestSuite) TestGroupByWithLimits() {
838746func (suite * QueryTestSuite ) TestUpdateQuery () {
839747 numMajor , numPerMajor := 1 , 10
840748 recordKB := 2
841- minRead := suite .minRead ()
842- prepCost := suite .minQueryCost ()
843749
844750 suite .loadRowsToTable (numMajor , numPerMajor , recordKB , suite .table )
845751
@@ -849,26 +755,14 @@ func (suite *QueryTestSuite) TestUpdateQuery() {
849755 stmt := fmt .Sprintf (`update %s set longString = "%s" where sid = 0 and id = 0` ,
850756 suite .table , longStr )
851757 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 )
860760
861761 // Update non-exsiting row.
862762 stmt = fmt .Sprintf (`update %s set longString = "test" where sid = 1000 and id = 0` , suite .table )
863763 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 )
872766
873767 // Update using prepared statement
874768 stmt = fmt .Sprintf ("declare $sval string; $sid integer; $id integer;" +
@@ -879,25 +773,13 @@ func (suite *QueryTestSuite) TestUpdateQuery() {
879773 //TODO
880774 // suite.NotNil(prepRes.PreparedStatement.Statement)
881775
882- if test .IsCloud () {
883- suite .AssertReadWriteKB (prepRes , prepCost , 0 , 0 , false )
884- } else {
885- suite .AssertZeroReadWriteKB (prepRes )
886- }
887-
888776 prepRes .PreparedStatement .SetVariable ("$sval" , longStr )
889777 prepRes .PreparedStatement .SetVariable ("$sid" , 0 )
890778 prepRes .PreparedStatement .SetVariable ("$id" , 1 )
891779
892780 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 )
901783 }
902784
903785}
0 commit comments