Merge pull request #38 from danielgtaylor/responses-unit-tests

chore: adding unit test for responses.ServeContent for codecov
This commit is contained in:
Drew Orban 2022-04-06 23:32:25 -07:00 committed by GitHub
commit 9d74b11de6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -96,3 +96,23 @@ func TestResponses(t *testing.T) {
String(http.StatusOK)
assert.Equal(t, 200, status)
}
func contains(r []huma.Response, e huma.Response) bool {
for _, i := range r {
if i.GetStatus() == e.GetStatus() {
return true
}
}
return false
}
func TestWriteContentResponses(t *testing.T) {
r := ServeContent()
assert.Equal(t, 5, len(r))
assert.True(t, contains(r, OK()))
assert.True(t, contains(r, PartialContent()))
assert.True(t, contains(r, NotModified()))
assert.True(t, contains(r, PreconditionFailed()))
assert.True(t, contains(r, InternalServerError()))
}