69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
|
import os, uuid
|
||
|
|
from random import randint
|
||
|
|
from time import sleep
|
||
|
|
from enum import IntEnum, auto
|
||
|
|
from nacl.signing import SigningKey, VerifyKey
|
||
|
|
from nacl.exceptions import BadSignatureError
|
||
|
|
import nacl
|
||
|
|
import secrets
|
||
|
|
import onionrblocks
|
||
|
|
|
||
|
|
|
||
|
|
TEST_DIR = 'testdata/%s-%s' % (str(uuid.uuid4())[:6], os.path.basename(__file__)) + '/'
|
||
|
|
print("Test directory:", TEST_DIR)
|
||
|
|
os.environ["ONIONR_HOME"] = TEST_DIR
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
sys.path.append(".")
|
||
|
|
sys.path.append('static-data/default-plugins/wot/')
|
||
|
|
sys.path.append("src/")
|
||
|
|
from wot.identityprocessing import process_identity_revoke
|
||
|
|
from wot import identity
|
||
|
|
from wot.identity.identityset import identities
|
||
|
|
|
||
|
|
|
||
|
|
class WotCommand(IntEnum):
|
||
|
|
TRUST = 1
|
||
|
|
REVOKE_TRUST = auto()
|
||
|
|
ANNOUNCE = auto()
|
||
|
|
REVOKE = auto()
|
||
|
|
|
||
|
|
|
||
|
|
class TestRevokeIdentityPayload(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_revoke_identity_invalid(self):
|
||
|
|
identities.clear()
|
||
|
|
|
||
|
|
signing_key = SigningKey.generate()
|
||
|
|
main_iden = identity.Identity(signing_key, "test")
|
||
|
|
identities.add(main_iden)
|
||
|
|
|
||
|
|
wot_cmd = int(WotCommand.REVOKE).to_bytes(1, 'big')
|
||
|
|
|
||
|
|
signed = signing_key.sign(wot_cmd + bytes(main_iden.key))
|
||
|
|
revoke_payload = wot_cmd + bytes(signing_key.verify_key) + signed.signature
|
||
|
|
|
||
|
|
self.assertRaises(nacl.exceptions.Inv process_identity_revoke(revoke_payload)
|
||
|
|
|
||
|
|
self.assertEqual(len(identities), 1)
|
||
|
|
|
||
|
|
def test_revoke_identity_payload(self):
|
||
|
|
identities.clear()
|
||
|
|
|
||
|
|
signing_key = SigningKey.generate()
|
||
|
|
main_iden = identity.Identity(signing_key, "test")
|
||
|
|
identities.add(main_iden)
|
||
|
|
|
||
|
|
wot_cmd = int(WotCommand.REVOKE).to_bytes(1, 'big')
|
||
|
|
|
||
|
|
signed = signing_key.sign(wot_cmd + bytes(main_iden.key))
|
||
|
|
revoke_payload = wot_cmd + bytes(signing_key.verify_key) + signed.signature
|
||
|
|
|
||
|
|
process_identity_revoke(revoke_payload)
|
||
|
|
|
||
|
|
self.assertEqual(len(identities), 0)
|
||
|
|
|
||
|
|
|
||
|
|
unittest.main()
|