improve test

middleware/jwt/jwt_test.go

@@ -64,11 +64,12 @@ func TestHmacSignVerify(t *testing.T) {
 
 func TestEd25519Verify(t *testing.T) {
 	tests := []struct {
-		name        string
-		claims      *Claims
-		certFile    string
-		keyFile     string
-		expectError bool
+		name              string
+		claims            *Claims
+		certFile          string
+		keyFile           string
+		expectedErr       error
+		expectedCustomErr interface{}
 	}{
 		{
 			name:     "valid token",
@@ -77,11 +78,11 @@ func TestEd25519Verify(t *testing.T) {
 			keyFile:  "testdata/key.pem",
 		},
 		{
-			name:        "invalid token",
-			claims:      &Claims{},
-			certFile:    "testdata/certs_invalid.crt",
-			keyFile:     "testdata/key_invalid.pem",
-			expectError: true,
+			name:              "invalid token",
+			claims:            &Claims{},
+			certFile:          "testdata/certs_invalid.crt",
+			keyFile:           "testdata/key_invalid.pem",
+			expectedCustomErr: x509.UnknownAuthorityError{},
 		},
 	}
 
@@ -93,14 +94,16 @@ func TestEd25519Verify(t *testing.T) {
 			}
 
 			_, err = VerifyToken(token)
-			if tt.expectError {
-				if errors.Is(err, x509.UnknownAuthorityError{}) {
-					t.Error("Token verification failed: expected error not returned")
-				}
-			} else {
-				if err != nil {
-					t.Errorf("Token verification failed: %v", err)
-				}
+			if tt.expectedErr != nil && !errors.Is(err, tt.expectedErr) {
+				t.Errorf("Token verification failed: expected error %v, got %v", tt.expectedErr, err)
+				return
+			}
+			if tt.expectedCustomErr != nil && !errors.As(err, &tt.expectedCustomErr) {
+				t.Errorf("Token verification failed: expected error type %T, got %v", tt.expectedCustomErr, err)
+				return
+			}
+			if tt.expectedErr == nil && tt.expectedCustomErr == nil && err != nil {
+				t.Errorf("Token verification failed: %v", err)
 			}
 		})
 	}