fix weak test code

taikoxyz · Jan 22, 2024 · 9808160 · 9808160
1 parent 3f8c3a4
commit 9808160
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/prover/prover_test.go b/prover/prover_test.go
@@ -260,6 +260,15 @@ func (s *ProverTestSuite) TestContestWrongBlocks() {
 		close(sink)
 	}()
 
+	// Contest the transition.
+	contestedSink := make(chan *bindings.TaikoL1ClientTransitionContested, 3)
+	contestedSub, err := s.p.rpc.TaikoL1.WatchTransitionContested(nil, contestedSink, nil)
+	s.Nil(err)
+	defer func() {
+		contestedSub.Unsubscribe()
+		close(contestedSink)
+	}()
+
 	s.Nil(s.p.proveOp())
 	proofWithHeader := <-s.p.proofGenerationCh
 	proofWithHeader.Opts.BlockHash = testutils.RandomHash()
@@ -271,15 +280,6 @@ func (s *ProverTestSuite) TestContestWrongBlocks() {
 	s.NotEqual(header.Hash(), common.BytesToHash(event.Tran.BlockHash[:]))
 	s.Equal(header.ParentHash, common.BytesToHash(event.Tran.ParentHash[:]))
 
-	// Contest the transition.
-	contestedSink := make(chan *bindings.TaikoL1ClientTransitionContested)
-	contestedSub, err := s.p.rpc.TaikoL1.WatchTransitionContested(nil, contestedSink, nil)
-	s.Nil(err)
-	defer func() {
-		contestedSub.Unsubscribe()
-		close(contestedSink)
-	}()
-
 	s.Greater(header.Number.Uint64(), uint64(0))
 	s.Nil(s.p.onTransitionProved(context.Background(), event))