mirror of
https://github.com/savoirfairelinux/opendht.git
synced 2026-11-08 04:21:51 +00:00
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
import unittest
|
|
import opendht as dht
|
|
|
|
class OpenDhtTester(unittest.TestCase):
|
|
|
|
# test that DhtRunner can be instatiated and deleted without throwing
|
|
def test_instance(self):
|
|
for i in range(10):
|
|
r = dht.DhtRunner()
|
|
r.run()
|
|
del r
|
|
|
|
# test that bootstraping works (raw address)
|
|
def test_bootstrap(self):
|
|
a = dht.DhtRunner()
|
|
a.run()
|
|
b = dht.DhtRunner()
|
|
b.run()
|
|
self.assertTrue(b.ping(a.getBound()))
|
|
|
|
def test_crypto(self):
|
|
i = dht.Identity.generate("id")
|
|
message = dht.InfoHash.getRandom().toString()
|
|
encrypted = i.publickey.encrypt(message)
|
|
decrypted = i.key.decrypt(encrypted)
|
|
self.assertTrue(message == decrypted)
|
|
|
|
def test_crypto_ec(self):
|
|
key = dht.PrivateKey.generateEC()
|
|
cert = dht.Certificate.generate(key, "CA", is_ca=True)
|
|
ca_id = dht.Identity(key, cert)
|
|
self.assertTrue(cert.getId() == key.getPublicKey().getId())
|
|
key2 = dht.PrivateKey.generateEC()
|
|
cert2 = dht.Certificate.generate(key2, "cert", ca_id)
|
|
trust = dht.TrustList()
|
|
trust.add(cert)
|
|
self.assertTrue(trust.verify(cert2))
|
|
|
|
def test_trust(self):
|
|
main_id = dht.Identity.generate("id_1")
|
|
sub_id1 = dht.Identity.generate("sid_1", main_id)
|
|
sub_id2 = dht.Identity.generate("sid_2", main_id)
|
|
main_id.certificate.revoke(main_id.key, sub_id2.certificate)
|
|
main_id2 = dht.Identity.generate("id_2")
|
|
trust = dht.TrustList()
|
|
trust.add(main_id.certificate)
|
|
self.assertTrue(trust.verify(main_id.certificate))
|
|
self.assertTrue(trust.verify(sub_id1.certificate))
|
|
self.assertFalse(trust.verify(sub_id2.certificate))
|
|
self.assertFalse(trust.verify(main_id2.certificate))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|