mirror of
https://github.com/netbirdio/netbird.git
synced 2026-08-25 09:01:29 +02:00
a144e8c144
* switch to go.uber.org/mock/gomock Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> * updated go:generate commands + regenerated mocks Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> * update go:generate mockgen commands Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> * removed duplicate import Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> * fix go:generate Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> --------- Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io>
64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"go.uber.org/mock/gomock"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
nbconfig "github.com/netbirdio/netbird/management/internals/server/config"
|
|
"github.com/netbirdio/netbird/management/server/store"
|
|
)
|
|
|
|
func TestResolveDomains_FreshInstallUsesDefault(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
t.Cleanup(ctrl.Finish)
|
|
|
|
mockStore := store.NewMockStore(ctrl)
|
|
mockStore.EXPECT().GetAccountsCounter(gomock.Any()).Return(int64(0), nil)
|
|
|
|
srv := NewServer(&Config{NbConfig: &nbconfig.Config{}})
|
|
Inject[store.Store](srv, mockStore)
|
|
|
|
srv.ResolveDomains(context.Background())
|
|
|
|
require.Equal(t, DefaultSelfHostedDomain, srv.dnsDomain)
|
|
require.Equal(t, DefaultSelfHostedDomain, srv.mgmtSingleAccModeDomain)
|
|
}
|
|
|
|
func TestResolveDomains_ExistingInstallUsesPersistedDomain(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
t.Cleanup(ctrl.Finish)
|
|
|
|
mockStore := store.NewMockStore(ctrl)
|
|
mockStore.EXPECT().GetAccountsCounter(gomock.Any()).Return(int64(1), nil)
|
|
mockStore.EXPECT().GetAnyAccountID(gomock.Any()).Return("acc-1", nil)
|
|
mockStore.EXPECT().GetAccountDomainAndCategory(gomock.Any(), store.LockingStrengthNone, "acc-1").Return("vpn.mycompany.com", "", nil)
|
|
|
|
srv := NewServer(&Config{NbConfig: &nbconfig.Config{}})
|
|
Inject[store.Store](srv, mockStore)
|
|
|
|
srv.ResolveDomains(context.Background())
|
|
|
|
require.Equal(t, "vpn.mycompany.com", srv.dnsDomain)
|
|
require.Equal(t, "vpn.mycompany.com", srv.mgmtSingleAccModeDomain)
|
|
}
|
|
|
|
func TestResolveDomains_StoreErrorFallsBackToDefault(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
t.Cleanup(ctrl.Finish)
|
|
|
|
mockStore := store.NewMockStore(ctrl)
|
|
mockStore.EXPECT().GetAccountsCounter(gomock.Any()).Return(int64(0), errors.New("db failed"))
|
|
|
|
srv := NewServer(&Config{NbConfig: &nbconfig.Config{}})
|
|
Inject[store.Store](srv, mockStore)
|
|
|
|
srv.ResolveDomains(context.Background())
|
|
|
|
require.Equal(t, DefaultSelfHostedDomain, srv.dnsDomain)
|
|
require.Equal(t, DefaultSelfHostedDomain, srv.mgmtSingleAccModeDomain)
|
|
}
|