mirror of
https://github.com/netbirdio/netbird.git
synced 2025-01-11 16:38:27 +01:00
Call start services function for tests (#257)
* Call start services function for tests when testing CMDs we were using some global variables which got replaced by parallel test Now we will call a single function independently for each test
This commit is contained in:
parent
605ca03519
commit
612ef98f03
@ -2,35 +2,17 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/wiretrustee/wiretrustee/client/internal"
|
"github.com/wiretrustee/wiretrustee/client/internal"
|
||||||
"github.com/wiretrustee/wiretrustee/iface"
|
"github.com/wiretrustee/wiretrustee/iface"
|
||||||
mgmt "github.com/wiretrustee/wiretrustee/management/server"
|
|
||||||
"github.com/wiretrustee/wiretrustee/util"
|
"github.com/wiretrustee/wiretrustee/util"
|
||||||
)
|
)
|
||||||
|
|
||||||
var mgmAddr string
|
|
||||||
|
|
||||||
func TestLogin_Start(t *testing.T) {
|
|
||||||
config := &mgmt.Config{}
|
|
||||||
_, err := util.ReadJson("../testdata/management.json", config)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
testDir := t.TempDir()
|
|
||||||
config.Datadir = testDir
|
|
||||||
err = util.CopyFileContents("../testdata/store.json", filepath.Join(testDir, "store.json"))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
_, listener := startManagement(t, config)
|
|
||||||
mgmAddr = listener.Addr().String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLogin(t *testing.T) {
|
func TestLogin(t *testing.T) {
|
||||||
|
mgmAddr := startTestingServices(t)
|
||||||
|
|
||||||
tempDir := t.TempDir()
|
tempDir := t.TempDir()
|
||||||
confPath := tempDir + "/config.json"
|
confPath := tempDir + "/config.json"
|
||||||
mgmtURL := fmt.Sprintf("http://%s", mgmAddr)
|
mgmtURL := fmt.Sprintf("http://%s", mgmAddr)
|
||||||
|
@ -2,7 +2,9 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"github.com/wiretrustee/wiretrustee/util"
|
||||||
"net"
|
"net"
|
||||||
|
"path/filepath"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -15,6 +17,28 @@ import (
|
|||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func startTestingServices(t *testing.T) string {
|
||||||
|
config := &mgmt.Config{}
|
||||||
|
_, err := util.ReadJson("../testdata/management.json", config)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
testDir := t.TempDir()
|
||||||
|
config.Datadir = testDir
|
||||||
|
err = util.CopyFileContents("../testdata/store.json", filepath.Join(testDir, "store.json"))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, signalLis := startSignal(t)
|
||||||
|
signalAddr := signalLis.Addr().String()
|
||||||
|
config.Signal.URI = signalAddr
|
||||||
|
|
||||||
|
_, mgmLis := startManagement(t, config)
|
||||||
|
mgmAddr := mgmLis.Addr().String()
|
||||||
|
return mgmAddr
|
||||||
|
}
|
||||||
|
|
||||||
func startSignal(t *testing.T) (*grpc.Server, net.Listener) {
|
func startSignal(t *testing.T) (*grpc.Server, net.Listener) {
|
||||||
lis, err := net.Listen("tcp", ":0")
|
lis, err := net.Listen("tcp", ":0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -2,37 +2,15 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/wiretrustee/wiretrustee/client/internal"
|
"github.com/wiretrustee/wiretrustee/client/internal"
|
||||||
mgmt "github.com/wiretrustee/wiretrustee/management/server"
|
|
||||||
"github.com/wiretrustee/wiretrustee/util"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestUpDaemon_Start(t *testing.T) {
|
|
||||||
config := &mgmt.Config{}
|
|
||||||
_, err := util.ReadJson("../testdata/management.json", config)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
testDir := t.TempDir()
|
|
||||||
config.Datadir = testDir
|
|
||||||
err = util.CopyFileContents("../testdata/store.json", filepath.Join(testDir, "store.json"))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, signalLis := startSignal(t)
|
|
||||||
signalAddr = signalLis.Addr().String()
|
|
||||||
config.Signal.URI = signalAddr
|
|
||||||
|
|
||||||
_, mgmLis := startManagement(t, config)
|
|
||||||
mgmAddr = mgmLis.Addr().String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUpDaemon(t *testing.T) {
|
func TestUpDaemon(t *testing.T) {
|
||||||
|
mgmAddr := startTestingServices(t)
|
||||||
|
|
||||||
tempDir := t.TempDir()
|
tempDir := t.TempDir()
|
||||||
confPath := tempDir + "/config.json"
|
confPath := tempDir + "/config.json"
|
||||||
|
|
||||||
|
@ -2,42 +2,20 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"net/url"
|
"net/url"
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/wiretrustee/wiretrustee/iface"
|
"github.com/wiretrustee/wiretrustee/iface"
|
||||||
mgmt "github.com/wiretrustee/wiretrustee/management/server"
|
|
||||||
"github.com/wiretrustee/wiretrustee/util"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
signalAddr string
|
//signalAddr string
|
||||||
cliAddr string
|
cliAddr string
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestUp_Start(t *testing.T) {
|
|
||||||
config := &mgmt.Config{}
|
|
||||||
_, err := util.ReadJson("../testdata/management.json", config)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
testDir := t.TempDir()
|
|
||||||
config.Datadir = testDir
|
|
||||||
err = util.CopyFileContents("../testdata/store.json", filepath.Join(testDir, "store.json"))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, signalLis := startSignal(t)
|
|
||||||
signalAddr = signalLis.Addr().String()
|
|
||||||
config.Signal.URI = signalAddr
|
|
||||||
|
|
||||||
_, mgmLis := startManagement(t, config)
|
|
||||||
mgmAddr = mgmLis.Addr().String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUp(t *testing.T) {
|
func TestUp(t *testing.T) {
|
||||||
|
mgmAddr := startTestingServices(t)
|
||||||
|
|
||||||
tempDir := t.TempDir()
|
tempDir := t.TempDir()
|
||||||
confPath := tempDir + "/config.json"
|
confPath := tempDir + "/config.json"
|
||||||
mgmtURL, err := url.Parse("http://" + mgmAddr)
|
mgmtURL, err := url.Parse("http://" + mgmAddr)
|
||||||
@ -66,7 +44,7 @@ func TestUp(t *testing.T) {
|
|||||||
}()
|
}()
|
||||||
time.Sleep(time.Second * 2)
|
time.Sleep(time.Second * 2)
|
||||||
|
|
||||||
timeout := 15 * time.Second
|
timeout := 30 * time.Second
|
||||||
timeoutChannel := time.After(timeout)
|
timeoutChannel := time.After(timeout)
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
|
Loading…
Reference in New Issue
Block a user