test case for session admin without account

0xsequence · Oct 23, 2024 · 753f269 · 753f269
1 parent fc5febd
commit 753f269
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/middleware.go b/middleware.go
@@ -4,10 +4,9 @@ import (
 	"errors"
 	"net/http"
 
+	"github.com/0xsequence/authcontrol/proto"
 	"github.com/go-chi/jwtauth/v5"
 	"github.com/lestrrat-go/jwx/v2/jwt"
-
-	"github.com/0xsequence/authcontrol/proto"
 )
 
 type Options struct {

diff --git a/middleware_test.go b/middleware_test.go
@@ -10,12 +10,11 @@ import (
 	"testing"
 	"time"
 
+	"github.com/0xsequence/authcontrol"
+	"github.com/0xsequence/authcontrol/proto"
 	"github.com/go-chi/chi/v5"
 	"github.com/go-chi/jwtauth/v5"
 	"github.com/stretchr/testify/assert"
-
-	"github.com/0xsequence/authcontrol"
-	"github.com/0xsequence/authcontrol/proto"
 )
 
 type mockStore map[string]bool
@@ -104,6 +103,7 @@ func TestSession(t *testing.T) {
 		{Session: proto.SessionType_User},
 		{Session: proto.SessionType_User, Admin: true},
 		{Session: proto.SessionType_Admin},
+		{Session: proto.SessionType_Admin, Admin: true},
 		{Session: proto.SessionType_Admin, AccessKey: AccessKey},
 		{Session: proto.SessionType_Service},
 		{Session: proto.SessionType_Service, AccessKey: AccessKey},
@@ -128,6 +128,9 @@ func TestSession(t *testing.T) {
 						claims = map[string]any{"account": address}
 					case proto.SessionType_Admin:
 						claims = map[string]any{"account": WalletAddress, "admin": true}
+						if tc.Admin {
+							claims = map[string]any{"admin": true}
+						}
 					case proto.SessionType_Service:
 						claims = map[string]any{"service": ServiceName}
 					}