diff --git a/generate/parse.go b/generate/parse.go index cb4a7a5..e6e1ca3 100644 --- a/generate/parse.go +++ b/generate/parse.go @@ -25,8 +25,8 @@ func defaultProtocol() Protocol { } } -func ParseReader(reader io.Reader) (*Protocol, error) { - lx, err := Lex("test.pdl", reader) +func ParseReader(fileName string, reader io.Reader) (*Protocol, error) { + lx, err := Lex(fileName, reader) if err != nil { return nil, err } return Parse(lx) } diff --git a/generate/parse_test.go b/generate/parse_test.go index 1916580..a447ebb 100644 --- a/generate/parse_test.go +++ b/generate/parse_test.go @@ -33,7 +33,7 @@ func TestParse(test *testing.T) { } test.Log("CORRECT:", &correct) - got, err := ParseReader(strings.NewReader(` + got, err := ParseReader("test.pdl", strings.NewReader(` M0000 Connect { 0000 Name String, 0001 Password String,