import pytest
from src.pseudo_nkode_api import PseudoNKodeAPI, CustomerDBModel


@pytest.fixture()
def pseudo_nkode_api() -> PseudoNKodeAPI:
    return PseudoNKodeAPI()


@pytest.mark.parametrize("numb_keys,attrs_per_key,user_passcode", [
    (10, 7, [3, 10, 27, 68]),
])
def test_create_new_user(pseudo_nkode_api, numb_keys, attrs_per_key, user_passcode):
    username = "test_username"
    customer = pseudo_nkode_api.create_new_customer(numb_keys, attrs_per_key)
    session_id, set_interface = pseudo_nkode_api.generate_index_interface(customer.customer_id)

    key_selection = lambda interface: [interface.index(attr) % attrs_per_key for attr in user_passcode]
    set_key_selection = key_selection(set_interface)

    confirm_interface = pseudo_nkode_api.set_nkode(username, customer.customer_id, set_key_selection, session_id)
    confirm_key_selection = key_selection(confirm_interface)
    response = pseudo_nkode_api.confirm_nkode(
        username,
        customer.customer_id,
        confirm_key_selection,
        session_id
    )
    assert ("success" == response)