30 lines
646 B
Go
30 lines
646 B
Go
package deploy
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestWantsDeferMiningFlag(t *testing.T) {
|
|
old := os.Args
|
|
defer func() { os.Args = old }()
|
|
os.Args = []string{"agent", "--run"}
|
|
if WantsDeferMining() {
|
|
t.Fatal("expected false without defer flag or env")
|
|
}
|
|
os.Args = []string{"agent", "--run", "--defer-mining"}
|
|
if !WantsDeferMining() {
|
|
t.Fatal("expected true with --defer-mining")
|
|
}
|
|
}
|
|
|
|
func TestRunFlagsIncludesDeferWhenSet(t *testing.T) {
|
|
old := os.Args
|
|
defer func() { os.Args = old }()
|
|
os.Args = []string{"agent", "--defer-mining"}
|
|
flags := RunFlags()
|
|
if flags != "--run --defer-mining" {
|
|
t.Fatalf("RunFlags = %q", flags)
|
|
}
|
|
}
|