support $INPUT_REMAP_FILE

This commit is contained in:
Bel LaPointe
2023-03-02 15:33:52 -07:00
parent b0767818e2
commit 74e54c4efa
4 changed files with 88 additions and 1 deletions

View File

@@ -18,6 +18,37 @@ func TestNew(t *testing.T) {
}
}
func TestNewRemapped(t *testing.T) {
d := t.TempDir()
remap := path.Join(d, "remap")
if err := os.WriteFile(remap, []byte(`{"a":"b"}`), os.ModePerm); err != nil {
t.Fatal(err)
}
rand := path.Join(d, "rand")
if err := os.WriteFile(rand, []byte(`{"a":1}`), os.ModePerm); err != nil {
t.Fatal(err)
}
os.Setenv("INPUT_REMAP_FILE", remap)
os.Setenv("INPUT_RANDOM_WEIGHT_FILE", rand)
t.Cleanup(func() {
os.Unsetenv("INPUT_REMAP_FILE")
os.Unsetenv("INPUT_RANDOM_WEIGHT_FILE")
})
r := input.New(context.Background())
for i := 0; i < 15; i++ {
batch := r.Read()
for j := range batch {
if batch[j].Char != 'b' {
t.Errorf("%c", batch[j].Char)
}
t.Logf("[%d][%d] %c|%v", i, j, batch[j].Char, batch[j].Down)
}
}
}
func TestNewBuffered(t *testing.T) {
os.Setenv("INPUT_BUFFERED", "true")
t.Cleanup(func() {