one place tst should be mapped to test

university-of-auckland · Dec 4, 2019 · 82386a5 · 82386a5
1 parent 8fc8e36
commit 82386a5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/handler/common.go b/handler/common.go
@@ -51,15 +51,23 @@ var (
 	qualifications map[string]string
 )
 
+func iif(cond bool, truePart, falsePart string) string {
+	if cond {
+		return truePart
+	}
+	return falsePart
+}
+
 func init() {
 	godotenv.Load()
 
 	env = os.Getenv("ENV")
 	verboseEnv := os.Getenv("VERBOSE")
 	verbose = verboseEnv != "" && verboseEnv != "n" && verboseEnv != "0" && verboseEnv != "false"
 	if env != "" && env != "prd" {
-		APIBaseURL = "https://api." + env + ".auckland.ac.nz/service"
-		OHBaseURL = "https://" + env + ".orcidhub.org.nz"
+		var e = iif(env == "tst", "test", env)
+		APIBaseURL = "https://api." + e + ".auckland.ac.nz/service"
+		OHBaseURL = "https://" + e + ".orcidhub.org.nz"
 	} else {
 		APIBaseURL = "https://api.auckland.ac.nz/service"
 		OHBaseURL = "https://orcidhub.org.nz"

diff --git a/handler/handler_internal_test.go b/handler/handler_internal_test.go
@@ -401,6 +401,11 @@ func testHealthCheck(t *testing.T) {
 	assert.NotNil(t, err)
 }
 
+func TestIif(t *testing.T) {
+	assert.Equal(t, "T", iif(true, "T", "F"))
+	assert.Equal(t, "F", iif(false, "T", "F"))
+}
+
 func TestIdentityGetORCID(t *testing.T) {
 	var id Identity
 	json.Unmarshal([]byte(`{