ensure Token implements the new AttributesAccessor interface

xmidt-org · Aug 13, 2024 · a287834 · a287834
1 parent ecc370c
commit a287834
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/basculejwt/token.go b/basculejwt/token.go
@@ -86,6 +86,10 @@ func (t token) Capabilities() (caps []string) {
 	return
 }
 
+func (t token) Get(key string) (any, bool) {
+	return t.jwt.Get(key)
+}
+
 // tokenParser is the canonical parser for bascule that deals with JWTs.
 // This parser does not use the source.
 type tokenParser struct {

diff --git a/basculejwt/token_test.go b/basculejwt/token_test.go
@@ -130,6 +130,11 @@ func (suite *TokenTestSuite) TestTokenParser() {
 		suite.Equal(suite.capabilities, caps)
 		suite.True(ok)
 
+		suite.Require().Implements((*bascule.AttributesAccessor)(nil), token)
+		v, ok := bascule.GetAttribute[string](token.(bascule.AttributesAccessor), "version")
+		suite.True(ok)
+		suite.Equal(suite.version, v)
+
 		suite.Require().Implements((*Claims)(nil), token)
 		claims := token.(Claims)
 		suite.Equal(suite.audience, claims.Audience())