implement generate login interface and login
This commit is contained in:
35
test/test_user_cipher_keys.py
Normal file
35
test/test_user_cipher_keys.py
Normal file
@@ -0,0 +1,35 @@
|
||||
import pytest
|
||||
from src.user_cipher_keys import UserCipherKeys, CustomerInterface
|
||||
from src.utils import generate_random_nonrepeating_list
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"passcode_len",
|
||||
[
|
||||
6
|
||||
]
|
||||
)
|
||||
def test_encode_decode_base64(passcode_len):
|
||||
data = generate_random_nonrepeating_list(passcode_len)
|
||||
encoded = UserCipherKeys.encode_base64_str(data)
|
||||
decoded = UserCipherKeys.decode_base64_str(encoded)
|
||||
assert (len(data) == len(decoded))
|
||||
assert (all(data[idx] == decoded[idx] for idx in range(passcode_len)))
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"numb_of_keys,attrs_per_key",
|
||||
[
|
||||
(10, 7,)
|
||||
])
|
||||
def test_decode_mask(numb_of_keys, attrs_per_key):
|
||||
customer = CustomerInterface.new_interface(numb_of_keys, attrs_per_key)
|
||||
passcode_entry = generate_random_nonrepeating_list(numb_of_keys*attrs_per_key, max_val=70)[:4]
|
||||
passcode_values = [customer.customer_interface[idx] for idx in passcode_entry]
|
||||
set_vals = customer.set_vals
|
||||
user_keys = UserCipherKeys.new_user_encipher_keys(numb_of_keys, attrs_per_key, set_vals)
|
||||
passcode = user_keys.encipher_nkode(passcode_entry, customer)
|
||||
|
||||
orig_passcode_set_vals = [customer.get_attr_set_val(attr) for attr in passcode_values]
|
||||
passcode_set_vals = user_keys.decipher_mask(passcode.mask, set_vals, len(passcode_entry))
|
||||
assert(len(passcode_set_vals) == len(orig_passcode_set_vals))
|
||||
assert(all(orig_passcode_set_vals[idx] == passcode_set_vals[idx] for idx in range(len(passcode_set_vals))))
|
||||
Reference in New Issue
Block a user