Skip to content

Commit 95b13df

Browse files
committed
Generate random no-ack query in chain test
1 parent 2f894e5 commit 95b13df

2 files changed

Lines changed: 11 additions & 4 deletions

File tree

sqlchain/chain_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ func TestMultiChain(t *testing.T) {
360360
// Send a random query
361361
err = cli.query(types.ReadQuery, []types.Query{
362362
buildQuery(`SELECT v FROM t1 WHERE k=?`, rand.Intn(5)),
363-
})
363+
}, rand.Int()%10 != 0)
364364
if err != nil {
365365
t.Errorf("error occurred: %v", err)
366366
}

sqlchain/xxx_test.go

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -416,8 +416,11 @@ func (p *nodeProfile) buildQuery(
416416
}
417417
return
418418
}
419-
420419
func (p *nodeProfile) sendQuery(req *types.Request) (err error) {
420+
return p.sendQueryEx(req, true)
421+
}
422+
423+
func (p *nodeProfile) sendQueryEx(req *types.Request, genAck bool) (err error) {
421424
tracker, resp, err := p.Chain.Query(req, p.IsLeader)
422425
if err != nil {
423426
return
@@ -430,6 +433,10 @@ func (p *nodeProfile) sendQuery(req *types.Request) (err error) {
430433
}
431434
tracker.UpdateResp(resp)
432435

436+
if !genAck {
437+
return
438+
}
439+
433440
ack, err := createRandomQueryAckWithResponse(&resp.Header, p)
434441
if err != nil {
435442
return
@@ -441,11 +448,11 @@ func (p *nodeProfile) sendQuery(req *types.Request) (err error) {
441448
}
442449

443450
func (p *nodeProfile) query(
444-
qt types.QueryType, qs []types.Query) (err error,
451+
qt types.QueryType, qs []types.Query, genAck bool) (err error,
445452
) {
446453
req, err := p.buildQuery(qt, qs)
447454
if err != nil {
448455
return
449456
}
450-
return p.sendQuery(req)
457+
return p.sendQueryEx(req, genAck)
451458
}

0 commit comments

Comments
 (0)