fix data race in the suit

Signed-off-by: Weizhen Wang <[email protected]>

suite/suite.go

@@ -7,6 +7,7 @@ import (
  "reflect"
  "regexp"
  "runtime/debug"
+ "sync"
  "testing"
  "time"
 
@@ -21,24 +22,31 @@ var matchMethod = flag.String("testify.m", "", "regular expression to select tes
 // retrieving the current *testing.T context.
 type Suite struct {
  *assert.Assertions
+ mu sync.Mutex
  require *require.Assertions
  t *testing.T
 }
 
 // T retrieves the current *testing.T context.
 func (suite *Suite) T() *testing.T {
+ suite.mu.Lock()
+ defer suite.mu.Unlock()
  return suite.t
 }
 
 // SetT sets the current *testing.T context.
 func (suite *Suite) SetT(t *testing.T) {
+ suite.mu.Lock()
+ defer suite.mu.Unlock()
  suite.t = t
  suite.Assertions = assert.New(t)
  suite.require = require.New(t)
 }
 
 // Require returns a require context for suite.
 func (suite *Suite) Require() *require.Assertions {
+ suite.mu.Lock()
+ defer suite.mu.Unlock()
  if suite.require == nil {
  suite.require = require.New(suite.T())
  }
@@ -51,6 +59,8 @@ func (suite *Suite) Require() *require.Assertions {
 // assert.Assertions with require.Assertions), this method is provided so you
 // can call `suite.Assert().NoError()`.
 func (suite *Suite) Assert() *assert.Assertions {
+ suite.mu.Lock()
+ defer suite.mu.Unlock()
  if suite.Assertions == nil {
  suite.Assertions = assert.New(suite.T())
  }