add flag testify.shuffle

suite/suite.go

@@ -3,10 +3,12 @@ package suite
 import (
 	"flag"
 	"fmt"
+	"math/rand"
 	"os"
 	"reflect"
 	"regexp"
 	"runtime/debug"
+	"strconv"
 	"sync"
 	"testing"
 	"time"
@@ -17,6 +19,7 @@ import (
 
 var allTestsFilter = func(_, _ string) (bool, error) { return true, nil }
 var matchMethod = flag.String("testify.m", "", "regular expression to select tests of the testify suite to run")
+var shuffle = flag.String("testify.shuffle", "off", "randomize the execution order of tests")
 
 // Suite is a basic testing suite with methods for storing and
 // retrieving the current *testing.T context.
@@ -199,6 +202,22 @@ func Run(t *testing.T, suite TestingSuite) {
 		}
 		tests = append(tests, test)
 	}
+	if *shuffle != "off" {
+		var n int64
+		var err error
+		if *shuffle == "on" {
+			n = time.Now().UnixNano()
+		} else {
+			n, err = strconv.ParseInt(*shuffle, 10, 64)
+			if err != nil {
+				fmt.Fprintln(os.Stderr, `testing: -shuffle should be "off", "on", or a valid integer:`, err)
+				return
+			}
+		}
+		fmt.Println("-testify.shuffle", n)
+		rng := rand.New(rand.NewSource(n))
+		rng.Shuffle(len(tests), func(i, j int) { tests[i], tests[j] = tests[j], tests[i] })
+	}
 	if suiteSetupDone {
 		defer func() {
 			if tearDownAllSuite, ok := suite.(TearDownAllSuite); ok {

suite/suite_test.go

@@ -617,3 +617,47 @@ func (s *FailfastSuite) Test_B_Passes() {
 	s.call("Test B Passes")
 	s.Require().True(true)
 }
+
+type ShuffleOrderSuite struct {
+	Suite
+	callOrder []string
+}
+
+func (s *ShuffleOrderSuite) call(method string) {
+	s.callOrder = append(s.callOrder, method)
+}
+
+func TestSuiteShuffleOrder(t *testing.T) {
+	Run(t, new(ShuffleOrderSuite))
+}
+
+func TestShuffleOrderSuiteShuffleOn(t *testing.T) {
+	// To test this with testify.shuffle on we launch it in its own process
+	cmd := exec.Command("go", "test", "-v", "-race", "-run", "TestSuiteShuffleOrder", "-testify.shuffle", "2")
+	var out bytes.Buffer
+	cmd.Stdout = &out
+	t.Logf("Running %s", strings.Join(cmd.Args, " "))
+	err := cmd.Run()
+	t.Log(out.String())
+	if err != nil {
+		t.Log(err)
+		t.Fail()
+	}
+}
+
+func (s *ShuffleOrderSuite) TearDownSuite() {
+	shuffle := flag.Lookup("testify.shuffle").Value.(flag.Getter).Get().(string)
+	if shuffle == "off" {
+		assert.Equal(s.T(), "Test A;Test B", strings.Join(s.callOrder, ";"))
+	} else {
+		assert.Equal(s.T(), "Test B;Test A", strings.Join(s.callOrder, ";"))
+	}
+}
+
+func (s *ShuffleOrderSuite) Test_A() {
+	s.call("Test A")
+}
+
+func (s *ShuffleOrderSuite) Test_B() {
+	s.call("Test B")
+}