dbmate/pkg/dbmate/driver_test.go
2018-04-15 20:47:24 -07:00

27 lines
551 B
Go

package dbmate
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestGetDriver_Postgres(t *testing.T) {
drv, err := GetDriver("postgres")
require.NoError(t, err)
_, ok := drv.(PostgresDriver)
require.Equal(t, true, ok)
}
func TestGetDriver_MySQL(t *testing.T) {
drv, err := GetDriver("mysql")
require.NoError(t, err)
_, ok := drv.(MySQLDriver)
require.Equal(t, true, ok)
}
func TestGetDriver_Error(t *testing.T) {
drv, err := GetDriver("foo")
require.EqualError(t, err, "unsupported driver: foo")
require.Nil(t, drv)
}