extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-06-26 04:31:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source 
Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
This commit is contained in:
bcmmbaga 2025-01-02 17:37:17 +03:00
parent 7e3ff3044c
commit c76683a8d3
No known key found for this signature in database
GPG Key ID: 511EED5C928AD547
11 changed files with 61 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
go.mod
View File

@ -39,6 +39,7 @@ require (
github.com/coreos/go-iptables v0.7.0 github.com/coreos/go-iptables v0.7.0
github.com/creack/pty v1.1.18 github.com/creack/pty v1.1.18
github.com/davecgh/go-spew v1.1.1 github.com/davecgh/go-spew v1.1.1
github.com/docker/docker v26.1.5+incompatible
github.com/eko/gocache/v3 v3.1.1 github.com/eko/gocache/v3 v3.1.1
github.com/fsnotify/fsnotify v1.7.0 github.com/fsnotify/fsnotify v1.7.0
github.com/gliderlabs/ssh v0.3.4 github.com/gliderlabs/ssh v0.3.4
@ -140,7 +141,6 @@ require (
github.com/dgraph-io/ristretto v0.1.1 // indirect github.com/dgraph-io/ristretto v0.1.1 // indirect
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
github.com/distribution/reference v0.6.0 // indirect github.com/distribution/reference v0.6.0 // indirect
github.com/docker/docker v26.1.5+incompatible // indirect
github.com/docker/go-connections v0.5.0 // indirect github.com/docker/go-connections v0.5.0 // indirect
github.com/docker/go-units v0.5.0 // indirect github.com/docker/go-units v0.5.0 // indirect
github.com/felixge/httpsnoop v1.0.4 // indirect github.com/felixge/httpsnoop v1.0.4 // indirect

37
management/server/account_test.go
View File

@ -3,7 +3,6 @@ package server
import ( import (
"context" "context"
"crypto/sha256" "crypto/sha256"
"database/sql"
b64 "encoding/base64" b64 "encoding/base64"
"encoding/json" "encoding/json"
"fmt" "fmt"
@ -17,6 +16,7 @@ import (
"time" "time"
"github.com/golang-jwt/jwt" "github.com/golang-jwt/jwt"
"github.com/netbirdio/netbird/management/server/util"
resourceTypes "github.com/netbirdio/netbird/management/server/networks/resources/types" resourceTypes "github.com/netbirdio/netbird/management/server/networks/resources/types"
routerTypes "github.com/netbirdio/netbird/management/server/networks/routers/types" routerTypes "github.com/netbirdio/netbird/management/server/networks/routers/types"
@ -188,7 +188,7 @@ func TestAccount_GetPeerNetworkMap(t *testing.T) {
LoginExpired: true, LoginExpired: true,
}, },
UserID: userID, UserID: userID,
LastLogin: time.Now().UTC().Add(-time.Hour * 24 * 30 * 30), LastLogin: util.ToPtr(time.Now().UTC().Add(-time.Hour * 24 * 30 * 30)),
}, },
"peer-2": { "peer-2": {
ID: peerID2, ID: peerID2,
@ -202,7 +202,7 @@ func TestAccount_GetPeerNetworkMap(t *testing.T) {
LoginExpired: false, LoginExpired: false,
}, },
UserID: userID, UserID: userID,
LastLogin: time.Now().UTC(), LastLogin: util.ToPtr(time.Now().UTC()),
LoginExpirationEnabled: true, LoginExpirationEnabled: true,
}, },
}, },
@ -226,7 +226,7 @@ func TestAccount_GetPeerNetworkMap(t *testing.T) {
LoginExpired: true, LoginExpired: true,
}, },
UserID: userID, UserID: userID,
LastLogin: time.Now().UTC().Add(-time.Hour * 24 * 30 * 30), LastLogin: util.ToPtr(time.Now().UTC().Add(-time.Hour * 24 * 30 * 30)),
LoginExpirationEnabled: true, LoginExpirationEnabled: true,
}, },
"peer-2": { "peer-2": {
@ -241,7 +241,7 @@ func TestAccount_GetPeerNetworkMap(t *testing.T) {
LoginExpired: true, LoginExpired: true,
}, },
UserID: userID, UserID: userID,
LastLogin: time.Now().UTC().Add(-time.Hour * 24 * 30 * 30), LastLogin: util.ToPtr(time.Now().UTC().Add(-time.Hour * 24 * 30 * 30)),
LoginExpirationEnabled: true, LoginExpirationEnabled: true,
}, },
}, },
@ -814,7 +814,6 @@ func TestDefaultAccountManager_MarkPATUsed(t *testing.T) {
"tokenId": { "tokenId": {
ID: "tokenId", ID: "tokenId",
HashedToken: encodedHashedToken, HashedToken: encodedHashedToken,
LastUsed: time.Time{},
}, },
}, },
} }
@ -1097,7 +1096,7 @@ func genUsers(p string, n int) map[string]*types.User {
users[fmt.Sprintf("%s-%d", p, i)] = &types.User{ users[fmt.Sprintf("%s-%d", p, i)] = &types.User{
Id: fmt.Sprintf("%s-%d", p, i), Id: fmt.Sprintf("%s-%d", p, i),
Role: types.UserRoleAdmin, Role: types.UserRoleAdmin,
LastLogin: sql.NullTime{Time: now, Valid: !now.IsZero()}, LastLogin: util.ToPtr(now),
CreatedAt: now, CreatedAt: now,
Issued: "api", Issued: "api",
AutoGroups: []string{"one", "two", "three", "four", "five", "six", "seven", "eight", "nine", "ten"}, AutoGroups: []string{"one", "two", "three", "four", "five", "six", "seven", "eight", "nine", "ten"},
@ -1749,10 +1748,10 @@ func TestAccount_Copy(t *testing.T) {
ID: "pat1", ID: "pat1",
Name: "First PAT", Name: "First PAT",
HashedToken: "SoMeHaShEdToKeN", HashedToken: "SoMeHaShEdToKeN",
ExpirationDate: time.Now().UTC().AddDate(0, 0, 7), ExpirationDate: util.ToPtr(time.Now().UTC().AddDate(0, 0, 7)),
CreatedBy: "user1", CreatedBy: "user1",
CreatedAt: time.Now().UTC(), CreatedAt: time.Now().UTC(),
LastUsed: time.Now().UTC(), LastUsed: util.ToPtr(time.Now().UTC()),
}, },
}, },
}, },
@ -2108,7 +2107,7 @@ func TestAccount_GetExpiredPeers(t *testing.T) {
Connected: true, Connected: true,
LoginExpired: false, LoginExpired: false,
}, },
LastLogin: time.Now().UTC().Add(-30 * time.Minute), LastLogin: util.ToPtr(time.Now().UTC().Add(-30 * time.Minute)),
UserID: userID, UserID: userID,
}, },
"peer-2": { "peer-2": {
@ -2119,7 +2118,7 @@ func TestAccount_GetExpiredPeers(t *testing.T) {
Connected: true, Connected: true,
LoginExpired: false, LoginExpired: false,
}, },
LastLogin: time.Now().UTC().Add(-2 * time.Hour), LastLogin: util.ToPtr(time.Now().UTC().Add(-2 * time.Hour)),
UserID: userID, UserID: userID,
}, },
@ -2131,7 +2130,7 @@ func TestAccount_GetExpiredPeers(t *testing.T) {
Connected: true, Connected: true,
LoginExpired: false, LoginExpired: false,
}, },
LastLogin: time.Now().UTC().Add(-1 * time.Hour), LastLogin: util.ToPtr(time.Now().UTC().Add(-1 * time.Hour)),
UserID: userID, UserID: userID,
}, },
}, },
@ -2193,7 +2192,7 @@ func TestAccount_GetInactivePeers(t *testing.T) {
Connected: false, Connected: false,
LoginExpired: false, LoginExpired: false,
}, },
LastLogin: time.Now().UTC().Add(-30 * time.Minute), LastLogin: util.ToPtr(time.Now().UTC().Add(-30 * time.Minute)),
UserID: userID, UserID: userID,
}, },
"peer-2": { "peer-2": {
@ -2204,7 +2203,7 @@ func TestAccount_GetInactivePeers(t *testing.T) {
Connected: false, Connected: false,
LoginExpired: false, LoginExpired: false,
}, },
LastLogin: time.Now().UTC().Add(-2 * time.Hour), LastLogin: util.ToPtr(time.Now().UTC().Add(-2 * time.Hour)),
UserID: userID, UserID: userID,
}, },
"peer-3": { "peer-3": {
@ -2215,7 +2214,7 @@ func TestAccount_GetInactivePeers(t *testing.T) {
Connected: true, Connected: true,
LoginExpired: false, LoginExpired: false,
}, },
LastLogin: time.Now().UTC().Add(-1 * time.Hour), LastLogin: util.ToPtr(time.Now().UTC().Add(-1 * time.Hour)),
UserID: userID, UserID: userID,
}, },
}, },
@ -2485,7 +2484,7 @@ func TestAccount_GetNextPeerExpiration(t *testing.T) {
LoginExpired: false, LoginExpired: false,
}, },
LoginExpirationEnabled: true, LoginExpirationEnabled: true,
LastLogin: time.Now().UTC(), LastLogin: util.ToPtr(time.Now().UTC()),
UserID: userID, UserID: userID,
}, },
"peer-2": { "peer-2": {
@ -2645,7 +2644,7 @@ func TestAccount_GetNextInactivePeerExpiration(t *testing.T) {
LastSeen: time.Now().Add(-1 * time.Second), LastSeen: time.Now().Add(-1 * time.Second),
}, },
InactivityExpirationEnabled: true, InactivityExpirationEnabled: true,
LastLogin: time.Now().UTC(), LastLogin: util.ToPtr(time.Now().UTC()),
UserID: userID, UserID: userID,
}, },
"peer-2": { "peer-2": {
@ -2723,8 +2722,8 @@ func TestAccount_SetJWTGroups(t *testing.T) {
}, },
Settings: &types.Settings{GroupsPropagationEnabled: true, JWTGroupsEnabled: true, JWTGroupsClaimName: "groups"}, Settings: &types.Settings{GroupsPropagationEnabled: true, JWTGroupsEnabled: true, JWTGroupsClaimName: "groups"},
Users: map[string]*types.User{ Users: map[string]*types.User{
"user1": {Id: "user1", AccountID: "accountID"}, "user1": {Id: "user1", AccountID: "accountID", CreatedAt: time.Now()},
"user2": {Id: "user2", AccountID: "accountID"}, "user2": {Id: "user2", AccountID: "accountID", CreatedAt: time.Now()},
}, },
} }

