Browse Source

Added TestGET404

Piotr Czajkowski 4 years ago
parent
commit
a4d68f7990
1 changed files with 23 additions and 0 deletions
  1. 23 0
      rest_test.go

+ 23 - 0
rest_test.go

@@ -84,3 +84,26 @@ func TestGET(t *testing.T) {
 		t.Errorf("Wrong result, %v", resultString)
 	}
 }
+
+func TestGET404(t *testing.T) {
+	expected := "Some text"
+	server := fakeServer(http.StatusNotFound, expected)
+	defer server.Close()
+
+	data, err := GET(server.URL)
+	defer data.Close()
+	if err == nil {
+		t.Error("There should be an error!")
+	}
+
+	result, errReading := ioutil.ReadAll(data)
+	if errReading != nil {
+		t.Error(err)
+	}
+
+	resultString := string(result)
+
+	if expected != resultString {
+		t.Errorf("Wrong result, %v", resultString)
+	}
+}