def test_keystore_pairkeys():
ks = KeyStore(None)
ks.add_key("a1b2c3d4", "hmac", "sig")
ks.add_key("a1b2c3d4", "hmac", "ver")
ks.add_key("e5f6g7h8", "hmac", "sig", "http://www.example.org")
ks.add_key("e5f6g7h8", "hmac", "ver", "http://www.example.org")
ks.add_key("-rsa-key-", "rsa", "enc", "http://www.example.org")
ks.add_key("-rsa-key-", "rsa", "dec", "http://www.example.org")
ks.add_key("i9j10k11l12", "hmac", "sig", "http://www.example.com")
ks.add_key("i9j10k11l12", "hmac", "ver", "http://www.example.com")
collection = ks.pairkeys("http://www.example.org")
assert _eq(collection.keys(), ["sig", "ver", "enc", "dec"])
def test_keystore_remove_key_type():
ks = KeyStore(None)
ks.add_key("a1b2c3d4", "hmac", "sig")
ks.add_key("a1b2c3d4", "hmac", "ver")
ks.add_key("e5f6g7h8", "hmac", "sig", "http://www.example.org")
ks.add_key("e5f6g7h8", "hmac", "ver", "http://www.example.org")
ks.add_key("-rsa-key-", "rsa", "enc", "http://www.example.org")
ks.add_key("-rsa-key-", "rsa", "dec", "http://www.example.org")
ks.add_key("i9j10k11l12", "hmac", "sig", "http://www.example.com")
ks.add_key("i9j10k11l12", "hmac", "ver", "http://www.example.com")
ks.remove_key_type("rsa", "http://www.example.org")
coll = ks.keys_by_owner("http://www.example.org")
assert _eq(coll.keys(), ["sig", "ver"])
ks = KeyStore(None)
ks.add_key("a1b2c3d4", "hmac", "sig")
ks.add_key("a1b2c3d4", "hmac", "ver")
ks.add_key("e5f6g7h8", "hmac", "sig", "http://www.example.org")
ks.add_key("e5f6g7h8", "hmac", "ver", "http://www.example.org")
ks.add_key("-rsa-key-", "rsa", "enc", "http://www.example.org")
ks.add_key("-rsa-key-", "rsa", "dec", "http://www.example.org")
ks.add_key("i9j10k11l12", "hmac", "sig", "http://www.example.com")
ks.add_key("i9j10k11l12", "hmac", "ver", "http://www.example.com")
ks.remove_key_type("rsa", "http://www.example.org")
coll = ks.keys_by_owner("http://www.example.org")
assert _eq(coll.keys(), ["sig", "ver"])
KEYSTORE = KeyStore(None)
KEYSTORE.add_key("a1b2c3d4", "hmac", "sig")
KEYSTORE.add_key("a1b2c3d4", "hmac", "ver")
KEYSTORE.add_key("e5f6g7h8", "hmac", "sig", "http://www.example.org")
KEYSTORE.add_key("e5f6g7h8", "hmac", "ver", "http://www.example.org")
KEYSTORE.add_key("-rsa-key-", "rsa", "enc", "http://www.example.org")
KEYSTORE.add_key("-rsa-key-", "rsa", "dec", "http://www.example.org")
KEYSTORE.add_key("i9j10k11l12", "hmac", "sig", "http://www.example.com")
KEYSTORE.add_key("i9j10k11l12", "hmac", "ver", "http://www.example.com")
def test_keystore_collect_keys():
col = KEYSTORE.collect_keys("http://www.example.org/oic")
print col
assert col == {'hmac': ['e5f6g7h8', 'a1b2c3d4']}
