From d7523e6f3ff55373be0b8ff55ce600541161ba1e Mon Sep 17 00:00:00 2001 From: Pascal Fischer Date: Fri, 21 Feb 2025 16:43:52 +0100 Subject: [PATCH] fix test --- management/server/account_benchmark_test.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/management/server/account_benchmark_test.go b/management/server/account_benchmark_test.go index 498bceec3..360a23042 100644 --- a/management/server/account_benchmark_test.go +++ b/management/server/account_benchmark_test.go @@ -6,26 +6,30 @@ package server import ( "context" "fmt" + "io" + "log" "net" "os" "strconv" "testing" "time" - "github.com/netbirdio/netbird/management/server/jwtclaims" + "github.com/stretchr/testify/assert" + + nbcontext "github.com/netbirdio/netbird/management/server/context" nbpeer "github.com/netbirdio/netbird/management/server/peer" "github.com/netbirdio/netbird/management/server/types" "github.com/netbirdio/netbird/management/server/util" ) func BenchmarkTest_GetAccountWithclaims(b *testing.B) { - claims := jwtclaims.AuthorizationClaims{ + claims := nbcontext.UserAuth{ Domain: "example.com", UserId: "pvt-domain-user", DomainCategory: types.PrivateCategory, } - publicClaims := jwtclaims.AuthorizationClaims{ + publicClaims := nbcontext.UserAuth{ Domain: "test.com", UserId: "public-domain-user", DomainCategory: types.PublicCategory,