From 7642c78eea162b843a9762ddf8d7062c9ed03e2b Mon Sep 17 00:00:00 2001
From: Ryo Nihei <nihei.dev@gmail.com>
Date: Sat, 17 Jul 2021 00:14:05 +0900
Subject: Add a line number to token error messages

---
 spec/lexer_test.go | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'spec/lexer_test.go')

diff --git a/spec/lexer_test.go b/spec/lexer_test.go
index a5de136..89553d0 100644
--- a/spec/lexer_test.go
+++ b/spec/lexer_test.go
@@ -3,6 +3,8 @@ package spec
 import (
 	"strings"
 	"testing"
+
+	verr "github.com/nihei9/vartan/error"
 )
 
 func TestLexer_Run(t *testing.T) {
@@ -155,8 +157,18 @@ bar // This is the fourth comment.
 					break
 				}
 			}
-			if err != tt.err {
-				t.Fatalf("unexpected error; want: %v, got: %v", tt.err, err)
+			if tt.err != nil {
+				synErr, ok := err.(*verr.SpecError)
+				if !ok {
+					t.Fatalf("unexpected error; want: %v, got: %v", tt.err, err)
+				}
+				if tt.err != synErr.Cause {
+					t.Fatalf("unexpected error; want: %v, got: %v", tt.err, synErr.Cause)
+				}
+			} else {
+				if err != nil {
+					t.Fatalf("unexpected error; want: %v, got: %v", tt.err, err)
+				}
 			}
 		})
 	}
-- 
cgit v1.2.3