internal/query/errors_test.go - master - githubmirror/ydb-go-sdk

internal/query/errors_test.go
86 строк3 KB
Aleksey Myasnikov
Pessimize endpoint on attach stream hints (#2174)
29 май 2026, 19:21
29 май 2026, 19:2160b42dd
package query

import (
	"errors"
	"testing"

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

func TestErrors(t *testing.T) {
	t.Run("errNilClient", func(t *testing.T) {
		require.NotNil(t, errNilClient)
		require.Contains(t, errNilClient.Error(), "table client is not initialized")
	})

	t.Run("errClosedClient", func(t *testing.T) {
		require.NotNil(t, errClosedClient)
		require.Contains(t, errClosedClient.Error(), "query client closed early")
	})

	t.Run("ErrTransactionRollingBack", func(t *testing.T) {
		require.NotNil(t, ErrTransactionRollingBack)
		require.Contains(t, ErrTransactionRollingBack.Error(), "the transaction is rolling back")
	})

	t.Run("errWrongNextResultSetIndex", func(t *testing.T) {
		require.NotNil(t, errWrongNextResultSetIndex)
		require.Equal(t, "wrong result set index", errWrongNextResultSetIndex.Error())
	})

	t.Run("errWrongResultSetIndex", func(t *testing.T) {
		require.NotNil(t, errWrongResultSetIndex)
		require.Equal(t, "critical violation of the logic - wrong result set index", errWrongResultSetIndex.Error())
	})

	t.Run("ErrMoreThanOneRow", func(t *testing.T) {
		require.NotNil(t, ErrMoreThanOneRow)
		require.Equal(t, "unexpected more than one row in result set", ErrMoreThanOneRow.Error())
	})

	t.Run("ErrMoreThanOneResultSet", func(t *testing.T) {
		require.NotNil(t, ErrMoreThanOneResultSet)
		require.Equal(t, "unexpected more than one result set", ErrMoreThanOneResultSet.Error())
	})

	t.Run("ErrNoResultSets", func(t *testing.T) {
		require.NotNil(t, ErrNoResultSets)
		require.Equal(t, "no result sets", ErrNoResultSets.Error())
	})

	t.Run("errNilOption", func(t *testing.T) {
		require.NotNil(t, errNilOption)
		require.Equal(t, "nil option", errNilOption.Error())
	})

	t.Run("ErrOptionNotForTxExecute", func(t *testing.T) {
		require.NotNil(t, ErrOptionNotForTxExecute)
		require.Equal(t, "option is not for execute on transaction", ErrOptionNotForTxExecute.Error())
	})

	t.Run("errExecuteOnCompletedTx", func(t *testing.T) {
		require.NotNil(t, errExecuteOnCompletedTx)
		require.Equal(t, "execute on completed transaction", errExecuteOnCompletedTx.Error())
	})

	t.Run("errSessionClosed", func(t *testing.T) {
		require.NotNil(t, errSessionClosed)
		require.Equal(t, "session is closed", errSessionClosed.Error())
	})

	t.Run("errNodeShutdownHint", func(t *testing.T) {
		require.NotNil(t, errNodeShutdownHint)
		require.Contains(t, errNodeShutdownHint.Error(), "received node shutdown hint")
	})

	t.Run("ErrorsAreUnique", func(t *testing.T) {
		// Verify that error variables are distinct
		require.False(t, errors.Is(errNilClient, errClosedClient))
		require.False(t, errors.Is(errClosedClient, ErrTransactionRollingBack))
		require.False(t, errors.Is(errWrongNextResultSetIndex, errWrongResultSetIndex))
		require.False(t, errors.Is(ErrMoreThanOneRow, ErrMoreThanOneResultSet))
		require.False(t, errors.Is(ErrNoResultSets, errNilOption))
		require.False(t, errors.Is(ErrOptionNotForTxExecute, errExecuteOnCompletedTx))
		require.False(t, errors.Is(errExecuteOnCompletedTx, errSessionClosed))
	})
}