github.com/ydb-platform/ydb-go-sdk/v3@v3.89.2/tests/integration/database_sql_get_index_columns_test.go (about)

     1  //go:build integration
     2  // +build integration
     3  
     4  package integration
     5  
     6  import (
     7  	"context"
     8  	"database/sql"
     9  	"errors"
    10  	"testing"
    11  
    12  	"github.com/stretchr/testify/require"
    13  
    14  	"github.com/ydb-platform/ydb-go-sdk/v3"
    15  	"github.com/ydb-platform/ydb-go-sdk/v3/retry"
    16  )
    17  
    18  func TestDatabaseSqlGetIndexColumns(t *testing.T) {
    19  	var (
    20  		scope = newScope(t)
    21  		db    = scope.SQLDriverWithFolder()
    22  	)
    23  
    24  	defer func() {
    25  		_ = db.Close()
    26  	}()
    27  
    28  	t.Run("create-tables", func(t *testing.T) {
    29  		err := retry.Do(scope.Ctx, db, func(ctx context.Context, cc *sql.Conn) (err error) {
    30  			_, err = cc.ExecContext(
    31  				ydb.WithQueryMode(ctx, ydb.SchemeQueryMode),
    32  				`CREATE TABLE seasons (
    33  				series_id Uint64,
    34  				season_id Uint64,
    35  				title UTF8,
    36  				first_aired Date,
    37  				last_aired Date,
    38  				INDEX index_series_title GLOBAL ON (title),
    39  				INDEX index_seasons_aired_date GLOBAL ON (first_aired, last_aired),
    40  				PRIMARY KEY (
    41  					series_id,
    42  					season_id
    43  				)
    44  			)`,
    45  			)
    46  			if err != nil {
    47  				return err
    48  			}
    49  
    50  			return nil
    51  		}, retry.WithIdempotent(true))
    52  
    53  		require.NoError(t, err)
    54  	})
    55  
    56  	t.Run("get-index-columns", func(t *testing.T) {
    57  		for _, test := range []struct {
    58  			IndexName      string
    59  			IndexedColumns []string
    60  		}{
    61  			{
    62  				IndexName:      "index_series_title",
    63  				IndexedColumns: []string{"title"},
    64  			},
    65  			{
    66  				IndexName:      "index_seasons_aired_date",
    67  				IndexedColumns: []string{"first_aired", "last_aired"},
    68  			},
    69  		} {
    70  			err := retry.Do(scope.Ctx, db, func(ctx context.Context, cc *sql.Conn) (err error) {
    71  				columns := make([]string, 0)
    72  				err = cc.Raw(func(drvConn interface{}) (err error) {
    73  					q, ok := drvConn.(interface {
    74  						GetIndexColumns(context.Context, string, string) ([]string, error)
    75  					})
    76  
    77  					if !ok {
    78  						return errors.New("drvConn does not implement extended API")
    79  					}
    80  
    81  					columns, err = q.GetIndexColumns(ctx, "./seasons", test.IndexName)
    82  					return err
    83  				})
    84  				if err != nil {
    85  					return err
    86  				}
    87  
    88  				require.ElementsMatch(t,
    89  					test.IndexedColumns,
    90  					columns)
    91  				return nil
    92  			}, retry.WithIdempotent(true))
    93  
    94  			require.NoError(t, err)
    95  		}
    96  	})
    97  }