2022-08-25 10:46:58 +08:00
|
|
|
//go:build windows
|
|
|
|
|
|
|
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"bytes"
|
|
|
|
|
"os"
|
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func TestWindowsFlagsAreSet(t *testing.T) {
|
|
|
|
|
expectedString := "test"
|
|
|
|
|
|
|
|
|
|
commands := []string{
|
|
|
|
|
"--service", expectedString,
|
|
|
|
|
"--service-name", expectedString,
|
|
|
|
|
"--service-display-name", expectedString,
|
|
|
|
|
"--service-restart-delay", expectedString,
|
|
|
|
|
"--service-auto-restart",
|
|
|
|
|
"--console",
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
buf := new(bytes.Buffer)
|
|
|
|
|
args := os.Args[0:1]
|
|
|
|
|
args = append(args, commands...)
|
|
|
|
|
m := NewMockTelegraf()
|
|
|
|
|
err := runApp(args, buf, NewMockServer(), NewMockConfig(buf), m)
|
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
|
|
require.Equal(t, expectedString, m.service)
|
|
|
|
|
require.Equal(t, expectedString, m.serviceName)
|
|
|
|
|
require.Equal(t, expectedString, m.serviceDisplayName)
|
|
|
|
|
require.Equal(t, expectedString, m.serviceRestartDelay)
|
2023-11-02 20:40:30 +08:00
|
|
|
require.True(t, m.serviceAutoRestart)
|
|
|
|
|
require.True(t, m.console)
|
2022-08-25 10:46:58 +08:00
|
|
|
}
|