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)
	}
}