mirror of
https://github.com/netbirdio/netbird.git
synced 2024-11-24 00:54:01 +01:00
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/netbirdio/netbird/client/internal"
|
|
)
|
|
|
|
var cliAddr string
|
|
|
|
func TestUpDaemon(t *testing.T) {
|
|
mgmAddr := startTestingServices(t)
|
|
|
|
tempDir := t.TempDir()
|
|
confPath := tempDir + "/config.json"
|
|
|
|
ctx := internal.CtxInitState(context.Background())
|
|
state := internal.CtxGetState(ctx)
|
|
|
|
_, cliLis := startClientDaemon(t, ctx, "http://"+mgmAddr, confPath)
|
|
|
|
cliAddr = cliLis.Addr().String()
|
|
|
|
daemonAddr = "tcp://" + cliAddr
|
|
rootCmd.SetArgs([]string{
|
|
"login",
|
|
"--daemon-addr", "tcp://" + cliAddr,
|
|
"--setup-key", "A2C8E62B-38F5-4553-B31E-DD66C696CEBB",
|
|
"--log-file", "",
|
|
})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Errorf("expected no error while running up command, got %v", err)
|
|
return
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
if status, err := state.Status(); err != nil && status != internal.StatusIdle {
|
|
t.Errorf("wrong status after login: %s, %v", internal.StatusIdle, err)
|
|
return
|
|
}
|
|
|
|
rootCmd.SetArgs([]string{
|
|
"up",
|
|
"--daemon-addr", "tcp://" + cliAddr,
|
|
"--log-file", "",
|
|
})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Errorf("expected no error while running up command, got %v", err)
|
|
return
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
if status, err := state.Status(); err != nil && status != internal.StatusConnected {
|
|
t.Errorf("wrong status after connect: %s, %v", status, err)
|
|
return
|
|
}
|
|
|
|
rootCmd.SetArgs([]string{
|
|
"status",
|
|
"--daemon-addr", "tcp://" + cliAddr,
|
|
})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Errorf("expected no error while running up command, got %v", err)
|
|
return
|
|
}
|
|
time.Sleep(time.Second * 3)
|
|
|
|
rootCmd.SetErr(nil)
|
|
rootCmd.SetArgs([]string{"down", "--daemon-addr", "tcp://" + cliAddr})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Errorf("expected no error while running up command, got %v", err)
|
|
return
|
|
}
|
|
// we can't check status here, because context already canceled
|
|
}
|