@@ -15,6 +15,7 @@ import (
1515
1616 "github.com/stretchr/testify/assert"
1717
18+ "github.com/sourcegraph/log/logtest"
1819 "github.com/sourcegraph/sourcegraph/internal/api"
1920 "github.com/sourcegraph/sourcegraph/internal/database"
2021 "github.com/sourcegraph/sourcegraph/internal/database/dbmocks"
@@ -77,6 +78,8 @@ func Test_serveRawWithHTTPRequestMethodHEAD(t *testing.T) {
7778 mockNewCommon = nil
7879 }()
7980
81+ logger := logtest .Scoped (t )
82+
8083 t .Run ("success response for HEAD request" , func (t * testing.T ) {
8184 // httptest server will return a 200 OK, so gitserver.Client.RepoInfo will not return
8285 // an error.
@@ -90,7 +93,7 @@ func Test_serveRawWithHTTPRequestMethodHEAD(t *testing.T) {
9093 db .ReposFunc .SetDefaultReturn (rstore )
9194 rstore .GetByNameFunc .SetDefaultReturn (& types.Repo {ID : 123 }, nil )
9295
93- err := serveRaw (db , gitserver .NewClient ())(w , req )
96+ err := serveRaw (logger , db , gitserver .NewClient ())(w , req )
9497 if err != nil {
9598 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
9699 }
@@ -113,7 +116,7 @@ func Test_serveRawWithHTTPRequestMethodHEAD(t *testing.T) {
113116 db .ReposFunc .SetDefaultReturn (rstore )
114117 rstore .GetByNameFunc .SetDefaultReturn (nil , & database.RepoNotFoundErr {ID : 123 })
115118
116- err := serveRaw (db , gitserver .NewClient ())(w , req )
119+ err := serveRaw (logger , db , gitserver .NewClient ())(w , req )
117120 if err == nil {
118121 t .Fatal ("Want error but got nil" )
119122 }
@@ -137,6 +140,8 @@ func Test_serveRawWithContentArchive(t *testing.T) {
137140 mockNewCommon = nil
138141 }()
139142
143+ logger := logtest .Scoped (t )
144+
140145 mockGitServerResponse := "this is a gitserver archive response"
141146
142147 t .Run ("success response for format=zip" , func (t * testing.T ) {
@@ -148,7 +153,7 @@ func Test_serveRawWithContentArchive(t *testing.T) {
148153 w := httptest .NewRecorder ()
149154
150155 db := dbmocks .NewMockDB ()
151- err := serveRaw (db , gitserver .NewClient ())(w , req )
156+ err := serveRaw (logger , db , gitserver .NewClient ())(w , req )
152157 if err != nil {
153158 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
154159 }
@@ -188,7 +193,7 @@ func Test_serveRawWithContentArchive(t *testing.T) {
188193 w := httptest .NewRecorder ()
189194
190195 db := dbmocks .NewMockDB ()
191- err := serveRaw (db , gitserver .NewClient ())(w , req )
196+ err := serveRaw (logger , db , gitserver .NewClient ())(w , req )
192197 if err != nil {
193198 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
194199 }
@@ -234,6 +239,8 @@ func Test_serveRawWithContentTypePlain(t *testing.T) {
234239 mockNewCommon = nil
235240 }()
236241
242+ logger := logtest .Scoped (t )
243+
237244 assertHeaders := func (w http.ResponseWriter ) {
238245 t .Helper ()
239246
@@ -264,7 +271,7 @@ func Test_serveRawWithContentTypePlain(t *testing.T) {
264271 w := httptest .NewRecorder ()
265272
266273 db := dbmocks .NewMockDB ()
267- err := serveRaw (db , gsClient )(w , req )
274+ err := serveRaw (logger , db , gsClient )(w , req )
268275 if err != nil {
269276 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
270277 }
@@ -294,7 +301,7 @@ func Test_serveRawWithContentTypePlain(t *testing.T) {
294301 w := httptest .NewRecorder ()
295302
296303 db := dbmocks .NewMockDB ()
297- err := serveRaw (db , gsClient )(w , req )
304+ err := serveRaw (logger , db , gsClient )(w , req )
298305 if err != nil {
299306 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
300307 }
@@ -327,7 +334,7 @@ c.go`
327334 req := httptest .NewRequest ("GET" , "/github.com/sourcegraph/sourcegraph/-/raw" , nil )
328335 w := httptest .NewRecorder ()
329336
330- err := serveRaw (dbmocks .NewMockDB (), gitserverClient )(w , req )
337+ err := serveRaw (logger , dbmocks .NewMockDB (), gitserverClient )(w , req )
331338 if err != nil {
332339 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
333340 }
@@ -360,7 +367,7 @@ c.go`
360367 req := httptest .NewRequest ("GET" , "/github.com/sourcegraph/sourcegraph/-/raw?format=exe" , nil )
361368 w := httptest .NewRecorder ()
362369
363- err := serveRaw (dbmocks .NewMockDB (), gitserverClient )(w , req )
370+ err := serveRaw (logger , dbmocks .NewMockDB (), gitserverClient )(w , req )
364371 if err != nil {
365372 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
366373 }
@@ -389,6 +396,7 @@ func Test_serveRawRepoCloning(t *testing.T) {
389396 t .Cleanup (func () {
390397 mockNewCommon = nil
391398 })
399+ logger := logtest .Scoped (t )
392400 // Fail git server calls, as they should not be invoked for a cloning repo.
393401 initHTTPTestGitServer (t , http .StatusInternalServerError , "{should not be invoked}" )
394402 gsClient := gitserver .NewMockClient ()
@@ -399,7 +407,7 @@ func Test_serveRawRepoCloning(t *testing.T) {
399407 db := dbmocks .NewMockDB ()
400408 // Former implementation would sleep awaiting repository to be available.
401409 // Await request to be served with a timeout by racing done channel with time.After.
402- err := serveRaw (db , gsClient )(w , req )
410+ err := serveRaw (logger , db , gsClient )(w , req )
403411 if err != nil {
404412 t .Fatalf ("Failed to invoke serveRaw: %v" , err )
405413 }
0 commit comments