From ee14a3c9bed326ae9577358549c7124b6e2cc252 Mon Sep 17 00:00:00 2001
From: Sung Won Cho <sung@monomax.sh>
Date: Tue, 23 Jul 2019 17:06:03 +1000
Subject: [PATCH] Fix test by enabling database SSL only in production (#226)

* Fix test

* Always specify sslmode
---
 pkg/server/database/database.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pkg/server/database/database.go b/pkg/server/database/database.go
index c2c35118..2404c8a7 100644
--- a/pkg/server/database/database.go
+++ b/pkg/server/database/database.go
@@ -34,13 +34,21 @@ var (
 )
 
 func getPGConnectionString() string {
+	var sslmode string
+	if os.Getenv("GO_ENV") == "PRODUCTION" {
+		sslmode = "require"
+	} else {
+		sslmode = "disable"
+	}
+
 	return fmt.Sprintf(
-		"host=%s port=%s dbname=%s user=%s password=%s",
+		"host=%s port=%s dbname=%s user=%s password=%s sslmode=%s",
 		os.Getenv("DBHost"),
 		os.Getenv("DBPort"),
 		os.Getenv("DBName"),
 		os.Getenv("DBUser"),
 		os.Getenv("DBPassword"),
+		sslmode,
 	)
 }