mirror of
https://github.com/TECHNOFAB11/dbmate.git
synced 2025-12-16 18:03:48 +01:00
27 lines
540 B
Go
27 lines
540 B
Go
package main
|
|
|
|
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)
|
|
}
|