From 70c896be119a79fbf05890cc2b956784ad2ca3fa Mon Sep 17 00:00:00 2001
From: mochi-w650z <mochice.mls.ntl@gmail.com>
Date: Fri, 1 Oct 2021 11:13:09 +0900
Subject: [PATCH] :muscle: Add test pattern

---
 main_test.go | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/main_test.go b/main_test.go
index 23f3e7f..26d1d6c 100644
--- a/main_test.go
+++ b/main_test.go
@@ -111,6 +111,31 @@ func TestValidateResponse(t *testing.T) {
 			}
 		}
 	}
-	// TODO test when body is valid/invalid as json
+	// test when body is valid/invalid as json
+	invalidResp := http.Response{
+		Status:     "444 not exists error",
+		StatusCode: 444,
+		Body:       ioutil.NopCloser(bytes.NewBuffer([]byte("test"))),
+	}
+	err = ValidateResponse(&invalidResp)
+	if err == nil {
+		t.Fatalf("If status code is invalid, should occur error\nActual: %d", invalidResp.StatusCode)
+	} else if !strings.HasSuffix(err.Error(), "]") {
+		t.Fatalf("If body is invalid as json, error is formated as %s",
+			"`Invalid response [statuscode statustext]`")
+	}
 
+	expectedMessage := "This is test"
+	validResp := http.Response{
+		Status:     "444 not exists error",
+		StatusCode: 444,
+		Body:       ioutil.NopCloser(bytes.NewBuffer([]byte(fmt.Sprintf(`{"message": "%s"}`, expectedMessage)))),
+	}
+	err = ValidateResponse(&validResp)
+	if err == nil {
+		t.Fatalf("If is status code invalid, should occur error\nActual: %d", validResp.StatusCode)
+	} else if !strings.HasSuffix(err.Error(), expectedMessage) {
+		t.Fatalf("If body is valid as json, error suffix should have `%s`\nActual: %s",
+			expectedMessage, err.Error())
+	}
 }