helper/args: MustNewCheckedArgs for cleaner hardcoded args

Signed-off-by: Ophestra Umiker <cat@ophivana.moe>
This commit is contained in:
2024-10-07 13:33:18 +09:00
parent 9647eb6a6b
commit 0fb9e40191
4 changed files with 68 additions and 18 deletions

View File

@@ -6,7 +6,6 @@ import (
"os"
"strconv"
"strings"
"sync"
"testing"
"time"
@@ -23,20 +22,9 @@ var (
"--talk=org.freedesktop.UPower",
}
wantPayload string
argsWt io.WriterTo
argsOnce sync.Once
)
func prepareArgs() {
wantPayload = strings.Join(want, "\x00") + "\x00"
if a, err := helper.NewCheckedArgs(want); err != nil {
panic(err.Error())
} else {
argsWt = a
}
}
argsWt = helper.MustNewCheckedArgs(want)
)
func argF(argsFD int, _ int) []string {
return []string{"--args", strconv.Itoa(argsFD)}
@@ -48,7 +36,6 @@ func argFStatus(argsFD int, statFD int) []string {
func TestHelper_StartNotify_Close_Wait(t *testing.T) {
helper.InternalReplaceExecCommand(t)
argsOnce.Do(prepareArgs)
t.Run("start non-existent helper path", func(t *testing.T) {
h := helper.New(argsWt, "/nonexistent", argF)
@@ -143,7 +130,6 @@ func TestHelper_StartNotify_Close_Wait(t *testing.T) {
}
func TestHelper_Start_Close_Wait(t *testing.T) {
helper.InternalReplaceExecCommand(t)
argsOnce.Do(prepareArgs)
var wt io.WriterTo
if a, err := helper.NewCheckedArgs(want); err != nil {