Skip to content

Commit

Permalink
Merge pull request #643 from upper/default-sslmode
Browse files Browse the repository at this point in the history
Update default sslmode
  • Loading branch information
xiam authored Jan 3, 2022
2 parents 4e2011c + eb6759b commit cb6aebc
Show file tree
Hide file tree
Showing 8 changed files with 42 additions and 34 deletions.
7 changes: 4 additions & 3 deletions adapter/cockroachdb/connection_pgx.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build !pq
// +build !pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -49,7 +50,7 @@ func (c ConnectionURL) String() (s string) {
host = "127.0.0.1"
}
if port == "" {
port = "5432"
port = "26257"
}
u = append(u, "host="+escaper.Replace(host))
u = append(u, "port="+escaper.Replace(port))
Expand All @@ -75,9 +76,9 @@ func (c ConnectionURL) String() (s string) {
c.Options = map[string]string{}
}

// If not present, SSL mode is assumed disabled.
// If not present, SSL mode is assumed "prefer".
if sslMode, ok := c.Options["sslmode"]; !ok || sslMode == "" {
c.Options["sslmode"] = "disable"
c.Options["sslmode"] = "prefer"
}

// Disabled by default
Expand Down
13 changes: 7 additions & 6 deletions adapter/cockroachdb/connection_pgx_test.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build !pq
// +build !pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -37,27 +38,27 @@ func TestConnectionURL(t *testing.T) {

// Adding a host with port.
c.Host = "localhost:1234"
assert.Equal(t, "host=localhost port=1234 sslmode=disable statement_cache_capacity=0", c.String())
assert.Equal(t, "host=localhost port=1234 sslmode=prefer statement_cache_capacity=0", c.String())

// Adding a host.
c.Host = "localhost"
assert.Equal(t, "host=localhost sslmode=disable statement_cache_capacity=0", c.String())
assert.Equal(t, "host=localhost sslmode=prefer statement_cache_capacity=0", c.String())

