switch->if, TODO

m-lab · Jan 2, 2019 · f49a079 · f49a079
1 parent 73d2258
commit f49a079
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/api/v2/api-v2_test.go b/api/v2/api-v2_test.go
@@ -25,10 +25,9 @@ func TestDoRequest(t *testing.T) {
 	callCount := 0
 
 	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		switch {
-		case callCount < 3:
+		if callCount < 3 {
 			w.WriteHeader(http.StatusServiceUnavailable)
-		default:
+		} else {
 			fmt.Fprint(w, expectedJson)
 		}
 		callCount++
@@ -67,11 +66,10 @@ func TestDoRequest(t *testing.T) {
 func TestSomeErrors(t *testing.T) {
 	callCount := 0
 	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		switch {
-		case callCount == 0:
+		if callCount == 0 {
 			w.WriteHeader(http.StatusInternalServerError)
 			fmt.Fprint(w, "body message")
-		default:
+		} else {
 			w.WriteHeader(http.StatusServiceUnavailable)
 		}
 		callCount++

diff --git a/handler/handler_test.go b/handler/handler_test.go
@@ -1,5 +1,8 @@
 package handler_test
 
+// TODO 201901 - TestStatusServiceUnavailable()
+// TODO 201901 - TestStatusInternalError()
+
 import (
 	"bytes"
 	"errors"