1
management/server/group_test.go
View File

@ -355,6 +355,7 @@ func initTestGroupAccount(am *DefaultAccountManager) (*DefaultAccountManager, *t
setupKey := &types.SetupKey{ setupKey := &types.SetupKey{
Id: "example setup key", Id: "example setup key",
AutoGroups: []string{groupForSetupKeys.ID}, AutoGroups: []string{groupForSetupKeys.ID},
UpdatedAt: time.Now(),
} }
user := &types.User{ user := &types.User{

17
management/server/http/handlers/routes/routes_handler_test.go
View File

@ -11,6 +11,7 @@ import (
"net/netip" "net/netip"
"testing" "testing"
"github.com/netbirdio/netbird/management/server/util"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"github.com/netbirdio/netbird/management/server/http/api" "github.com/netbirdio/netbird/management/server/http/api"
@ -239,7 +240,7 @@ func TestRoutesHandlers(t *testing.T) {
Id: existingRouteID, Id: existingRouteID,
Description: "Post", Description: "Post",
NetworkId: "awesomeNet", NetworkId: "awesomeNet",
Network: toPtr("192.168.0.0/16"), Network: util.ToPtr("192.168.0.0/16"),
Peer: &existingPeerID, Peer: &existingPeerID,
NetworkType: route.IPv4NetworkString, NetworkType: route.IPv4NetworkString,
Masquerade: false, Masquerade: false,
@ -259,7 +260,7 @@ func TestRoutesHandlers(t *testing.T) {
Id: existingRouteID, Id: existingRouteID,
Description: "Post", Description: "Post",
NetworkId: "domainNet", NetworkId: "domainNet",
Network: toPtr("invalid Prefix"), Network: util.ToPtr("invalid Prefix"),
KeepRoute: true, KeepRoute: true,
Domains: &[]string{existingDomain}, Domains: &[]string{existingDomain},
Peer: &existingPeerID, Peer: &existingPeerID,
@ -281,7 +282,7 @@ func TestRoutesHandlers(t *testing.T) {
Id: existingRouteID, Id: existingRouteID,
Description: "Post", Description: "Post",
NetworkId: "awesomeNet", NetworkId: "awesomeNet",
Network: toPtr("192.168.0.0/16"), Network: util.ToPtr("192.168.0.0/16"),
Peer: &existingPeerID, Peer: &existingPeerID,
NetworkType: route.IPv4NetworkString, NetworkType: route.IPv4NetworkString,
Masquerade: false, Masquerade: false,
@ -385,7 +386,7 @@ func TestRoutesHandlers(t *testing.T) {
Id: existingRouteID, Id: existingRouteID,
Description: "Post", Description: "Post",
NetworkId: "awesomeNet", NetworkId: "awesomeNet",
Network: toPtr("192.168.0.0/16"), Network: util.ToPtr("192.168.0.0/16"),
Peer: &existingPeerID, Peer: &existingPeerID,
NetworkType: route.IPv4NetworkString, NetworkType: route.IPv4NetworkString,
Masquerade: false, Masquerade: false,
@ -404,7 +405,7 @@ func TestRoutesHandlers(t *testing.T) {
Id: existingRouteID, Id: existingRouteID,
Description: "Post", Description: "Post",
NetworkId: "awesomeNet", NetworkId: "awesomeNet",
Network: toPtr("invalid Prefix"), Network: util.ToPtr("invalid Prefix"),
Domains: &[]string{existingDomain}, Domains: &[]string{existingDomain},
Peer: &existingPeerID, Peer: &existingPeerID,
NetworkType: route.DomainNetworkString, NetworkType: route.DomainNetworkString,
@ -425,7 +426,7 @@ func TestRoutesHandlers(t *testing.T) {
Id: existingRouteID, Id: existingRouteID,
Description: "Post", Description: "Post",
NetworkId: "awesomeNet", NetworkId: "awesomeNet",
Network: toPtr("192.168.0.0/16"), Network: util.ToPtr("192.168.0.0/16"),
Peer: &emptyString, Peer: &emptyString,
PeerGroups: &[]string{existingGroupID}, PeerGroups: &[]string{existingGroupID},
NetworkType: route.IPv4NetworkString, NetworkType: route.IPv4NetworkString,
@ -663,7 +664,3 @@ func toApiRoute(t *testing.T, r *route.Route) *api.Route {
require.NoError(t, err, "Failed to convert route") require.NoError(t, err, "Failed to convert route")
return apiRoute return apiRoute
} }
func toPtr[T any](v T) *T {
return &v
}

2
management/server/http/handlers/setup_keys/setupkeys_handler.go
View File

@ -240,7 +240,7 @@ func toResponseBody(key *types.SetupKey) *api.SetupKey {
Id: key.Id, Id: key.Id,
Key: key.KeySecret, Key: key.KeySecret,
Name: key.Name, Name: key.Name,
Expires: key.ExpiresAt, Expires: key.ExpirationTime(),
Type: string(key.Type), Type: string(key.Type),
Valid: key.IsValid(), Valid: key.IsValid(),
Revoked: key.Revoked, Revoked: key.Revoked,

13
management/server/http/handlers/users/pat_handler_test.go
View File

@ -12,6 +12,7 @@ import (
"github.com/google/go-cmp/cmp" "github.com/google/go-cmp/cmp"
"github.com/gorilla/mux" "github.com/gorilla/mux"
"github.com/netbirdio/netbird/management/server/util"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/netbirdio/netbird/management/server/http/api" "github.com/netbirdio/netbird/management/server/http/api"
@ -42,19 +43,19 @@ var testAccount = &types.Account{
ID: existingTokenID, ID: existingTokenID,
Name: "My first token", Name: "My first token",
HashedToken: "someHash", HashedToken: "someHash",
ExpirationDate: time.Now().UTC().AddDate(0, 0, 7), ExpirationDate: util.ToPtr(time.Now().UTC().AddDate(0, 0, 7)),
CreatedBy: existingUserID, CreatedBy: existingUserID,
CreatedAt: time.Now().UTC(), CreatedAt: time.Now().UTC(),
LastUsed: time.Now().UTC(), LastUsed: util.ToPtr(time.Now().UTC()),
}, },
"token2": { "token2": {
ID: "token2", ID: "token2",
Name: "My second token", Name: "My second token",
HashedToken: "someOtherHash", HashedToken: "someOtherHash",
ExpirationDate: time.Now().UTC().AddDate(0, 0, 7), ExpirationDate: util.ToPtr(time.Now().UTC().AddDate(0, 0, 7)),
CreatedBy: existingUserID, CreatedBy: existingUserID,
CreatedAt: time.Now().UTC(), CreatedAt: time.Now().UTC(),
LastUsed: time.Now().UTC(), LastUsed: util.ToPtr(time.Now().UTC()),
}, },
}, },
}, },
@ -248,8 +249,8 @@ func toTokenResponse(serverToken types.PersonalAccessToken) api.PersonalAccessTo
Id: serverToken.ID, Id: serverToken.ID,
Name: serverToken.Name, Name: serverToken.Name,
CreatedAt: serverToken.CreatedAt, CreatedAt: serverToken.CreatedAt,
LastUsed: &serverToken.LastUsed, LastUsed: serverToken.LastUsed,
CreatedBy: serverToken.CreatedBy, CreatedBy: serverToken.CreatedBy,
ExpirationDate: serverToken.ExpirationDate, ExpirationDate: serverToken.ExpirationTime(),
} }
} }

5
management/server/http/middleware/auth_middleware_test.go
View File

@ -9,6 +9,7 @@ import (
"time" "time"
"github.com/golang-jwt/jwt" "github.com/golang-jwt/jwt"
"github.com/netbirdio/netbird/management/server/util"
"github.com/netbirdio/netbird/management/server/http/middleware/bypass" "github.com/netbirdio/netbird/management/server/http/middleware/bypass"
"github.com/netbirdio/netbird/management/server/jwtclaims" "github.com/netbirdio/netbird/management/server/jwtclaims"
@ -39,10 +40,10 @@ var testAccount = &types.Account{
ID: tokenID, ID: tokenID,
Name: "My first token", Name: "My first token",
HashedToken: "someHash", HashedToken: "someHash",
ExpirationDate: time.Now().UTC().AddDate(0, 0, 7), ExpirationDate: util.ToPtr(time.Now().UTC().AddDate(0, 0, 7)),
CreatedBy: userID, CreatedBy: userID,
CreatedAt: time.Now().UTC(), CreatedAt: time.Now().UTC(),
LastUsed: time.Now().UTC(), LastUsed: util.ToPtr(time.Now().UTC()),
}, },
}, },
}, },

