64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
package fejkdata
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// newGenerator creates a generator over a single data directory, failing on
|
|
// error. Most tests load one dir; newGeneratorN loads several (last-loaded wins
|
|
// on conflicts).
|
|
func newGenerator(t *testing.T, dir string, opts ...Option) *Generator {
|
|
t.Helper()
|
|
return newGeneratorN(t, []string{dir}, opts...)
|
|
}
|
|
|
|
func newGeneratorN(t *testing.T, dirs []string, opts ...Option) *Generator {
|
|
t.Helper()
|
|
all := []Option{WithoutShippedData()}
|
|
for _, dir := range dirs {
|
|
all = append(all, WithDataPath(dir))
|
|
}
|
|
f, err := New(append(all, opts...)...)
|
|
if err != nil {
|
|
t.Fatalf("New(%q): %v", dirs, err)
|
|
}
|
|
return f
|
|
}
|
|
|
|
// fake generates a value, failing the test on error.
|
|
func fake(t *testing.T, f *Generator, path string) string {
|
|
t.Helper()
|
|
s, err := f.Fake(path)
|
|
if err != nil {
|
|
t.Fatalf("Fake(%q): %v", path, err)
|
|
}
|
|
return s
|
|
}
|
|
|
|
func TestNewMissingDirectory(t *testing.T) {
|
|
_, err := New(WithoutShippedData(), WithDataPath("data/de_DE"))
|
|
if err == nil || !strings.Contains(err.Error(), "de_DE") {
|
|
t.Fatalf("New(missing) error = %v, want it to name the path", err)
|
|
}
|
|
}
|
|
|
|
func TestWithSeedIsDeterministic(t *testing.T) {
|
|
a, b := newGenerator(t, "data/sv_SE", WithSeed(42)), newGenerator(t, "data/sv_SE", WithSeed(42))
|
|
for i := 0; i < 50; i++ {
|
|
if x, y := fake(t, a, "person"), fake(t, b, "person"); x != y {
|
|
t.Fatalf("same seed diverged at %d: %q != %q", i, x, y)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDifferentSeedsDiffer(t *testing.T) {
|
|
a, b := newGenerator(t, "data/en_US", WithSeed(1)), newGenerator(t, "data/en_US", WithSeed(2))
|
|
for i := 0; i < 50; i++ {
|
|
if fake(t, a, "person") != fake(t, b, "person") {
|
|
return
|
|
}
|
|
}
|
|
t.Fatal("seeds 1 and 2 produced identical sequences")
|
|
}
|