mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-16 07:16:38 +00:00
63c83aa8d2
CLI: new expose command to publish a local port with flags for PIN, password, user groups, custom domain, name prefix and protocol (HTTP default). Management/API: create/renew/stop expose sessions (streamed status), automatic naming/domain, TTL renewals, background expiration, new management RPCs and client methods. UI/API: account settings now include peer_expose_enabled and peer_expose_groups; new activity codes for peer expose events.
96 lines
3.0 KiB
Go
96 lines
3.0 KiB
Go
package expose
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
daemonProto "github.com/netbirdio/netbird/client/proto"
|
|
mgm "github.com/netbirdio/netbird/shared/management/client"
|
|
)
|
|
|
|
func TestManager_Expose_Success(t *testing.T) {
|
|
mock := &mgm.MockClient{
|
|
CreateExposeFunc: func(ctx context.Context, req mgm.ExposeRequest) (*mgm.ExposeResponse, error) {
|
|
return &mgm.ExposeResponse{
|
|
ServiceName: "my-service",
|
|
ServiceURL: "https://my-service.example.com",
|
|
Domain: "my-service.example.com",
|
|
}, nil
|
|
},
|
|
}
|
|
|
|
m := NewManager(context.Background(), mock)
|
|
result, err := m.Expose(context.Background(), Request{Port: 8080})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "my-service", result.ServiceName, "service name should match")
|
|
assert.Equal(t, "https://my-service.example.com", result.ServiceURL, "service URL should match")
|
|
assert.Equal(t, "my-service.example.com", result.Domain, "domain should match")
|
|
}
|
|
|
|
func TestManager_Expose_Error(t *testing.T) {
|
|
mock := &mgm.MockClient{
|
|
CreateExposeFunc: func(ctx context.Context, req mgm.ExposeRequest) (*mgm.ExposeResponse, error) {
|
|
return nil, errors.New("permission denied")
|
|
},
|
|
}
|
|
|
|
m := NewManager(context.Background(), mock)
|
|
_, err := m.Expose(context.Background(), Request{Port: 8080})
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "permission denied", "error should propagate")
|
|
}
|
|
|
|
func TestManager_Renew_Success(t *testing.T) {
|
|
mock := &mgm.MockClient{
|
|
RenewExposeFunc: func(ctx context.Context, domain string) error {
|
|
assert.Equal(t, "my-service.example.com", domain, "domain should be passed through")
|
|
return nil
|
|
},
|
|
}
|
|
|
|
m := NewManager(context.Background(), mock)
|
|
err := m.renew(context.Background(), "my-service.example.com")
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestManager_Renew_Timeout(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
mock := &mgm.MockClient{
|
|
RenewExposeFunc: func(ctx context.Context, domain string) error {
|
|
return ctx.Err()
|
|
},
|
|
}
|
|
|
|
m := NewManager(ctx, mock)
|
|
err := m.renew(ctx, "my-service.example.com")
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestNewRequest(t *testing.T) {
|
|
req := &daemonProto.ExposeServiceRequest{
|
|
Port: 8080,
|
|
Protocol: daemonProto.ExposeProtocol_EXPOSE_HTTPS,
|
|
Pin: "123456",
|
|
Password: "secret",
|
|
UserGroups: []string{"group1", "group2"},
|
|
Domain: "custom.example.com",
|
|
NamePrefix: "my-prefix",
|
|
}
|
|
|
|
exposeReq := NewRequest(req)
|
|
|
|
assert.Equal(t, uint16(8080), exposeReq.Port, "port should match")
|
|
assert.Equal(t, int(daemonProto.ExposeProtocol_EXPOSE_HTTPS), exposeReq.Protocol, "protocol should match")
|
|
assert.Equal(t, "123456", exposeReq.Pin, "pin should match")
|
|
assert.Equal(t, "secret", exposeReq.Password, "password should match")
|
|
assert.Equal(t, []string{"group1", "group2"}, exposeReq.UserGroups, "user groups should match")
|
|
assert.Equal(t, "custom.example.com", exposeReq.Domain, "domain should match")
|
|
assert.Equal(t, "my-prefix", exposeReq.NamePrefix, "name prefix should match")
|
|
}
|