sq/drivers/sqlite3/extension_test.go
Neil O'Toole 26f0c9a381
Refactor source.Files (#363)
* Moved `source.Files` to its own package, thus the type is now `files.Files`.
* Moved much of the location functionality from pkg `source` to its own package `location`.
2024-01-24 23:29:55 -07:00

67 lines
2.4 KiB
Go

//go:build sqlite_vtable && sqlite_fts5
package sqlite3_test
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/neilotoole/sq/libsq/core/kind"
"github.com/neilotoole/sq/libsq/core/sqlz"
"github.com/neilotoole/sq/libsq/source"
"github.com/neilotoole/sq/libsq/source/drivertype"
"github.com/neilotoole/sq/testh"
"github.com/neilotoole/sq/testh/sakila"
"github.com/neilotoole/sq/testh/tu"
)
func TestExtension_fts5(t *testing.T) {
const tblActorFts = "actor_fts"
th := testh.New(t)
src := th.Add(&source.Source{
Handle: "@fts",
Type: drivertype.SQLite,
Location: "sqlite3://" + tu.MustAbsFilepath("testdata", "sakila_fts5.db"),
})
srcMeta, err := th.SourceMetadata(src)
require.NoError(t, err)
require.Equal(t, src.Handle, srcMeta.Handle)
tblMeta1 := srcMeta.Table(tblActorFts)
require.NotNil(t, tblMeta1)
require.Equal(t, tblActorFts, tblMeta1.Name)
require.Equal(t, sqlz.TableTypeVirtual, tblMeta1.TableType)
require.Equal(t, "actor_id", tblMeta1.Columns[0].Name)
require.Equal(t, "integer", tblMeta1.Columns[0].ColumnType)
require.Equal(t, "integer", tblMeta1.Columns[0].BaseType)
require.Equal(t, kind.Int, tblMeta1.Columns[0].Kind)
require.Equal(t, "first_name", tblMeta1.Columns[1].Name)
require.Equal(t, "text", tblMeta1.Columns[1].ColumnType)
require.Equal(t, "text", tblMeta1.Columns[1].BaseType)
require.Equal(t, kind.Text, tblMeta1.Columns[1].Kind)
require.Equal(t, "last_name", tblMeta1.Columns[2].Name)
require.Equal(t, "text", tblMeta1.Columns[2].ColumnType)
require.Equal(t, "text", tblMeta1.Columns[2].BaseType)
require.Equal(t, kind.Text, tblMeta1.Columns[2].Kind)
require.Equal(t, "last_update", tblMeta1.Columns[3].Name)
require.Equal(t, "text", tblMeta1.Columns[3].ColumnType)
require.Equal(t, "text", tblMeta1.Columns[3].BaseType)
require.Equal(t, kind.Text, tblMeta1.Columns[3].Kind)
tblMeta2, err := th.TableMetadata(src, tblActorFts)
require.NoError(t, err)
require.Equal(t, tblActorFts, tblMeta2.Name)
require.Equal(t, sqlz.TableTypeVirtual, tblMeta2.TableType)
require.EqualValues(t, *tblMeta1, *tblMeta2)
// Verify that the (non-virtual) "actor" table has its type set correctly.
actorMeta1 := srcMeta.Table(sakila.TblActor)
actorMeta2, err := th.TableMetadata(src, sakila.TblActor)
require.NoError(t, err)
require.Equal(t, actorMeta1.TableType, sqlz.TableTypeTable)
require.Equal(t, *actorMeta1, *actorMeta2)
}