mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 03:11:02 +01:00
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/netbirdio/netbird/iface"
|
|
)
|
|
|
|
func TestInitCommands(t *testing.T) {
|
|
helpFlag := "-h"
|
|
commandArgs := [][]string{{"root", helpFlag}}
|
|
for _, command := range rootCmd.Commands() {
|
|
commandArgs = append(commandArgs, []string{command.Name(), command.Name(), helpFlag})
|
|
for _, subcommand := range command.Commands() {
|
|
commandArgs = append(commandArgs, []string{command.Name() + " " + subcommand.Name(), command.Name(), subcommand.Name(), helpFlag})
|
|
}
|
|
}
|
|
|
|
for _, args := range commandArgs {
|
|
t.Run(fmt.Sprintf("Testing Command %s", args[0]), func(t *testing.T) {
|
|
defer func() {
|
|
err := recover()
|
|
if err != nil {
|
|
t.Fatalf("got an panic error while running the command: %s -h. Error: %s", args[0], err)
|
|
}
|
|
}()
|
|
|
|
rootCmd.SetArgs(args[1:])
|
|
rootCmd.SetOut(io.Discard)
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Errorf("expected no error while running %s command, got %v", args[0], err)
|
|
return
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSetFlagsFromEnvVars(t *testing.T) {
|
|
var cmd = &cobra.Command{
|
|
Use: "netbird",
|
|
Long: "test",
|
|
SilenceUsage: true,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
SetFlagsFromEnvVars(cmd)
|
|
},
|
|
}
|
|
|
|
cmd.PersistentFlags().StringSliceVar(&natExternalIPs, externalIPMapFlag, nil,
|
|
`comma separated list of external IPs to map to the Wireguard interface`)
|
|
cmd.PersistentFlags().StringVar(&interfaceName, interfaceNameFlag, iface.WgInterfaceDefault, "Wireguard interface name")
|
|
cmd.PersistentFlags().BoolVar(&rosenpassEnabled, enableRosenpassFlag, false, "Enable Rosenpass feature Rosenpass.")
|
|
cmd.PersistentFlags().Uint16Var(&wireguardPort, wireguardPortFlag, iface.DefaultWgPort, "Wireguard interface listening port")
|
|
|
|
t.Setenv("NB_EXTERNAL_IP_MAP", "abc,dec")
|
|
t.Setenv("NB_INTERFACE_NAME", "test-name")
|
|
t.Setenv("NB_ENABLE_ROSENPASS", "true")
|
|
t.Setenv("NB_WIREGUARD_PORT", "10000")
|
|
err := cmd.Execute()
|
|
if err != nil {
|
|
t.Fatalf("expected no error while running netbird command, got %v", err)
|
|
}
|
|
if len(natExternalIPs) != 2 {
|
|
t.Errorf("expected 2 external ips, got %d", len(natExternalIPs))
|
|
}
|
|
if natExternalIPs[0] != "abc" || natExternalIPs[1] != "dec" {
|
|
t.Errorf("expected abc,dec, got %s,%s", natExternalIPs[0], natExternalIPs[1])
|
|
}
|
|
if interfaceName != "test-name" {
|
|
t.Errorf("expected test-name, got %s", interfaceName)
|
|
}
|
|
if !rosenpassEnabled {
|
|
t.Errorf("expected rosenpassEnabled to be true, got false")
|
|
}
|
|
if wireguardPort != 10000 {
|
|
t.Errorf("expected wireguardPort to be 10000, got %d", wireguardPort)
|
|
}
|
|
}
|