fix: tests.

corazawaf · Oct 7, 2024 · 2398c1a · 2398c1a
1 parent 660f381
commit 2398c1a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/internal/seclang/rule_parser_test.go b/internal/seclang/rule_parser_test.go
@@ -4,7 +4,6 @@
 package seclang
 
 import (
-	"errors"
 	"reflect"
 	"strings"
 	"testing"
@@ -111,19 +110,20 @@ func TestSecRuleUpdateTargetVariableNegation(t *testing.T) {
 		SecRule REQUEST_URI|REQUEST_COOKIES "abc" "id:8,phase:2"
 		SecRuleUpdateTargetById 8 "!REQUEST_HEADERS:"
 	`)
-	expectedErr := errors.New("unknown variable")
-	if errors.Unwrap(err).Error() != expectedErr.Error() {
-		t.Fatalf("unexpexted error, want %q, have %q", expectedErr, errors.Unwrap(err).Error())
+
+	expectedErrMsg := "unknown variable"
+	if !strings.Contains(err.Error(), expectedErrMsg) {
+		t.Fatalf("unexpexted error, want to contain %q, have %q", expectedErrMsg, err)
 	}
 
 	// Try to update undefined rule
 	err = p.FromString(`
 		SecRule REQUEST_URI|REQUEST_COOKIES "abc" "id:9,phase:2"
 		SecRuleUpdateTargetById 99 "!REQUEST_HEADERS:xyz"
 	`)
-	expectedErr = errors.New("SecRuleUpdateTargetById: rule \"99\" not found")
-	if errors.Unwrap(err).Error() != expectedErr.Error() {
-		t.Fatalf("unexpected error, want %q, have %q", expectedErr, errors.Unwrap(err).Error())
+	expectedErrMsg = "SecRuleUpdateTargetById: rule \"99\" not found"
+	if !strings.Contains(err.Error(), expectedErrMsg) {
+		t.Fatalf("unexpexted error, want to contain %q, have %q", expectedErrMsg, err)
 	}
 }