mirror of https://github.com/jackc/pgx.git
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package pgx
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func unescape(s string) string {
|
|
s = strings.Replace(s, `\:`, `:`, -1)
|
|
s = strings.Replace(s, `\\`, `\`, -1)
|
|
return s
|
|
}
|
|
|
|
var passfile = [][]string{
|
|
[]string{"test1", "5432", "larrydb", "larry", "whatstheidea"},
|
|
[]string{"test1", "5432", "moedb", "moe", "imbecile"},
|
|
[]string{"test1", "5432", "curlydb", "curly", "nyuknyuknyuk"},
|
|
[]string{"test2", "5432", "*", "shemp", "heymoe"},
|
|
[]string{"test2", "5432", "*", "*", `test\\ing\:`},
|
|
}
|
|
|
|
func TestPGPass(t *testing.T) {
|
|
tf, err := ioutil.TempFile("", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer tf.Close()
|
|
defer os.Remove(tf.Name())
|
|
os.Setenv("PGPASSFILE", tf.Name())
|
|
for _, l := range passfile {
|
|
_, err := fmt.Fprintln(tf, strings.Join(l, `:`))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if err = tf.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for i, l := range passfile {
|
|
cfg := ConnConfig{Host: l[0], Database: l[2], User: l[3]}
|
|
found := pgpass(&cfg)
|
|
if !found {
|
|
t.Fatalf("Entry %v not found", i)
|
|
}
|
|
if cfg.Password != unescape(l[4]) {
|
|
t.Fatalf(`Password mismatch entry %v want %s got %s`, i, unescape(l[4]), cfg.Password)
|
|
}
|
|
}
|
|
cfg := ConnConfig{Host: "derp", Database: "herp", User: "joe"}
|
|
found := pgpass(&cfg)
|
|
if found {
|
|
t.Fatal("bad found")
|
|
}
|
|
}
|