dbmate/driver_test.go
2017-05-04 20:58:23 -07:00

27 lines
542 B
Go

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