upgrade TestRetryWithCustomErrors

Author: asmyasnikov

internal/table/retry_test.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/errors"
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/rand"
+	"github.com/ydb-platform/ydb-go-sdk/v3/retry"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table/options"
 	"github.com/ydb-platform/ydb-go-sdk/v3/testutil"
@@ -376,6 +377,16 @@ func TestRetryWithCustomErrors(t *testing.T) {
 		retriable     bool
 		deleteSession bool
 	}{
+		{
+			error: &CustomError{
+				Err: retry.RetryableError(
+					fmt.Errorf("custom error"),
+					retry.WithDeleteSession(),
+				),
+			},
+			retriable:     true,
+			deleteSession: true,
+		},
 		{
 			error: &CustomError{
 				Err: errors.NewOpError(
@@ -422,22 +433,28 @@ func TestRetryWithCustomErrors(t *testing.T) {
 				sessions = make(map[table.Session]int)
 			)
 			err := do(ctx, p, func(ctx context.Context, s table.Session) (err error) {
-				if test.deleteSession {
-					if _, has := sessions[s]; has {
-						t.Fatalf("session already used: %s", s.ID())
-					}
-				}
 				sessions[s]++
 				i++
 				if i < limit {
 					return test.error
 				}
 				return nil
 			})
+			// nolint:nestif
 			if test.retriable {
 				if i != limit {
 					t.Fatalf("unexpected i: %d, err: %v", i, err)
 				}
+				if test.deleteSession {
+					if len(sessions) != limit {
+						t.Fatalf("unexpected len(sessions): %d, err: %v", len(sessions), err)
+					}
+					for s, n := range sessions {
+						if n != 1 {
+							t.Fatalf("unexpected session usage: %d, session: %v", n, s.ID())
+						}
+					}
+				}
 			} else {
 				if i != 1 {
 					t.Fatalf("unexpected i: %d, err: %v", i, err)

retry/retry_test.go

@@ -620,6 +620,15 @@ func TestRetryWithCustomErrors(t *testing.T) {
 		error     error
 		retriable bool
 	}{
+		{
+			error: &CustomError{
+				Err: RetryableError(
+					fmt.Errorf("custom error"),
+					WithDeleteSession(),
+				),
+			},
+			retriable: true,
+		},
 		{
 			error: &CustomError{
 				Err: errors.NewOpError(