sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-18 16:26:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve addition of new peers in Management service. (#56)

Browse Source 
* Store refactoring
* Improve addition of new peers in Management service.
This commit is contained in:
andpar83
2021-07-18 11:51:09 -07:00
committed by GitHub
parent f9c3ed784f
commit 7b52049333
4 changed files with 139 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
management/management_test.go
View File

@@ -2,6 +2,13 @@ package management_test
import (
"context"
"io/ioutil"
"net"
"os"
"path/filepath"
"strings"
"time"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
log "github.com/sirupsen/logrus"
@@ -11,12 +18,6 @@ import (
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
"google.golang.org/grpc"
"google.golang.org/grpc/keepalive"
"io/ioutil"
"net"
"os"
"path/filepath"
"strings"
"time"
)
var _ = Describe("Client", func() {
@@ -114,11 +115,11 @@ var _ = Describe("Client", func() {
Expect(err).NotTo(HaveOccurred())
store, err := util.ReadJson(filepath.Join(dataDir, "store.json"), &mgmt.Store{})
store, err := util.ReadJson(filepath.Join(dataDir, "store.json"), &mgmt.FileStore{})
Expect(err).NotTo(HaveOccurred())
Expect(store.(*mgmt.Store)).NotTo(BeNil())
user := store.(*mgmt.Store).Accounts["bf1c8084-ba50-4ce7-9439-34653001fc3b"]
Expect(store.(*mgmt.FileStore)).NotTo(BeNil())
user := store.(*mgmt.FileStore).Accounts["bf1c8084-ba50-4ce7-9439-34653001fc3b"]
Expect(user.Peers[key.PublicKey().String()]).NotTo(BeNil())
Expect(user.SetupKeys[strings.ToLower(setupKey)]).NotTo(BeNil())