nhyatt/putty-source
1
0
Fork 0
mirror of https://git.tartarus.org/simon/putty.git synced 2025-04-17 19:18:06 -05:00
Code Issues Projects Releases Wiki Activity

testcrypt: permit multiple OO function prefixes for a type.

Browse Source 
This means if I have functions like foo_subfoo_bar and foo_baz that
both operate on a foo, the Python testcrypt system can translate both
into .bar() and .baz() methods on the object, even though they don't
start with the same prefix.
This commit is contained in:
Simon Tatham 2022-04-15 17:18:58 +01:00
parent 31db2e67bb
commit e66e1ebeae
1 changed files with 20 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
test/testcrypt.py
View File

@ -87,19 +87,20 @@ class ChildProcess(object):
childprocess = ChildProcess() childprocess = ChildProcess()
method_prefixes = { method_prefixes = {
'val_wpoint': 'ecc_weierstrass_', 'val_wpoint': ['ecc_weierstrass_'],
'val_mpoint': 'ecc_montgomery_', 'val_mpoint': ['ecc_montgomery_'],
'val_epoint': 'ecc_edwards_', 'val_epoint': ['ecc_edwards_'],
'val_hash': 'ssh_hash_', 'val_hash': ['ssh_hash_'],
'val_mac': 'ssh2_mac_', 'val_mac': ['ssh2_mac_'],
'val_key': 'ssh_key_', 'val_key': ['ssh_key_'],
'val_cipher': 'ssh_cipher_', 'val_cipher': ['ssh_cipher_'],
'val_dh': 'dh_', 'val_dh': ['dh_'],
'val_ecdh': 'ssh_ecdhkex_', 'val_ecdh': ['ssh_ecdhkex_'],
'val_rsakex': 'ssh_rsakex_', 'val_rsakex': ['ssh_rsakex_'],
'val_prng': 'prng_', 'val_prng': ['prng_'],
'val_pcs': 'pcs_', 'val_pcs': ['pcs_'],
'val_pockle': 'pockle_', 'val_pockle': ['pockle_'],
'val_ntruencodeschedule': ['ntru_encode_schedule_', 'ntru_'],
} }
method_lists = {t: [] for t in method_prefixes} method_lists = {t: [] for t in method_prefixes}
@ -420,10 +421,12 @@ def _setup(scope):
scope[function] = func scope[function] = func
if len(argtypes) > 0: if len(argtypes) > 0:
t = argtypes[0][0] t = argtypes[0][0]
if (t in method_prefixes and if t in method_prefixes:
function.startswith(method_prefixes[t])): for prefix in method_prefixes[t]:
methodname = function[len(method_prefixes[t]):] if function.startswith(prefix):
method_lists[t].append((methodname, func)) methodname = function[len(prefix):]
method_lists[t].append((methodname, func))
break
_setup(globals()) _setup(globals())
del _setup del _setup