mirror of
https://github.com/TECHNOFAB11/dbmate.git
synced 2026-02-02 09:25:07 +01:00
35 lines
695 B
Go
35 lines
695 B
Go
package dbmate
|
|
|
|
import (
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDatabaseName(t *testing.T) {
|
|
u, err := url.Parse("ignore://localhost/foo?query")
|
|
require.NoError(t, err)
|
|
|
|
name := databaseName(u)
|
|
require.Equal(t, "foo", name)
|
|
}
|
|
|
|
func TestDatabaseName_Empty(t *testing.T) {
|
|
u, err := url.Parse("ignore://localhost")
|
|
require.NoError(t, err)
|
|
|
|
name := databaseName(u)
|
|
require.Equal(t, "", name)
|
|
}
|
|
|
|
func TestTrimLeadingSQLComments(t *testing.T) {
|
|
in := "--\n" +
|
|
"-- foo\n\n" +
|
|
"-- bar\n\n" +
|
|
"real stuff\n" +
|
|
"-- end\n"
|
|
out, err := trimLeadingSQLComments([]byte(in))
|
|
require.NoError(t, err)
|
|
require.Equal(t, "real stuff\n-- end\n", string(out))
|
|
}
|