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

[client] Fix mgmProber interface to match unexported GetServerPublicKey (#5815)

Browse Source 
Update the mgmProber interface to use HealthCheck() instead of the
now-unexported GetServerPublicKey(), aligning with the changes in the
management client API.
This commit is contained in:
Zoltan Papp
2026-04-07 13:13:38 +02:00
committed by GitHub
parent 0efef671d7
commit 6da34e483c
2 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
client/internal/profilemanager/config.go
View File

@@ -41,7 +41,7 @@ const (
// mgmProber is the subset of management client needed for URL migration probes. // mgmProber is the subset of management client needed for URL migration probes.
type mgmProber interface { type mgmProber interface {
GetServerPublicKey() (*wgtypes.Key, error) HealthCheck() error
Close() error Close() error
} }

14
client/internal/profilemanager/config_test.go
View File

@@ -17,12 +17,10 @@ import (
"github.com/netbirdio/netbird/util" "github.com/netbirdio/netbird/util"
) )
type mockMgmProber struct { type mockMgmProber struct{}
key wgtypes.Key
}
func (m *mockMgmProber) GetServerPublicKey() (*wgtypes.Key, error) { func (m *mockMgmProber) HealthCheck() error {
return &m.key, nil return nil
} }
func (m *mockMgmProber) Close() error { return nil } func (m *mockMgmProber) Close() error { return nil }
@@ -247,11 +245,7 @@ func TestWireguardPortDefaultVsExplicit(t *testing.T) {
func TestUpdateOldManagementURL(t *testing.T) { func TestUpdateOldManagementURL(t *testing.T) {
origProber := newMgmProber origProber := newMgmProber
newMgmProber = func(_ context.Context, _ string, _ wgtypes.Key, _ bool) (mgmProber, error) { newMgmProber = func(_ context.Context, _ string, _ wgtypes.Key, _ bool) (mgmProber, error) {
key, err := wgtypes.GenerateKey() return &mockMgmProber{}, nil
if err != nil {
return nil, err
}
return &mockMgmProber{key: key.PublicKey()}, nil
} }
t.Cleanup(func() { newMgmProber = origProber }) t.Cleanup(func() { newMgmProber = origProber })