internal/xsql/rows_test.go - master - githubmirror/ydb-go-sdk

internal/xsql/rows_test.go

97 строк2 KB

Aleksey Myasnikov

refactoring for better perf: replaced done signal channels to `atomic.Bool`, simlpified internals of database/sql driver and query-client (#2148)

11 май 2026, 22:25

Не верифицирован

11 май 2026, 22:25be40ba3

package xsql

import (
	"database/sql"
	"database/sql/driver"
	"io"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/ydb-platform/ydb-go-sdk/v3/internal/xsql/common"
)

var _ common.Rows = (*singleRow)(nil)

func TestRowByAstPlan(t *testing.T) {
	t.Run("CreateRow", func(t *testing.T) {
		ast := "test ast"
		plan := "test plan"
		row := rowByAstPlan(ast, plan)

		require.NotNil(t, row)
		require.Len(t, row.values, 2)
		require.Equal(t, "Ast", row.values[0].Name)
		require.Equal(t, ast, row.values[0].Value)
		require.Equal(t, "Plan", row.values[1].Name)
		require.Equal(t, plan, row.values[1].Value)
		require.False(t, row.readAll)
	})
}

func TestSingleRow_Columns(t *testing.T) {
	row := rowByAstPlan("ast", "plan")
	columns := row.Columns(t.Context())

	require.Len(t, columns, 2)
	require.Equal(t, "Ast", columns[0])
	require.Equal(t, "Plan", columns[1])
}

func TestSingleRow_Close(t *testing.T) {
	row := rowByAstPlan("ast", "plan")
	err := row.Close(t.Context())
	require.NoError(t, err)
}

func TestSingleRow_Next(t *testing.T) {
	ctx := t.Context()

	t.Run("FirstRead", func(t *testing.T) {
		ast := "test ast"
		plan := "test plan"
		row := rowByAstPlan(ast, plan)

		dst := make([]driver.Value, 2)
		err := row.Next(ctx, dst)

		require.NoError(t, err)
		require.Equal(t, ast, dst[0])
		require.Equal(t, plan, dst[1])
		require.True(t, row.readAll)
	})

	t.Run("SecondRead", func(t *testing.T) {
		row := rowByAstPlan("ast", "plan")

		dst := make([]driver.Value, 2)
		err := row.Next(ctx, dst)
		require.NoError(t, err)

		err = row.Next(ctx, dst)
		require.ErrorIs(t, err, io.EOF)
	})

	t.Run("NilValues", func(t *testing.T) {
		row := &singleRow{
			values:  nil,
			readAll: false,
		}

		dst := make([]driver.Value, 2)
		err := row.Next(ctx, dst)
		require.ErrorIs(t, err, io.EOF)
	})

	t.Run("EmptyValues", func(t *testing.T) {
		row := &singleRow{
			values:  []sql.NamedArg{},
			readAll: false,
		}

		dst := make([]driver.Value, 0)
		err := row.Next(ctx, dst)
		require.NoError(t, err)
		require.True(t, row.readAll)
	})
}