dialect_postgresql_test.go - main - githubmirror/pop

dialect_postgresql_test.go
129 строк3 KB
Yonghwan SO
fixed url handling with reserved characters (fixes #402)
24 сен 2022, 08:29
24 сен 2022, 08:29e851c6c
package pop

import (
	"os"
	"os/user"
	"testing"

	"github.com/stretchr/testify/require"
)

func Test_PostgreSQL_ConnectionDetails_Values_Finalize(t *testing.T) {
	r := require.New(t)

	cd := &ConnectionDetails{
		Dialect:  "postgres",
		Database: "database",
		Host:     "host",
		Port:     "1234",
		User:     "user",
		Password: "pass#",
	}
	err := cd.Finalize()
	r.NoError(err)

	p := &postgresql{commonDialect: commonDialect{ConnectionDetails: cd}}

	r.Equal("postgres://user:pass%23@host:1234/database?", p.URL())
}

func Test_PostgreSQL_Connection_String(t *testing.T) {
	r := require.New(t)

	url := "host=host port=1234 dbname=database user=user password=pass#"
	cd := &ConnectionDetails{
		Dialect: "postgres",
		URL:     url,
	}
	err := cd.Finalize()
	r.NoError(err)

	r.Equal(url, cd.URL)
	r.Equal("postgres", cd.Dialect)
	r.Equal("host", cd.Host)
	r.Equal("pass#", cd.Password)
	r.Equal("1234", cd.Port)
	r.Equal("user", cd.User)
	r.Equal("database", cd.Database)
}

func Test_PostgreSQL_Connection_String_Options(t *testing.T) {
	r := require.New(t)

	url := "host=host port=1234 dbname=database user=user password=pass# sslmode=disable fallback_application_name=test_app connect_timeout=10 sslcert=/some/location sslkey=/some/other/location sslrootcert=/root/location"
	cd := &ConnectionDetails{
		Dialect: "postgres",
		URL:     url,
	}
	err := cd.Finalize()
	r.NoError(err)

	r.Equal(url, cd.URL)

	r.Equal("disable", cd.Options["sslmode"])
	r.Equal("test_app", cd.Options["fallback_application_name"])
}

func Test_PostgreSQL_Connection_String_Without_User(t *testing.T) {
	r := require.New(t)

	url := "dbname=database"
	cd := &ConnectionDetails{
		Dialect: "postgres",
		URL:     url,
	}
	err := cd.Finalize()
	r.NoError(err)

	uc := os.Getenv("PGUSER")
	if uc == "" {
		c, err := user.Current()
		if err == nil {
			uc = c.Username
		}
	}

	r.Equal(url, cd.URL)
	r.Equal("postgres", cd.Dialect)

	var foundHost bool
	for _, host := range []string{
		"/var/run/postgresql", // Debian
		"/private/tmp",        // OSX - homebrew
		"/tmp",                // standard PostgreSQL
		"localhost",           // Windows does not do sockets
	} {
		if cd.Host == host {
			foundHost = true
			break
		}
	}
	r.True(foundHost, `Got host: "%s"`, cd.Host)

	r.Equal(os.Getenv("PGPASSWORD"), cd.Password)
	r.Equal(portPostgreSQL, cd.Port) // fallback
	r.Equal(uc, cd.User)
	r.Equal("database", cd.Database)
}

func Test_PostgreSQL_Connection_String_Failure(t *testing.T) {
	r := require.New(t)

	url := "abc"
	cd := &ConnectionDetails{
		Dialect: "postgres",
		URL:     url,
	}
	err := cd.Finalize()
	r.Error(err)
	r.Equal("postgres", cd.Dialect)
}

func Test_PostgreSQL_Quotable(t *testing.T) {
	r := require.New(t)
	p := postgresql{}

	r.Equal(`"table_name"`, p.Quote("table_name"))
	r.Equal(`"schema"."table_name"`, p.Quote("schema.table_name"))
	r.Equal(`"schema"."table name"`, p.Quote(`"schema"."table name"`))
}