Ejemplo n.º 1
0
func TestLineDirectives(t *testing.T) {
	const (
		m = "main.c"
		f = "foo.c"
		b = "bar.c"
	)
	tab := []struct {
		s, file   string
		line, col int
	}{
		{"",
			m, 1, 1},
		{" ",
			m, 1, 2},
		{"  ",
			m, 1, 3},
		{"\n",
			m, 2, 1},
		{"\n ",
			m, 2, 2},
		{"\n  ",
			m, 2, 3},
		{" \n",
			m, 2, 1},
		{" \n ",
			m, 2, 2},
		{" \n  ",
			m, 2, 3},
		{"\n\n",
			m, 3, 1},
		{"\n\n ",
			m, 3, 2},
		{"\n \n",
			m, 3, 1},
		{"\n \n ",
			m, 3, 2},
		{" \n\n",
			m, 3, 1},
		{" \n\n ",
			m, 3, 2},
		{" \n \n",
			m, 3, 1},
		{" \n \n ",
			m, 3, 2},
		{`# 123456 "foo.c"`,
			f, 123455, 17},
		{`# 123456 "foo.c"` + "\n",
			f, 123456, 1},
		{` # 123456 "foo.c"`,
			m, 1, 18},
		{` # 123456 "foo.c"` + "\n",
			m, 2, 1},
		{`
# 123456 "foo.c"
`,
			f, 123456, 1},
		{`
# 123456 "foo.c"

`,
			f, 123457, 1},
		{`

# 123456 "foo.c"

`,
			f, 123457, 1},
		{`

# 123456 "foo.c"


`,
			f, 123458, 1},
		{`

# 999 "bar.c"

# 123456 "foo.c"

`,
			f, 123457, 1},

		{`#123456"foo.c"` + "\n",
			f, 123456, 1},
		{`#123456"foo.c" ` + "\n",
			f, 123456, 1},
		{`#123456 "foo.c"` + "\n",
			f, 123456, 1},
		{`#123456 "foo.c" ` + "\n",
			f, 123456, 1},
		{`# 123456"foo.c"` + "\n",
			f, 123456, 1},
		{`# 123456"foo.c" ` + "\n",
			f, 123456, 1},
		{`# 123456 "foo.c"` + "\n",
			f, 123456, 1},
		{`# 123456 "foo.c" ` + "\n",
			f, 123456, 1},
		{`# 123456 "foo.c"what so ever` + "\n",
			f, 123456, 1},

		{`#x 1000 "foo.c"` + "\n",
			m, 2, 1},
		{`# x1001 "foo.c"` + "\n",
			m, 2, 1},
		{`# 1002x "foo.c"` + "\n",
			m, 2, 1},
		{`# 1003 x"foo.c"` + "\n",
			m, 2, 1},
		{`# 1004 foo.c"` + "\n",
			m, 2, 1},
		{`# 1005 "foo.c` + "\n",
			m, 2, 1},
		{`# 1006 "foo.c`,
			m, 1, 14},
		{`# 1007 "` + "\n",
			m, 2, 1},
		{`# 1008 "`,
			m, 1, 9},
		{`# 1009 ` + "\n",
			m, 2, 1},
		{`# 1010 `,
			m, 1, 8},
		{`# 1011` + "\n",
			m, 2, 1},
		{`# 1012`,
			m, 1, 7},
		{`# ` + "\n",
			m, 2, 1},
		{`# `,
			m, 1, 3},
		{`#` + "\n",
			m, 2, 1},
		{`#`,
			m, 1, 2},
	}
	for i, test := range tab {
		r := strings.NewReader(test.s)
		d := strutil.NewDict()
		l := newLexer(
			d,
			m,
			r,
			func(file string, line, col int, msg string, args ...interface{}) bool {
				t.Fatal(i, "unexpected")
				return false
			},
			nil,
		)

		for j := 0; ; j++ {
			c := l.scan()
			if c == 0 {
				break
			}
			if j > 100 {
				t.Fatal(i, j, "missed EOF")
			}
		}

		if l.file != d.Id(test.file) || l.line != test.line || l.col != test.col {
			t.Fatalf(
				`%d: (%q) got "%s.%d:%d", exp "%s.%d:%d"`,
				i, test.s,
				l.file, l.line, l.col,
				test.file, test.line, test.col,
			)
		}
	}
}
Ejemplo n.º 2
0
func TestScanPeekNext(t *testing.T) {
	const N = 17
	b := make([]byte, N)
	for i := range b {
		b[i] = byte(i + 1)
	}
	f := func(rf func() io.Reader) {
		for n := 0; n <= N; n++ {
			b = b[:n]
			for sz := 0; sz <= len(b)+2; sz++ {
				r := rf()
				d := strutil.NewDict()
				failed := false
				l := newLexer(
					d,
					"file",
					r,
					func(file string, line, col int, msg string, args ...interface{}) bool {
						failed = true
						return true
					},
					make([]byte, sz),
				)
				t.Log("----")
				for i := 1; i <= n+1; i++ {
					t.Logf("n %d sz %d i %d", n, sz, i)
					c := l.peek()
					t.Logf("peek == %d", c)
					switch {
					case failed:
						t.Fatal("unexpected")
					case c == 0:
						if g, e := i, n+1; g != e {
							t.Fatalf("got peek == 0 at %d, expceted at %d", g, e)
						}

						c = l.next(c)
						t.Logf("next == %d", c)
						if g, e := c, 0; g != e {
							t.Fatalf("got next == %d, expected %d", g, e)
						}
					case c > 0 && i > n:
						t.Fatalf("got peek > 0 (%d), expected 0", c)
					default:
						if g, e := c, i; g != i {
							t.Fatalf("got peek == %d, expected %d", g, e)
						}

						c = l.next(c)
						t.Logf("next == %d", c)
						switch {
						case failed:
							t.Fatal("unexpected")
						case c == 0:
							if g, e := i, n; g != e {
								t.Fatalf("got next == 0 at %d, expected at %d", g, e)
							}
						default:
							if i > n {
								t.Fatalf("got next > 0 (%d), expected 0", c)
							}
						}
					}
				}
			}
		}
	}
	f(func() io.Reader { return bytes.NewReader(b) })
	f(func() io.Reader { return strings.NewReader(string(b)) })
	f(func() io.Reader { return bufio.NewReader(bytes.NewReader(b)) })

}