github.com/bingtel/dbmate@v1.4.1/pkg/dbmate/driver_test.go (about) 1 package dbmate 2 3 import ( 4 "testing" 5 6 "github.com/stretchr/testify/require" 7 ) 8 9 func TestGetDriver_Postgres(t *testing.T) { 10 drv, err := GetDriver("postgres") 11 require.NoError(t, err) 12 _, ok := drv.(PostgresDriver) 13 require.Equal(t, true, ok) 14 } 15 16 func TestGetDriver_MySQL(t *testing.T) { 17 drv, err := GetDriver("mysql") 18 require.NoError(t, err) 19 _, ok := drv.(MySQLDriver) 20 require.Equal(t, true, ok) 21 } 22 23 func TestGetDriver_Error(t *testing.T) { 24 drv, err := GetDriver("foo") 25 require.EqualError(t, err, "unsupported driver: foo") 26 require.Nil(t, drv) 27 }