7
management/server/peer_test.go
View File

@ -13,6 +13,7 @@ import (
"testing" "testing"
"time" "time"
"github.com/netbirdio/netbird/management/server/util"
"github.com/rs/xid" "github.com/rs/xid"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
@ -80,7 +81,7 @@ func TestPeer_LoginExpired(t *testing.T) {
t.Run(c.name, func(t *testing.T) { t.Run(c.name, func(t *testing.T) {
peer := &nbpeer.Peer{ peer := &nbpeer.Peer{
LoginExpirationEnabled: c.expirationEnabled, LoginExpirationEnabled: c.expirationEnabled,
LastLogin: c.lastLogin, LastLogin: util.ToPtr(c.lastLogin),
UserID: userID, UserID: userID,
} }
@ -141,7 +142,7 @@ func TestPeer_SessionExpired(t *testing.T) {
} }
peer := &nbpeer.Peer{ peer := &nbpeer.Peer{
InactivityExpirationEnabled: c.expirationEnabled, InactivityExpirationEnabled: c.expirationEnabled,
LastLogin: c.lastLogin, LastLogin: util.ToPtr(c.lastLogin),
Status: peerStatus, Status: peerStatus,
UserID: userID, UserID: userID,
} }
@ -1209,7 +1210,7 @@ func Test_RegisterPeerByUser(t *testing.T) {
UserID: existingUserID, UserID: existingUserID,
Status: &nbpeer.PeerStatus{Connected: false, LastSeen: time.Now()}, Status: &nbpeer.PeerStatus{Connected: false, LastSeen: time.Now()},
SSHEnabled: false, SSHEnabled: false,
LastLogin: time.Now(), LastLogin: util.ToPtr(time.Now()),
} }
addedPeer, _, _, err := am.AddPeer(context.Background(), "", existingUserID, newPeer) addedPeer, _, _, err := am.AddPeer(context.Background(), "", existingUserID, newPeer)

4
management/server/setupkey_test.go
View File

@ -66,7 +66,7 @@ func TestDefaultAccountManager_SaveSetupKey(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
assertKey(t, newKey, keyName, revoked, "reusable", 0, key.CreatedAt, key.ExpiresAt, assertKey(t, newKey, keyName, revoked, "reusable", 0, key.CreatedAt, key.ExpirationTime(),
key.Id, time.Now().UTC(), autoGroups, true) key.Id, time.Now().UTC(), autoGroups, true)
// check the corresponding events that should have been generated // check the corresponding events that should have been generated
@ -391,7 +391,7 @@ func TestSetupKey_Copy(t *testing.T) {
key, _ := types.GenerateSetupKey("key name", types.SetupKeyOneOff, time.Hour, []string{}, types.SetupKeyUnlimitedUsage, false) key, _ := types.GenerateSetupKey("key name", types.SetupKeyOneOff, time.Hour, []string{}, types.SetupKeyUnlimitedUsage, false)
keyCopy := key.Copy() keyCopy := key.Copy()
assertKey(t, keyCopy, key.Name, key.Revoked, string(key.Type), key.UsedTimes, key.CreatedAt, key.ExpiresAt, key.Id, assertKey(t, keyCopy, key.Name, key.Revoked, string(key.Type), key.UsedTimes, key.CreatedAt, key.ExpirationTime(), key.Id,
key.UpdatedAt, key.AutoGroups, true) key.UpdatedAt, key.AutoGroups, true)
} }

18
management/server/types/setupkey.go
View File

@ -39,7 +39,7 @@ type SetupKey struct {
Name string Name string
Type SetupKeyType Type SetupKeyType
CreatedAt time.Time CreatedAt time.Time
ExpiresAt time.Time ExpiresAt *time.Time
UpdatedAt time.Time `gorm:"autoUpdateTime:false"` UpdatedAt time.Time `gorm:"autoUpdateTime:false"`
// Revoked indicates whether the key was revoked or not (we don't remove them for tracking purposes) // Revoked indicates whether the key was revoked or not (we don't remove them for tracking purposes)
Revoked bool Revoked bool
@ -95,6 +95,14 @@ func (key *SetupKey) LastUsedTime() time.Time {
return time.Time{} return time.Time{}
} }
// ExpirationTime returns the expiration time of the setup key.
func (key *SetupKey) ExpirationTime() time.Time {
if key.ExpiresAt != nil {
return *key.ExpiresAt
}
return time.Time{}
}
// HiddenKey returns the Key value hidden with "*" and a 5 character prefix. // HiddenKey returns the Key value hidden with "*" and a 5 character prefix.
// E.g., "831F6*******************************" // E.g., "831F6*******************************"
func HiddenKey(key string, length int) string { func HiddenKey(key string, length int) string {
@ -125,10 +133,10 @@ func (key *SetupKey) IsRevoked() bool {
// IsExpired if key was expired // IsExpired if key was expired
func (key *SetupKey) IsExpired() bool { func (key *SetupKey) IsExpired() bool {
if key.ExpiresAt.IsZero() { if key.ExpirationTime().IsZero() {
return false return false
} }
return time.Now().After(key.ExpiresAt) return time.Now().After(key.ExpirationTime())
} }
// IsOverUsed if the key was used too many times. SetupKey.UsageLimit == 0 indicates the unlimited usage. // IsOverUsed if the key was used too many times. SetupKey.UsageLimit == 0 indicates the unlimited usage.
@ -149,9 +157,9 @@ func GenerateSetupKey(name string, t SetupKeyType, validFor time.Duration, autoG
limit = 1 limit = 1
} }
expiresAt := time.Time{} var expiresAt *time.Time
if validFor != 0 { if validFor != 0 {
expiresAt = time.Now().UTC().Add(validFor) expiresAt = util.ToPtr(time.Now().UTC().Add(validFor))
} }
hashedKey := sha256.Sum256([]byte(key)) hashedKey := sha256.Sum256([]byte(key))

8
management/server/user_test.go
View File

@ -2,7 +2,6 @@ package server
import ( import (
"context" "context"
"database/sql"
"fmt" "fmt"
"reflect" "reflect"
"testing" "testing"
@ -11,6 +10,7 @@ import (
"github.com/eko/gocache/v3/cache" "github.com/eko/gocache/v3/cache"
cacheStore "github.com/eko/gocache/v3/store" cacheStore "github.com/eko/gocache/v3/store"
"github.com/google/go-cmp/cmp" "github.com/google/go-cmp/cmp"
"github.com/netbirdio/netbird/management/server/util"
nbpeer "github.com/netbirdio/netbird/management/server/peer" nbpeer "github.com/netbirdio/netbird/management/server/peer"
"github.com/netbirdio/netbird/management/server/store" "github.com/netbirdio/netbird/management/server/store"
@ -322,14 +322,14 @@ func TestUser_Copy(t *testing.T) {
ID: "pat1", ID: "pat1",
Name: "First PAT", Name: "First PAT",
HashedToken: "SoMeHaShEdToKeN", HashedToken: "SoMeHaShEdToKeN",
ExpirationDate: time.Now().AddDate(0, 0, 7), ExpirationDate: util.ToPtr(time.Now().AddDate(0, 0, 7)),
CreatedBy: "userId", CreatedBy: "userId",
CreatedAt: time.Now(), CreatedAt: time.Now(),
LastUsed: time.Now(), LastUsed: util.ToPtr(time.Now()),
}, },
}, },
Blocked: false, Blocked: false,
LastLogin: sql.NullTime{Time: time.Now().UTC(), Valid: true}, LastLogin: util.ToPtr(time.Now().UTC()),
CreatedAt: time.Now().UTC(), CreatedAt: time.Now().UTC(),
Issued: "test", Issued: "test",
IntegrationReference: integration_reference.IntegrationReference{ IntegrationReference: integration_reference.IntegrationReference{