@@ -14,7 +14,6 @@ import (
1414 "github.com/stretchr/testify/require"
1515
1616 "github.com/ydb-platform/ydb-go-sdk/v3"
17- "github.com/ydb-platform/ydb-go-sdk/v3/internal/version"
1817 "github.com/ydb-platform/ydb-go-sdk/v3/internal/xtest"
1918 "github.com/ydb-platform/ydb-go-sdk/v3/query"
2019 "github.com/ydb-platform/ydb-go-sdk/v3/retry"
@@ -91,18 +90,16 @@ func TestRetryBudget(t *testing.T) {
9190 }, table .WithRetryBudget (retryBudget ))
9291 require .ErrorIs (t , err , budget .ErrNoQuota )
9392 })
94- if version .Gte (os .Getenv ("YDB_VERSION" ), "24.1" ) {
95- t .Run ("db.Query().Do" , func (t * testing.T ) {
96- err := nativeDriver .Query ().Do (ctx , func (ctx context.Context , s query.Session ) error {
97- return retry .RetryableError (errors .New ("custom error" ))
98- }, query .WithRetryBudget (retryBudget ))
99- require .ErrorIs (t , err , budget .ErrNoQuota )
100- })
101- t .Run ("db.Query().DoTx" , func (t * testing.T ) {
102- err := nativeDriver .Query ().DoTx (ctx , func (ctx context.Context , tx query.TxActor ) error {
103- return retry .RetryableError (errors .New ("custom error" ))
104- }, query .WithRetryBudget (retryBudget ))
105- require .ErrorIs (t , err , budget .ErrNoQuota )
106- })
107- }
93+ t .Run ("db.Query().Do" , func (t * testing.T ) {
94+ err := nativeDriver .Query ().Do (ctx , func (ctx context.Context , s query.Session ) error {
95+ return retry .RetryableError (errors .New ("custom error" ))
96+ }, query .WithRetryBudget (retryBudget ))
97+ require .ErrorIs (t , err , budget .ErrNoQuota )
98+ })
99+ t .Run ("db.Query().DoTx" , func (t * testing.T ) {
100+ err := nativeDriver .Query ().DoTx (ctx , func (ctx context.Context , tx query.TxActor ) error {
101+ return retry .RetryableError (errors .New ("custom error" ))
102+ }, query .WithRetryBudget (retryBudget ))
103+ require .ErrorIs (t , err , budget .ErrNoQuota )
104+ })
108105}
0 commit comments