Support both postgresql:// and posgres:// url prefix

pkg/client/client_test.go

@@ -81,6 +81,18 @@ func test_NewClientFromUrl(t *testing.T) {
 	assert.Equal(t, url, client.ConnectionString)
 }
 
+func test_NewClientFromUrl2(t *testing.T) {
+	url := "postgresql://postgres@localhost/booktown?sslmode=disable"
+	client, err := NewFromUrl(url)
+
+	if err != nil {
+		defer client.Close()
+	}
+
+	assert.Equal(t, nil, err)
+	assert.Equal(t, url, client.ConnectionString)
+}
+
 func test_Test(t *testing.T) {
 	assert.Equal(t, nil, testClient.Test())
 }

pkg/connection/connection_string.go

@@ -28,7 +28,7 @@ func FormatUrl(opts command.Options) (string, error) {
 	url := opts.Url
 
 	// Make sure to only accept urls in a standard format
-	if !strings.Contains(url, "postgres://") {
+	if !strings.HasPrefix(url, "postgres://") && !strings.HasPrefix(url, "postgresql://") {
 		return "", errors.New("Invalid URL. Valid format: postgres://user:password@host:port/db?sslmode=mode")
 	}
 

pkg/connection/connection_string_test.go

@@ -12,7 +12,7 @@ import (
 func Test_Invalid_Url(t *testing.T) {
 	opts := command.Options{}
 	examples := []string{
-		"postgresql://foobar",
+		"postgre://foobar",
 		"foobar",
 	}