fix tests

cosmos · Jan 7, 2022 · fac4cad · fac4cad
1 parent 6ad9160
commit fac4cad
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/x/distribution/client/testutil/cli_test.go b/x/distribution/client/testutil/cli_test.go
@@ -22,7 +22,5 @@ func TestGRPCQueryTestSuite(t *testing.T) {
 }
 
 func TestWithdrawAllSuite(t *testing.T) {
-	cfg1 := network.DefaultConfig()
-	cfg1.NumValidators = 2
-	suite.Run(t, NewWithdrawAllTestSuite(cfg1))
+	suite.Run(t, new(WithdrawAllTestSuite))
 }
diff --git a/x/distribution/client/testutil/withdraw_all_suite.go b/x/distribution/client/testutil/withdraw_all_suite.go
@@ -23,11 +23,10 @@ type WithdrawAllTestSuite struct {
 	network *network.Network
 }
 
-func NewWithdrawAllTestSuite(cfg network.Config) *IntegrationTestSuite {
-	return &IntegrationTestSuite{cfg: cfg}
-}
-
 func (s *WithdrawAllTestSuite) SetupTest() {
+	cfg := network.DefaultConfig()
+	cfg.NumValidators = 2
+	s.cfg = cfg
 	s.T().Log("setting up integration test suite")
 	network, err := network.New(s.T(), s.T().TempDir(), s.cfg)
 	s.Require().NoError(err)
@@ -37,6 +36,12 @@ func (s *WithdrawAllTestSuite) SetupTest() {
 	s.Require().NoError(err)
 }
 
+// TearDownTest cleans up the curret test network after _each_ test.
+func (s *WithdrawAllTestSuite) TearDownTest() {
+	s.T().Log("tearing down integration test suite")
+	s.network.Cleanup()
+}
+
 func (s *WithdrawAllTestSuite) TestNewWithdrawAllRewardsGenerateOnly() {
 	require := s.Require()
 	val := s.network.Validators[0]