tests/gobang.go: Add tests to more functions

1 files changed, 15 insertions, 28 deletions

diff --git a/src/gobang.go b/src/gobang.go
index ff8f68b..fff0503 100644
--- a/src/gobang.go
+++ b/src/gobang.go
@@ -740,38 +740,25 @@ func setLoggerOutput(w io.Writer) {
 	))
 }
 
-func levelFromString(name string) (bool, logLevel) {
-	label := strings.ToUpper(name)
-
-	if label == "NONE" {
-		return true, LevelNone
-	}
-
-	if label == "ERROR" {
-		return true, LevelError
-	}
-
-	if label == "WARNING" {
-		return true, LevelWarning
-	}
-
-	if label == "INFO" {
-		return true, LevelInfo
-	}
-
-	if label == "DEBUG" {
-		return true, LevelDebug
+func levelFromString(name string, fallback logLevel) logLevel {
+	switch strings.ToUpper(name) {
+		case "NONE":
+			return LevelNone
+		case "ERROR":
+			return LevelError
+		case "WARNING":
+			return LevelWarning
+		case "INFO":
+			return LevelInfo
+		case "DEBUG":
+			return LevelDebug
+		default:
+			return fallback
 	}
-
-	return false, level
 }
 
 func setLogLevel() {
-	ok, envLevel := levelFromString(os.Getenv("LOG_LEVEL"))
-
-	if ok {
-		level = envLevel
-	}
+	level = levelFromString(os.Getenv("LOG_LEVEL"), level)
 }
 
 func setMetric() {