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,max_nkode_len",
    [
        (10, 7, 10),
        (9, 7, 10),
        (8, 7, 12),
    ])
def test_decode_mask(numb_of_keys, attrs_per_key, max_nkode_len):
    customer = CustomerInterface.new(numb_of_keys, attrs_per_key)
    passcode_entry = generate_random_nonrepeating_list(
        numb_of_keys * attrs_per_key,
        max_val=(numb_of_keys*attrs_per_key))[:4]
    passcode_values = [customer.attr_vals[idx] for idx in passcode_entry]
    set_vals = customer.set_vals
    user_keys = UserCipherKeys.new(numb_of_keys, attrs_per_key, set_vals, max_nkode_len)
    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))))