// Copyright 2017 syzkaller project authors. All rights reserved.
|
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
|
|
package ast
|
|
import (
|
"bufio"
|
"bytes"
|
"io/ioutil"
|
"path/filepath"
|
"regexp"
|
"strings"
|
"testing"
|
)
|
|
type ErrorMatcher struct {
|
Data []byte
|
expect []*errorDesc
|
got []*errorDesc
|
}
|
|
type errorDesc struct {
|
file string
|
line int
|
col int
|
text string
|
matched bool
|
}
|
|
func NewErrorMatcher(t *testing.T, file string) *ErrorMatcher {
|
data, err := ioutil.ReadFile(file)
|
if err != nil {
|
t.Fatalf("failed to open input file: %v", err)
|
}
|
var stripped []byte
|
var errors []*errorDesc
|
s := bufio.NewScanner(bytes.NewReader(data))
|
for i := 1; s.Scan(); i++ {
|
ln := s.Bytes()
|
for {
|
pos := bytes.LastIndex(ln, []byte("###"))
|
if pos == -1 {
|
break
|
}
|
errors = append(errors, &errorDesc{
|
file: filepath.Base(file),
|
line: i,
|
text: strings.TrimSpace(string(ln[pos+3:])),
|
})
|
ln = ln[:pos]
|
}
|
stripped = append(stripped, ln...)
|
stripped = append(stripped, '\n')
|
}
|
if err := s.Err(); err != nil {
|
t.Fatalf("failed to scan input file: %v", err)
|
}
|
return &ErrorMatcher{
|
Data: stripped,
|
expect: errors,
|
}
|
}
|
|
var errorLocationRe = regexp.MustCompile(`at [a-z][a-z0-9]+\.txt:[0-9]+:[0-9]+`)
|
|
func (em *ErrorMatcher) ErrorHandler(pos Pos, msg string) {
|
if match := errorLocationRe.FindStringSubmatchIndex(msg); match != nil {
|
msg = msg[0:match[0]] + "at LOCATION" + msg[match[1]:]
|
}
|
em.got = append(em.got, &errorDesc{
|
file: pos.File,
|
line: pos.Line,
|
col: pos.Col,
|
text: msg,
|
})
|
}
|
|
func (em *ErrorMatcher) Count() int {
|
return len(em.got)
|
}
|
|
func (em *ErrorMatcher) Check(t *testing.T) {
|
nextErr:
|
for _, e := range em.got {
|
for _, want := range em.expect {
|
if want.matched || want.line != e.line || want.text != e.text {
|
continue
|
}
|
want.matched = true
|
continue nextErr
|
}
|
t.Errorf("unexpected error:\n%v:%v:%v: %v", e.file, e.line, e.col, e.text)
|
}
|
for _, want := range em.expect {
|
if want.matched {
|
continue
|
}
|
t.Errorf("unmatched error:\n%v:%v: %v", want.file, want.line, want.text)
|
}
|
}
|
|
func (em *ErrorMatcher) DumpErrors(t *testing.T) {
|
for _, e := range em.got {
|
t.Logf("%v:%v:%v: %v", e.file, e.line, e.col, e.text)
|
}
|
}
|
