can exit while reading from keyboard

This commit is contained in:
Bel LaPointe
2023-03-02 15:21:34 -07:00
parent 5c5a371f55
commit b0767818e2
6 changed files with 22 additions and 11 deletions

View File

@@ -1,6 +1,7 @@
package input_test
import (
"context"
"mayhem-party/src/device/input"
"os"
"path"
@@ -8,7 +9,7 @@ import (
)
func TestNew(t *testing.T) {
r := input.New()
r := input.New(context.Background())
for i := 0; i < 15; i++ {
batch := r.Read()
for _, got := range batch {
@@ -23,7 +24,7 @@ func TestNewBuffered(t *testing.T) {
os.Unsetenv("INPUT_BUFFERED")
})
r := input.New()
r := input.New(context.Background())
for i := 0; i < 15; i++ {
batch := r.Read()
for j := range batch {
@@ -44,7 +45,7 @@ func TestNewRandomWeightFile(t *testing.T) {
os.Unsetenv("INPUT_RANDOM_WEIGHT_FILE")
})
r := input.New()
r := input.New(context.Background())
for i := 0; i < 15; i++ {
batch := r.Read()
for _, got := range batch {