mirror of
https://github.com/safedep/pmg.git
synced 2026-08-03 07:24:09 +02:00
42 lines
941 B
Go
42 lines
941 B
Go
|
package proxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/safedep/pmg/config"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDaemonArgsPrependsChangedConfigFlags(t *testing.T) {
|
||
|
|
root := &cobra.Command{Use: "pmg"}
|
||
|
|
config.ApplyCobraFlags(root)
|
||
|
|
|
||
|
|
var got []string
|
||
|
|
start := &cobra.Command{
|
||
|
|
Use: "start",
|
||
|
|
Run: func(cmd *cobra.Command, _ []string) {
|
||
|
|
got = daemonArgs(cmd, "/tmp/proxy-state.json", "127.0.0.1", 9000)
|
||
|
|
},
|
||
|
|
}
|
||
|
|
proxyCmd := &cobra.Command{Use: "proxy"}
|
||
|
|
proxyCmd.AddCommand(start)
|
||
|
|
root.AddCommand(proxyCmd)
|
||
|
|
root.SetArgs([]string{
|
||
|
|
"--paranoid",
|
||
|
|
"--skip-dependency-cooldown",
|
||
|
|
"proxy", "start",
|
||
|
|
})
|
||
|
|
|
||
|
|
require.NoError(t, root.Execute())
|
||
|
|
assert.Equal(t, []string{
|
||
|
|
"--paranoid=true",
|
||
|
|
"--skip-dependency-cooldown=true",
|
||
|
|
"proxy", "start", "--foreground-internal",
|
||
|
|
"--state", "/tmp/proxy-state.json",
|
||
|
|
"--host", "127.0.0.1",
|
||
|
|
"--port", "9000",
|
||
|
|
}, got)
|
||
|
|
}
|