driver_test.go - master - githubmirror/ydb-go-sdk

driver_test.go

52 строки1 KB

Aleksey Myasnikov

refactor(conn): Get/Put ref-counting with balancer quarantine (#2235)

15 июл 2026, 17:24

15 июл 2026, 17:246dfb4a0

package ydb_test

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/ydb-platform/ydb-go-sdk/v3"
	"github.com/ydb-platform/ydb-go-sdk/v3/trace"
)

func TestOpen(t *testing.T) {
	t.Run("context already canceled", func(t *testing.T) {
		ctx, cancel := context.WithCancel(context.Background())
		cancel()

		_, err := ydb.Open(ctx, "grpc://localhost:2136/local")
		require.ErrorIs(t, err, context.Canceled)
		assert.Regexp(t, "^context canceled at", err.Error())
	})

	t.Run("context canceled at internal connect()", func(t *testing.T) {
		ctx, cancel := context.WithCancel(context.Background())

		_, err := ydb.Open(ctx, "grpc://localhost:2136/local", ydb.WithTraceDriver(trace.Driver{
			OnInit: func(disi trace.DriverInitStartInfo) func(trace.DriverInitDoneInfo) {
				cancel()

				return func(didi trace.DriverInitDoneInfo) {}
			},
		}))
		require.ErrorIs(t, err, context.Canceled)
		assert.Regexp(t, "^context canceled at", err.Error())
	})

	t.Run("connect failure releases pool ref", func(t *testing.T) {
		ctx, cancel := context.WithCancel(context.Background())

		_, err := ydb.Open(ctx, "grpc://127.0.0.1:1/local",
			ydb.WithTraceDriver(trace.Driver{
				OnBalancerInit: func(trace.DriverBalancerInitStartInfo) func(trace.DriverBalancerInitDoneInfo) {
					cancel()

					return func(trace.DriverBalancerInitDoneInfo) {}
				},
			}),
		)
		require.Error(t, err)
	})
}