input Getenvs to FlagXYZ
This commit is contained in:
@@ -3,6 +3,7 @@ package input_test
|
||||
import (
|
||||
"context"
|
||||
"mayhem-party/src/device/input"
|
||||
"mayhem-party/src/device/input/raw"
|
||||
"mayhem-party/src/device/input/wrap"
|
||||
"os"
|
||||
"path"
|
||||
@@ -32,10 +33,10 @@ func TestNewRemapped(t *testing.T) {
|
||||
}
|
||||
|
||||
wrap.FlagRemapFile = remap
|
||||
os.Setenv("RAW_RANDOM_WEIGHT_FILE", rand)
|
||||
raw.FlagRawRandomWeightFile = rand
|
||||
t.Cleanup(func() {
|
||||
wrap.FlagRemapFile = ""
|
||||
os.Unsetenv("RAW_RANDOM_WEIGHT_FILE")
|
||||
raw.FlagRawRandomWeightFile = ""
|
||||
})
|
||||
|
||||
r := input.New(context.Background())
|
||||
@@ -72,9 +73,9 @@ func TestNewRandomWeightFile(t *testing.T) {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
os.Setenv("RAW_RANDOM_WEIGHT_FILE", p)
|
||||
raw.FlagRawRandomWeightFile = p
|
||||
t.Cleanup(func() {
|
||||
os.Unsetenv("RAW_RANDOM_WEIGHT_FILE")
|
||||
raw.FlagRawRandomWeightFile = ""
|
||||
})
|
||||
|
||||
r := input.New(context.Background())
|
||||
|
||||
Reference in New Issue
Block a user