migration_test.go

package integration

// Basic imports
import (
	"os"
	"testing"

	gonanoid "github.com/matoous/go-nanoid/v2"
	"github.com/spf13/cobra"
	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"
	clicmd "github.com/supabase/cli/cmd"
	"github.com/supabase/cli/test/mocks/supabase"
)

type MigrationTestSuite struct {
	suite.Suite
	tempDir string
	cmd     *cobra.Command
}

// test functions
func (suite *MigrationTestSuite) TestNewMigration() {
	// run command
	migration, _, err := suite.cmd.Find([]string{"migration", "new"})
	require.NoError(suite.T(), err)
	name := gonanoid.MustGenerate(supabase.IDAlphabet, 10)
	require.NoError(suite.T(), migration.RunE(migration, []string{name}))

	// check migrations file created
	subs, err := os.ReadDir("supabase/migrations")
	require.NoError(suite.T(), err)
	require.Regexp(suite.T(), `[0-9]{14}_`+name+".sql", subs[0].Name())
}

// hooks
func (suite *MigrationTestSuite) SetupTest() {
	// init cli
	suite.cmd = clicmd.GetRootCmd()
	suite.tempDir = NewTempDir(Logger, TempDir)

	// init supabase
	init, _, err := suite.cmd.Find([]string{"init"})
	require.NoError(suite.T(), err)
	require.NoError(suite.T(), init.RunE(init, []string{}))
}

func (suite *MigrationTestSuite) TeardownTest() {
	require.NoError(suite.T(), os.Chdir(TempDir))
}

// In order for 'go test' to run this suite, we need to create
// a normal test function and pass our suite to suite.Run
func TestMigrationTestSuite(t *testing.T) {
	suite.Run(t, new(MigrationTestSuite))
}