diff options
Diffstat (limited to 'internal/api/fileserver')
| -rw-r--r-- | internal/api/fileserver/servefile_test.go | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/internal/api/fileserver/servefile_test.go b/internal/api/fileserver/servefile_test.go index c840d232f..cb3a35e45 100644 --- a/internal/api/fileserver/servefile_test.go +++ b/internal/api/fileserver/servefile_test.go @@ -19,7 +19,7 @@ package fileserver_test import ( "context" - "io/ioutil" + "io" "net/http" "net/http/httptest" "testing" @@ -28,6 +28,7 @@ import ( "github.com/superseriousbusiness/gotosocial/internal/api/fileserver" "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" "github.com/superseriousbusiness/gotosocial/internal/media" + "github.com/superseriousbusiness/gotosocial/internal/middleware" "github.com/superseriousbusiness/gotosocial/testrig" ) @@ -54,12 +55,15 @@ func (suite *ServeFileTestSuite) GetFile( ctx.AddParam(fileserver.MediaSizeKey, string(mediaSize)) ctx.AddParam(fileserver.FileNameKey, filename) + logger := middleware.Logger(false) suite.fileServer.ServeFile(ctx) + logger(ctx) + code = recorder.Code headers = recorder.Result().Header var err error - body, err = ioutil.ReadAll(recorder.Body) + body, err = io.ReadAll(recorder.Body) if err != nil { suite.FailNow(err.Error()) } |