set more check

taikoxyz · Apr 23, 2024 · e314feb · e314feb
1 parent d357a05
commit e314feb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/bindings/encoding/custom_error.go b/bindings/encoding/custom_error.go
@@ -22,18 +22,12 @@ func TryParsingCustomError(originalError error) error {
 
 	for _, customErrors := range customErrorMaps {
 		for _, customError := range customErrors {
-			if strings.HasPrefix(customError.ID.Hex(), errData) {
+			if strings.HasPrefix(customError.ID.Hex(), errData) || strings.Contains(errData, customError.Name) {
 				return errors.New(customError.Name)
 			}
 		}
 	}
 
-	for _, hookCustomError := range AssignmentHookABI.Errors {
-		if strings.HasPrefix(hookCustomError.ID.Hex(), errData) {
-			return errors.New(hookCustomError.Name)
-		}
-	}
-
 	return originalError
 }
 

diff --git a/prover/proof_submitter/proof_contester_test.go b/prover/proof_submitter/proof_contester_test.go
@@ -11,7 +11,7 @@ import (
 )
 
 func (s *ProofSubmitterTestSuite) TestSubmitContestNoTransition() {
-	s.NotNil(
+	s.Error(
 		s.contester.SubmitContest(
 			context.Background(),
 			common.Big256,

diff --git a/prover/proof_submitter/transaction/sender_test.go b/prover/proof_submitter/transaction/sender_test.go
@@ -83,7 +83,7 @@ func (s *TransactionTestSuite) TestSendTxWithBackoff() {
 	l1HeadChild, err := s.RPCClient.L1.HeaderByNumber(context.Background(), new(big.Int).Sub(l1Head.Number, common.Big1))
 	s.Nil(err)
 	meta := &bindings.TaikoDataBlockMetadata{L1Height: l1HeadChild.Number.Uint64(), L1Hash: l1HeadChild.Hash()}
-	s.NotNil(s.sender.Send(
+	s.Error(s.sender.Send(
 		context.Background(),
 		&producer.ProofWithHeader{
 			Meta:    meta,