pgpass_test.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")
	}
}