chore: fix test failure

zeromicro · Apr 17, 2024 · bd8cc06 · bd8cc06
1 parent 25f92c4
commit bd8cc06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/stores/sqlx/stmt_test.go b/core/stores/sqlx/stmt_test.go
@@ -294,15 +294,15 @@ func TestStmtBreaker(t *testing.T) {
 func TestQueryRowsScanTimeout(t *testing.T) {
 	dbtest.RunTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
 		rows := sqlmock.NewRows([]string{"foo"})
-		for i := 0; i < 100000; i++ {
+		for i := 0; i < 10000; i++ {
 			rows = rows.AddRow("bar" + strconv.Itoa(i))
 		}
 		mock.ExpectQuery("any").WillReturnRows(rows)
 		var val []struct {
 			Foo string
 		}
 		conn := NewSqlConnFromDB(db)
-		ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*5)
+		ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*2)
 		err := conn.QueryRowsCtx(ctx, &val, "any")
 		assert.ErrorIs(t, err, context.DeadlineExceeded)
 		cancel()