[#129] Fix race condition tests

Signed-off-by: Denis Kirillov <denis@nspcc.ru>
nspcc-dev · Mar 11, 2022 · acd8bbb · acd8bbb
1 parent c452130
commit acd8bbb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/integration_test.go b/integration_test.go
@@ -1,6 +1,3 @@
-//go:build !race
-// +build !race
-
 package main
 
 import (

diff --git a/uploader/upload.go b/uploader/upload.go
@@ -136,7 +136,10 @@ func (u *Uploader) Upload(c *fasthttp.RequestCtx) {
 	obj.SetOwnerID(id)
 	obj.SetAttributes(attributes...)
 
-	if idObj, err = u.pool.PutObject(c, *obj, file, pool.WithBearer(bt)); err != nil {
+	ctx, cancel := context.WithCancel(c)
+	defer cancel()
+
+	if idObj, err = u.pool.PutObject(ctx, *obj, file, pool.WithBearer(bt)); err != nil {
 		log.Error("could not store file in neofs", zap.Error(err))
 		response.Error(c, "could not store file in neofs", fasthttp.StatusBadRequest)
 		return