mirror of
				https://github.com/go-gitea/gitea
				synced 2025-10-31 19:38:23 +00:00 
			
		
		
		
	Prevent race in TestPersistableChannelQueue (#16468)
* Prevent race in TestPersistableChannelQueue A slight race has become apparent in the TestPersistableChannelQueue. This PR simply adds locking to prevent the race. * make print value of "$(GOTESTFLAGS)" on test-backend and unit-test-coverage Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: 6543 <6543@obermui.de>
This commit is contained in:
		| @@ -6,6 +6,7 @@ package queue | ||||
|  | ||||
| import ( | ||||
| 	"io/ioutil" | ||||
| 	"sync" | ||||
| 	"testing" | ||||
|  | ||||
| 	"code.gitea.io/gitea/modules/util" | ||||
| @@ -22,6 +23,7 @@ func TestPersistableChannelQueue(t *testing.T) { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	lock := sync.Mutex{} | ||||
| 	queueShutdown := []func(){} | ||||
| 	queueTerminate := []func(){} | ||||
|  | ||||
| @@ -41,8 +43,12 @@ func TestPersistableChannelQueue(t *testing.T) { | ||||
| 	assert.NoError(t, err) | ||||
|  | ||||
| 	go queue.Run(func(shutdown func()) { | ||||
| 		lock.Lock() | ||||
| 		defer lock.Unlock() | ||||
| 		queueShutdown = append(queueShutdown, shutdown) | ||||
| 	}, func(terminate func()) { | ||||
| 		lock.Lock() | ||||
| 		defer lock.Unlock() | ||||
| 		queueTerminate = append(queueTerminate, terminate) | ||||
| 	}) | ||||
|  | ||||
| @@ -69,7 +75,11 @@ func TestPersistableChannelQueue(t *testing.T) { | ||||
| 	assert.Error(t, err) | ||||
|  | ||||
| 	// Now shutdown the queue | ||||
| 	for _, callback := range queueShutdown { | ||||
| 	lock.Lock() | ||||
| 	callbacks := make([]func(), len(queueShutdown)) | ||||
| 	copy(callbacks, queueShutdown) | ||||
| 	lock.Unlock() | ||||
| 	for _, callback := range callbacks { | ||||
| 		callback() | ||||
| 	} | ||||
|  | ||||
| @@ -87,7 +97,11 @@ func TestPersistableChannelQueue(t *testing.T) { | ||||
| 	} | ||||
|  | ||||
| 	// terminate the queue | ||||
| 	for _, callback := range queueTerminate { | ||||
| 	lock.Lock() | ||||
| 	callbacks = make([]func(), len(queueTerminate)) | ||||
| 	copy(callbacks, queueTerminate) | ||||
| 	lock.Unlock() | ||||
| 	for _, callback := range callbacks { | ||||
| 		callback() | ||||
| 	} | ||||
|  | ||||
| @@ -110,8 +124,12 @@ func TestPersistableChannelQueue(t *testing.T) { | ||||
| 	assert.NoError(t, err) | ||||
|  | ||||
| 	go queue.Run(func(shutdown func()) { | ||||
| 		lock.Lock() | ||||
| 		defer lock.Unlock() | ||||
| 		queueShutdown = append(queueShutdown, shutdown) | ||||
| 	}, func(terminate func()) { | ||||
| 		lock.Lock() | ||||
| 		defer lock.Unlock() | ||||
| 		queueTerminate = append(queueTerminate, terminate) | ||||
| 	}) | ||||
|  | ||||
| @@ -122,10 +140,19 @@ func TestPersistableChannelQueue(t *testing.T) { | ||||
| 	result4 := <-handleChan | ||||
| 	assert.Equal(t, test2.TestString, result4.TestString) | ||||
| 	assert.Equal(t, test2.TestInt, result4.TestInt) | ||||
| 	for _, callback := range queueShutdown { | ||||
|  | ||||
| 	lock.Lock() | ||||
| 	callbacks = make([]func(), len(queueShutdown)) | ||||
| 	copy(callbacks, queueShutdown) | ||||
| 	lock.Unlock() | ||||
| 	for _, callback := range callbacks { | ||||
| 		callback() | ||||
| 	} | ||||
| 	for _, callback := range queueTerminate { | ||||
| 	lock.Lock() | ||||
| 	callbacks = make([]func(), len(queueTerminate)) | ||||
| 	copy(callbacks, queueTerminate) | ||||
| 	lock.Unlock() | ||||
| 	for _, callback := range callbacks { | ||||
| 		callback() | ||||
| 	} | ||||
|  | ||||
|   | ||||
		Reference in New Issue
	
	Block a user