瀏覽代碼

Added TestGET404

Piotr Czajkowski 4 年之前
父節點
當前提交
a4d68f7990
共有 1 個文件被更改,包括 23 次插入0 次删除
  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)
+	}
+}