Added tests to check large read guards

This commit is contained in:
Cecylia Bocovich 2019-05-10 15:39:05 -04:00
parent 1d76d3ca2e
commit 0842dad38e
2 changed files with 31 additions and 7 deletions

View file

@ -178,6 +178,16 @@ func TestBroker(t *testing.T) {
proxyAnswers(ctx, w, r)
So(w.Code, ShouldEqual, http.StatusBadRequest)
})
Convey("with error if the proxy writes too much data", func() {
data := bytes.NewReader(make([]byte, 100001, 100001))
r, err := http.NewRequest("POST", "snowflake.broker/answer", data)
r.Header.Set("X-Session-ID", "test")
So(err, ShouldBeNil)
proxyAnswers(ctx, w, r)
So(w.Code, ShouldEqual, http.StatusBadRequest)
})
})
})