Skip to content

Commit aefd95a

Browse files
authored
Merge pull request openwallet-foundation#1829 from andrewwhitehead/upd/pyjwt
Update pyjwt to 2.4
2 parents 8421116 + 4356060 commit aefd95a

3 files changed

Lines changed: 10 additions & 10 deletions

File tree

aries_cloudagent/multitenant/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ async def create_auth_token(
366366

367367
jwt_payload["wallet_key"] = wallet_key
368368

369-
token = jwt.encode(jwt_payload, jwt_secret, algorithm="HS256").decode()
369+
token = jwt.encode(jwt_payload, jwt_secret, algorithm="HS256")
370370

371371
# Store iat for verification later on
372372
wallet_record.jwt_iat = iat

aries_cloudagent/multitenant/tests/test_base.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -430,7 +430,7 @@ async def test_create_auth_token_managed(self):
430430

431431
expected_token = jwt.encode(
432432
{"wallet_id": wallet_record.wallet_id, "iat": iat}, "very_secret_jwt"
433-
).decode()
433+
)
434434

435435
with async_mock.patch.object(test_module, "datetime") as mock_datetime:
436436
mock_datetime.utcnow.return_value = utc_now
@@ -459,7 +459,7 @@ async def test_create_auth_token_unmanaged(self):
459459
"wallet_key": "test_key",
460460
},
461461
"very_secret_jwt",
462-
).decode()
462+
)
463463

464464
with async_mock.patch.object(test_module, "datetime") as mock_datetime:
465465
mock_datetime.utcnow.return_value = utc_now
@@ -471,7 +471,7 @@ async def test_create_auth_token_unmanaged(self):
471471
async def test_get_profile_for_token_invalid_token_raises(self):
472472
self.profile.settings["multitenant.jwt_secret"] = "very_secret_jwt"
473473

474-
token = jwt.encode({"wallet_id": "test"}, "some_random_key").decode()
474+
token = jwt.encode({"wallet_id": "test"}, "some_random_key")
475475

476476
with self.assertRaises(jwt.InvalidTokenError):
477477
await self.manager.get_profile_for_token(self.profile.context, token)
@@ -486,7 +486,7 @@ async def test_get_profile_for_token_wallet_key_missing_raises(self):
486486
await wallet_record.save(session)
487487
token = jwt.encode(
488488
{"wallet_id": wallet_record.wallet_id}, "very_secret_jwt", algorithm="HS256"
489-
).decode()
489+
)
490490

491491
with self.assertRaises(WalletKeyMissingError):
492492
await self.manager.get_profile_for_token(self.profile.context, token)
@@ -503,7 +503,7 @@ async def test_get_profile_for_token_managed_wallet_no_iat(self):
503503

504504
token = jwt.encode(
505505
{"wallet_id": wallet_record.wallet_id}, "very_secret_jwt", algorithm="HS256"
506-
).decode()
506+
)
507507

508508
with async_mock.patch.object(
509509
BaseMultitenantManager, "get_wallet_profile"
@@ -540,7 +540,7 @@ async def test_get_profile_for_token_managed_wallet_iat(self):
540540
{"wallet_id": wallet_record.wallet_id, "iat": iat},
541541
"very_secret_jwt",
542542
algorithm="HS256",
543-
).decode()
543+
)
544544

545545
with async_mock.patch.object(
546546
BaseMultitenantManager, "get_wallet_profile"
@@ -578,7 +578,7 @@ async def test_get_profile_for_token_managed_wallet_x_iat_no_match(self):
578578
{"wallet_id": wallet_record.wallet_id, "iat": 200},
579579
"very_secret_jwt",
580580
algorithm="HS256",
581-
).decode()
581+
)
582582

583583
with async_mock.patch.object(
584584
BaseMultitenantManager, "get_wallet_profile"
@@ -614,7 +614,7 @@ async def test_get_profile_for_token_unmanaged_wallet(self):
614614
{"wallet_id": wallet_record.wallet_id, "wallet_key": "wallet_key"},
615615
"very_secret_jwt",
616616
algorithm="HS256",
617-
).decode()
617+
)
618618

619619
with async_mock.patch.object(
620620
BaseMultitenantManager, "get_wallet_profile"

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ packaging~=20.4
1919
pyld~=2.0.3
2020
pyyaml~=5.4.0
2121
ConfigArgParse~=1.5.3
22-
pyjwt~=1.7.1
22+
pyjwt~=2.4.0
2323
pydid~=0.3.3
2424
jsonpath_ng==1.5.2
2525
pytz~=2021.1

0 commit comments

Comments
 (0)