Mikael Göransson
69 lines
1.9 KiB
Go
69 lines
1.9 KiB
Go
package fakes
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func benchFaker(b *testing.B, dir string) *Fakes {
|
|
b.Helper()
|
|
f, err := New([]string{dir}, WithSeed(1))
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
return f
|
|
}
|
|
|
|
func benchPath(b *testing.B, dir, path string) {
|
|
f := benchFaker(b, dir)
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
if _, err := f.Fake(path); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkPerson(b *testing.B) { benchPath(b, "data/sv_SE", "person") }
|
|
func BenchmarkAddress(b *testing.B) { benchPath(b, "data/sv_SE", "address") }
|
|
func BenchmarkWord(b *testing.B) { benchPath(b, "data/sv_SE", "word") }
|
|
func BenchmarkCreditcard(b *testing.B) { benchPath(b, "data/misc", "creditcard") }
|
|
func BenchmarkSSN(b *testing.B) { benchPath(b, "data/sv_SE", "ssn") }
|
|
func BenchmarkUUIDv7(b *testing.B) { benchPath(b, "data/misc", "uuid") }
|
|
|
|
func tmpData(b *testing.B, name, body string) string {
|
|
b.Helper()
|
|
dir := b.TempDir()
|
|
if err := os.WriteFile(filepath.Join(dir, name+".json"), []byte(body), 0o644); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
return dir
|
|
}
|
|
|
|
func BenchmarkCalc(b *testing.B) {
|
|
dir := tmpData(b, "inv", `{"format":"{net} x {qty} = {calc(net * qty, 2)}","net":["19.99"],"qty":["3"]}`)
|
|
benchPath(b, dir, "inv")
|
|
}
|
|
|
|
func BenchmarkLongLiteral(b *testing.B) {
|
|
dir := tmpData(b, "sql", `{"format":"INSERT INTO customers (id, name, city) V#ALUES (#1#2#3, '{word}', '{word}');","word":["alpha","beta","gamma","delta"]}`)
|
|
benchPath(b, dir, "sql")
|
|
}
|
|
|
|
func BenchmarkRepeat(b *testing.B) {
|
|
dir := tmpData(b, "many", `{"format":"{word}","repeat":20,"separator":", ","word":["alpha","beta","gamma","delta"]}`)
|
|
benchPath(b, dir, "many")
|
|
}
|
|
|
|
// BenchmarkNew measures load+compile+validate of the whole shipped tree.
|
|
func BenchmarkNew(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
if _, err := New([]string{"data"}, WithSeed(1)); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|