mirror of
https://github.com/wader/fq.git
synced 2024-12-27 15:42:07 +03:00
b08ef00dd1
Replaces []Format with a Group type. A bit more type safe. Breaking change for RegisterFormat, now takes a first argument that is a "single" format group. Lots of naming cleanup. This is also preparation for decode group argument which will enable doing intresting probing, ex a format decoder could know it's decode as part of probe group (html could be probed possibly), or have "arg probe" group for decoder who inspect args to know if they should probe (-d /path/to/schema etc) to enable nice CLI-ergonomics.
135 lines
2.9 KiB
Go
135 lines
2.9 KiB
Go
package format_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
_ "github.com/wader/fq/format/all"
|
|
"github.com/wader/fq/pkg/decode"
|
|
"github.com/wader/fq/pkg/interp"
|
|
)
|
|
|
|
type fuzzFS struct{}
|
|
|
|
func (fuzzFS) Open(name string) (fs.File, error) {
|
|
return nil, fmt.Errorf("%s: file not found", name)
|
|
}
|
|
|
|
type fuzzTest struct {
|
|
b []byte
|
|
f *decode.Format
|
|
}
|
|
|
|
type fuzzTestInput struct {
|
|
interp.FileReader
|
|
io.Writer
|
|
}
|
|
|
|
func (fuzzTestInput) IsTerminal() bool { return false }
|
|
func (fuzzTestInput) Size() (int, int) { return 120, 25 }
|
|
|
|
type fuzzTestOutput struct {
|
|
io.Writer
|
|
}
|
|
|
|
func (o fuzzTestOutput) Size() (int, int) { return 120, 25 }
|
|
func (o fuzzTestOutput) IsTerminal() bool { return false }
|
|
|
|
func (ft *fuzzTest) Platform() interp.Platform { return interp.Platform{} }
|
|
func (ft *fuzzTest) Stdin() interp.Input {
|
|
return fuzzTestInput{FileReader: interp.FileReader{R: bytes.NewBuffer(ft.b)}}
|
|
}
|
|
func (ft *fuzzTest) Stdout() interp.Output { return fuzzTestOutput{io.Discard} }
|
|
func (ft *fuzzTest) Stderr() interp.Output { return fuzzTestOutput{io.Discard} }
|
|
func (ft *fuzzTest) InterruptChan() chan struct{} { return nil }
|
|
func (ft *fuzzTest) Environ() []string { return nil }
|
|
func (ft *fuzzTest) Args() []string {
|
|
return []string{
|
|
`fq`,
|
|
`-d`, ft.f.Name,
|
|
`.`,
|
|
}
|
|
}
|
|
func (ft *fuzzTest) ConfigDir() (string, error) { return "/config", nil }
|
|
func (ft *fuzzTest) FS() fs.FS { return fuzzFS{} }
|
|
func (ft *fuzzTest) History() ([]string, error) { return nil, nil }
|
|
|
|
func (ft *fuzzTest) Readline(opts interp.ReadlineOpts) (string, error) {
|
|
return "", io.EOF
|
|
}
|
|
|
|
func FuzzFormats(f *testing.F) {
|
|
if os.Getenv("FUZZTEST") == "" {
|
|
f.Skip("run with FUZZTEST=1 to fuzz")
|
|
}
|
|
|
|
i := 0
|
|
|
|
if err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
|
|
if filepath.Base(path) != "testdata" {
|
|
return nil
|
|
}
|
|
|
|
if err := filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
|
|
if filepath.Ext(path) == ".fqtest" {
|
|
return nil
|
|
}
|
|
if st, err := os.Stat(path); err != nil || st.IsDir() {
|
|
return err
|
|
}
|
|
|
|
b, readErr := os.ReadFile(path)
|
|
if readErr != nil {
|
|
f.Fatal(err)
|
|
}
|
|
|
|
f.Logf("seed#%d %s", i, path)
|
|
f.Add(b)
|
|
i++
|
|
|
|
return nil
|
|
}); err != nil {
|
|
f.Fatal(f)
|
|
}
|
|
return nil
|
|
}); err != nil {
|
|
f.Fatal(f)
|
|
}
|
|
|
|
fi := 0
|
|
var g *decode.Group
|
|
|
|
if n := os.Getenv("GROUP"); n != "" {
|
|
var err error
|
|
g, err = interp.DefaultRegistry.Group(n)
|
|
if err != nil {
|
|
f.Fatal(err)
|
|
}
|
|
f.Logf("GROUP=%s", n)
|
|
} else {
|
|
g = interp.DefaultRegistry.MustAll()
|
|
}
|
|
|
|
f.Fuzz(func(t *testing.T, b []byte) {
|
|
fz := &fuzzTest{b: b, f: g.Formats[fi]}
|
|
q, err := interp.New(fz, interp.DefaultRegistry)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_ = q.Main(context.Background(), fz.Stdout(), "fuzz")
|
|
// if err != nil {
|
|
// // TODO: expect error
|
|
// t.Fatal(err)
|
|
// }
|
|
|
|
fi = (fi + 1) % len(g.Formats)
|
|
})
|
|
}
|