mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-18 21:52:28 +03:00
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
package mysql_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/neilotoole/sq/drivers/mysql"
|
|
"github.com/neilotoole/sq/libsq/core/kind"
|
|
"github.com/neilotoole/sq/libsq/core/lg"
|
|
"github.com/neilotoole/sq/libsq/core/lg/lgt"
|
|
"github.com/neilotoole/sq/testh"
|
|
"github.com/neilotoole/sq/testh/sakila"
|
|
)
|
|
|
|
func TestKindFromDBTypeName(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx := lg.NewContext(context.Background(), lgt.New(t))
|
|
|
|
testCases := map[string]kind.Kind{
|
|
"": kind.Unknown,
|
|
"INTEGER": kind.Int,
|
|
"INT": kind.Int,
|
|
"SMALLINT": kind.Int,
|
|
"TINYINT": kind.Int,
|
|
"MEDIUMINT": kind.Int,
|
|
"BIGINT": kind.Int,
|
|
"BIT": kind.Int,
|
|
"DECIMAL": kind.Decimal,
|
|
"DECIMAL(5,2)": kind.Decimal,
|
|
"NUMERIC": kind.Decimal,
|
|
"FLOAT": kind.Float,
|
|
"FLOAT(8)": kind.Float,
|
|
"FLOAT(7,4)": kind.Float,
|
|
"REAL": kind.Float,
|
|
"DOUBLE": kind.Float,
|
|
"DOUBLE PRECISION": kind.Float,
|
|
"DATE": kind.Date,
|
|
"DATETIME": kind.Datetime,
|
|
"TIMESTAMP": kind.Datetime,
|
|
"TIME": kind.Time,
|
|
"YEAR": kind.Int,
|
|
"CHAR": kind.Text,
|
|
"VARCHAR": kind.Text,
|
|
"VARCHAR(64)": kind.Text,
|
|
"TINYTEXT": kind.Text,
|
|
"TEXT": kind.Text,
|
|
"MEDIUMTEXT": kind.Text,
|
|
"LONGTEXT": kind.Text,
|
|
"BINARY": kind.Bytes,
|
|
"BINARY(4)": kind.Bytes,
|
|
"VARBINARY": kind.Bytes,
|
|
"BLOB": kind.Bytes,
|
|
"MEDIUMBLOB": kind.Bytes,
|
|
"LONGBLOB": kind.Bytes,
|
|
"ENUM": kind.Text,
|
|
"SET": kind.Text,
|
|
"BOOL": kind.Bool,
|
|
"BOOLEAN": kind.Bool,
|
|
}
|
|
|
|
for dbTypeName, wantKind := range testCases {
|
|
gotKind := mysql.KindFromDBTypeName(ctx, "col", dbTypeName)
|
|
require.Equal(t, wantKind, gotKind, "{%s} should produce %s but got %s", dbTypeName, wantKind, gotKind)
|
|
}
|
|
}
|
|
|
|
func TestDatabase_SourceMetadata_MySQL(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
handles := sakila.MyAll()
|
|
for _, handle := range handles {
|
|
handle := handle
|
|
|
|
t.Run(handle, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
th, _, _, grip, _ := testh.NewWith(t, handle)
|
|
md, err := grip.SourceMetadata(th.Context, false)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "sakila", md.Name)
|
|
require.Equal(t, handle, md.Handle)
|
|
|
|
tblActor := md.Tables[0]
|
|
require.Equal(t, sakila.TblActor, tblActor.Name)
|
|
require.Equal(t, int64(sakila.TblActorCount), tblActor.RowCount)
|
|
require.Equal(t, len(sakila.TblActorCols()), len(tblActor.Columns))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDatabase_TableMetadata(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
for _, handle := range sakila.MyAll() {
|
|
handle := handle
|
|
|
|
t.Run(handle, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
th, _, _, grip, _ := testh.NewWith(t, handle)
|
|
md, err := grip.TableMetadata(th.Context, sakila.TblActor)
|
|
require.NoError(t, err)
|
|
require.Equal(t, sakila.TblActor, md.Name)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetTableRowCounts(t *testing.T) {
|
|
th, _, _, _, db := testh.NewWith(t, sakila.My)
|
|
|
|
counts, err := mysql.GetTableRowCountsBatch(th.Context, db, []string{sakila.TblActor, sakila.TblFilm})
|
|
require.NoError(t, err)
|
|
require.Len(t, counts, 2)
|
|
require.Equal(t, int64(sakila.TblActorCount), counts[sakila.TblActor])
|
|
require.Equal(t, int64(sakila.TblFilmCount), counts[sakila.TblFilm])
|
|
}
|