1
1
mirror of https://github.com/wader/fq.git synced 2024-12-25 22:34:14 +03:00
fq/format/fuzz_test.go

82 lines
1.8 KiB
Go
Raw Normal View History

2021-08-20 16:10:16 +03:00
//go:build gofuzzbeta
2020-06-08 03:29:51 +03:00
package format_test
import (
"bytes"
"context"
"fmt"
_ "github.com/wader/fq/format/all"
"github.com/wader/fq/pkg/interp"
2020-06-08 03:29:51 +03:00
"io"
"io/ioutil"
"os"
"path/filepath"
"testing"
)
type fuzzTest struct {
b []byte
}
type fuzzTestOutput struct {
io.Writer
}
func (o fuzzTestOutput) Size() (int, int) { return 120, 25 }
func (o fuzzTestOutput) IsTerminal() bool { return false }
2021-11-01 14:57:55 +03:00
func (ft *fuzzTest) Stdin() io.Reader { return bytes.NewBuffer(ft.b) } // TODO: special file?
func (ft *fuzzTest) Stdout() interp.Output { return fuzzTestOutput{os.Stdout} }
func (ft *fuzzTest) Stderr() io.Writer { return os.Stderr }
func (ft *fuzzTest) InterruptChan() chan struct{} { return nil }
func (ft *fuzzTest) Environ() []string { return nil }
2020-06-08 03:29:51 +03:00
func (ft *fuzzTest) Args() []string {
return []string{}
}
func (ft *fuzzTest) ConfigDir() (string, error) { return "/config", nil }
func (ft *fuzzTest) Open(name string) (io.ReadSeeker, error) {
return nil, fmt.Errorf("%s: file not found", name)
}
func (ft *fuzzTest) Readline(prompt string, complete func(line string, pos int) (newLine []string, shared int)) (string, error) {
return "", io.EOF
}
func FuzzFormats(f *testing.F) {
2020-06-08 03:29:51 +03:00
filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
if filepath.Ext(path) == ".fqtest" {
return nil
}
if filepath.Base(filepath.Dir(path)) != "testdata" {
return nil
}
if st, err := os.Stat(path); err != nil || st.IsDir() {
return nil
}
b, err := ioutil.ReadFile(path)
if err != nil {
panic(err)
}
f.Add(b)
return nil
})
f.Fuzz(func(t *testing.T, b []byte) {
fz := &fuzzTest{b: b}
q, err := interp.New(fz, all.Registry)
if err != nil {
t.Fatal(err)
}
err = q.Main(context.Background(), fz.Stdout(), "dev")
if err != nil {
// TODO: expect error
t.Fatal(err)
}
})
}