tests/integration/database_sql_is_table_exists_test.go - master - githubmirror/ydb-go-sdk

tests/integration/database_sql_is_table_exists_test.go

107 строк2 KB

Timofey Koolin

update gofumpt in CI for use samve version as golangcilint

18 сен 2024, 19:39

18 сен 2024, 19:391b080f5

//go:build integration
// +build integration

package integration

import (
	"context"
	"database/sql"
	"errors"
	"testing"

	"github.com/stretchr/testify/require"

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

func TestDatabaseSqlIsTableExists(t *testing.T) {
	var (
		scope = newScope(t)
		db    = scope.SQLDriverWithFolder()
	)

	defer func() {
		_ = db.Close()
	}()

	t.Run("drop-if-exists", func(t *testing.T) {
		err := retry.Do(scope.Ctx, db, func(ctx context.Context, cc *sql.Conn) (err error) {
			exists := true
			err = cc.Raw(func(drvConn interface{}) (err error) {
				q, ok := drvConn.(interface {
					IsTableExists(context.Context, string) (bool, error)
				})

				if !ok {
					return errors.New("drvConn does not implement extended API")
				}

				exists, err = q.IsTableExists(ctx, "series")
				return err
			})
			if err != nil {
				return err
			}

			if exists {
				_, err = cc.ExecContext(
					ydb.WithQueryMode(ctx, ydb.SchemeQueryMode),
					"DROP TABLE `series`")
			}

			return err
		}, retry.WithIdempotent(true))

		require.NoError(t, err)
	})

	t.Run("create-tables", func(t *testing.T) {
		err := retry.Do(scope.Ctx, db, func(ctx context.Context, cc *sql.Conn) (err error) {
			_, err = cc.ExecContext(
				ydb.WithQueryMode(ctx, ydb.SchemeQueryMode),
				`
			CREATE TABLE series (
				series_id Uint64,
				title UTF8,
				series_info UTF8,
				release_date Date,
				comment UTF8,
				PRIMARY KEY (
					series_id
				)
			);`,
			)

			return err
		}, retry.WithIdempotent(true))

		require.NoError(t, err)
	})

	t.Run("is-table-exists", func(t *testing.T) {
		err := retry.Do(scope.Ctx, db, func(ctx context.Context, cc *sql.Conn) (err error) {
			exists := false
			err = cc.Raw(func(drvConn interface{}) (err error) {
				q, ok := drvConn.(interface {
					IsTableExists(context.Context, string) (bool, error)
				})

				if !ok {
					return errors.New("drvConn does not implement extended API")
				}

				exists, err = q.IsTableExists(ctx, "series")
				return err
			})
			if err != nil {
				return err
			}

			require.True(t, exists)
			return nil
		}, retry.WithIdempotent(true))

		require.NoError(t, err)
	})
}