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())