def test_get():
jwt_bundle = JwtBundle(trust_domain_1, authorities)
jwt_bundle_set = JwtBundleSet({trust_domain_1: jwt_bundle})
res = jwt_bundle_set.get(trust_domain_1)
assert res == jwt_bundle
assert res.trust_domain() == jwt_bundle.trust_domain()
def put(self, jwt_bundle: JwtBundle):
"""Adds a new bundle into the set.
If a bundle already exists for the trust domain, the existing bundle is
replaced.
Args:
jwt_bundle: The new JwtBundle to add.
"""
with self.lock:
self._bundles[jwt_bundle.trust_domain()] = jwt_bundle
def test_create_jwt_bundle_no_authorities():
jwt_bundle = JwtBundle(trust_domain, None)
assert jwt_bundle.trust_domain() == trust_domain
assert isinstance(jwt_bundle.jwt_authorities(), dict)
assert len(jwt_bundle.jwt_authorities().keys()) == 0
def test_create_jwt_bundle():
jwt_bundle = JwtBundle(trust_domain, authorities)
assert jwt_bundle.trust_domain() == trust_domain
assert len(jwt_bundle.jwt_authorities().keys()) == len(authorities.keys())