dbmate/driver_test.go

27 lines
539 B
Go
Raw Normal View History

2015-12-01 09:43:35 -08:00
package main
import (
"github.com/stretchr/testify/require"
"testing"
)
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)
}