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

Fix netbird_test.go

Browse Source
This commit is contained in:
Viktor Liu
2026-02-08 17:33:04 +08:00
parent ed58659a01
commit 3883b2fb41
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
proxy/internal/roundtrip/netbird_test.go
View File

@@ -7,15 +7,23 @@ import (
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"google.golang.org/grpc"
"github.com/netbirdio/netbird/proxy/internal/types" "github.com/netbirdio/netbird/proxy/internal/types"
"github.com/netbirdio/netbird/shared/management/domain" "github.com/netbirdio/netbird/shared/management/domain"
"github.com/netbirdio/netbird/shared/management/proto"
) )
type mockMgmtClient struct{}
func (m *mockMgmtClient) CreateProxyPeer(_ context.Context, _ *proto.CreateProxyPeerRequest, _ ...grpc.CallOption) (*proto.CreateProxyPeerResponse, error) {
return &proto.CreateProxyPeerResponse{Success: true}, nil
}
// mockNetBird creates a NetBird instance for testing without actually connecting. // mockNetBird creates a NetBird instance for testing without actually connecting.
// It uses an invalid management URL to prevent real connections. // It uses an invalid management URL to prevent real connections.
func mockNetBird() *NetBird { func mockNetBird() *NetBird {
return NewNetBird("http://invalid.test:9999", "test-proxy", nil, nil) return NewNetBird("http://invalid.test:9999", "test-proxy", nil, nil, &mockMgmtClient{})
} }
func TestNetBird_AddPeer_CreatesClientForNewAccount(t *testing.T) { func TestNetBird_AddPeer_CreatesClientForNewAccount(t *testing.T) {