// Adding a username.
c.User = "Anakin"
assert.Equal(t, `host=localhost sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `host=localhost sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding a password with special characters.
c.Password = "Some Sort of ' Password"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding a port.
c.Host = "localhost:1234"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding a database.
c.Database = "MyDatabase"
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding options.
c.Options = map[string]string{
Expand Down
7 changes: 4 additions & 3 deletions adapter/cockroachdb/connection_pq.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build pq
// +build pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -49,7 +50,7 @@ func (c ConnectionURL) String() (s string) {
host = "127.0.0.1"
}
if port == "" {
port = "5432"
port = "26257"
}
u = append(u, "host="+escaper.Replace(host))
u = append(u, "port="+escaper.Replace(port))
Expand All @@ -75,9 +76,9 @@ func (c ConnectionURL) String() (s string) {
c.Options = map[string]string{}
}

// If not present, SSL mode is assumed disabled.
// If not present, SSL mode is assumed "prefer".
if sslMode, ok := c.Options["sslmode"]; !ok || sslMode == "" {
c.Options["sslmode"] = "disable"
c.Options["sslmode"] = "prefer"
}

for k, v := range c.Options {
Expand Down
13 changes: 7 additions & 6 deletions adapter/cockroachdb/connection_pq_test.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build pq
// +build pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -37,27 +38,27 @@ func TestConnectionURL(t *testing.T) {

// Adding a host with port.
c.Host = "localhost:1234"
assert.Equal(t, "host=localhost port=1234 sslmode=disable", c.String())
assert.Equal(t, "host=localhost port=1234 sslmode=prefer", c.String())

// Adding a host.
c.Host = "localhost"
assert.Equal(t, "host=localhost sslmode=disable", c.String())
assert.Equal(t, "host=localhost sslmode=prefer", c.String())

// Adding a username.
c.User = "Anakin"
assert.Equal(t, `host=localhost sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `host=localhost sslmode=prefer user=Anakin`, c.String())

// Adding a password with special characters.
c.Password = "Some Sort of ' Password"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=prefer user=Anakin`, c.String())

// Adding a port.
c.Host = "localhost:1234"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer user=Anakin`, c.String())

// Adding a database.
c.Database = "MyDatabase"
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer user=Anakin`, c.String())

// Adding options.
c.Options = map[string]string{
Expand Down
5 changes: 3 additions & 2 deletions adapter/postgresql/connection_pgx.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build !pq
// +build !pq

package postgresql
Expand Down Expand Up @@ -75,9 +76,9 @@ func (c ConnectionURL) String() (s string) {
c.Options = map[string]string{}
}

// If not present, SSL mode is assumed disabled.
// If not present, SSL mode is assumed "prefer".
if sslMode, ok := c.Options["sslmode"]; !ok || sslMode == "" {
c.Options["sslmode"] = "disable"
c.Options["sslmode"] = "prefer"
}

// Disabled by default
Expand Down
13 changes: 7 additions & 6 deletions adapter/postgresql/connection_pgx_test.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build !pq
// +build !pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -37,27 +38,27 @@ func TestConnectionURL(t *testing.T) {

// Adding a host with port.
c.Host = "localhost:1234"
assert.Equal(t, "host=localhost port=1234 sslmode=disable statement_cache_capacity=0", c.String())
assert.Equal(t, "host=localhost port=1234 sslmode=prefer statement_cache_capacity=0", c.String())

// Adding a host.
c.Host = "localhost"
assert.Equal(t, "host=localhost sslmode=disable statement_cache_capacity=0", c.String())
assert.Equal(t, "host=localhost sslmode=prefer statement_cache_capacity=0", c.String())

// Adding a username.
c.User = "Anakin"
assert.Equal(t, `host=localhost sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `host=localhost sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding a password with special characters.
c.Password = "Some Sort of ' Password"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding a port.
c.Host = "localhost:1234"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding a database.
c.Database = "MyDatabase"
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable statement_cache_capacity=0 user=Anakin`, c.String())
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer statement_cache_capacity=0 user=Anakin`, c.String())

// Adding options.
c.Options = map[string]string{
Expand Down
5 changes: 3 additions & 2 deletions adapter/postgresql/connection_pq.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build pq
// +build pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -75,9 +76,9 @@ func (c ConnectionURL) String() (s string) {
c.Options = map[string]string{}
}

// If not present, SSL mode is assumed disabled.
// If not present, SSL mode is assumed "prefer".
if sslMode, ok := c.Options["sslmode"]; !ok || sslMode == "" {
c.Options["sslmode"] = "disable"
c.Options["sslmode"] = "prefer"
}

for k, v := range c.Options {
Expand Down
13 changes: 7 additions & 6 deletions adapter/postgresql/connection_pq_test.go
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
//go:build pq
// +build pq

// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
Expand Down Expand Up @@ -37,27 +38,27 @@ func TestConnectionURL(t *testing.T) {

// Adding a host with port.
c.Host = "localhost:1234"
assert.Equal(t, "host=localhost port=1234 sslmode=disable", c.String())
assert.Equal(t, "host=localhost port=1234 sslmode=prefer", c.String())

// Adding a host.
c.Host = "localhost"
assert.Equal(t, "host=localhost sslmode=disable", c.String())
assert.Equal(t, "host=localhost sslmode=prefer", c.String())

// Adding a username.
c.User = "Anakin"
assert.Equal(t, `host=localhost sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `host=localhost sslmode=prefer user=Anakin`, c.String())

// Adding a password with special characters.
c.Password = "Some Sort of ' Password"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password sslmode=prefer user=Anakin`, c.String())

// Adding a port.
c.Host = "localhost:1234"
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer user=Anakin`, c.String())

// Adding a database.
c.Database = "MyDatabase"
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=disable user=Anakin`, c.String())
assert.Equal(t, `dbname=MyDatabase host=localhost password=Some\ Sort\ of\ \'\ Password port=1234 sslmode=prefer user=Anakin`, c.String())

// Adding options.
c.Options = map[string]string{
Expand Down

0 comments on commit cb6aebc

Please sign in to comment.