import pytest
from nkode_api import NKodeAPI
from src.models import NKodePolicy, KeypadSize


@pytest.fixture()
def nkode_api() -> NKodeAPI:
    return NKodeAPI()


@pytest.mark.parametrize("keypad_size,passocode_len", [
    (KeypadSize(numb_of_keys=10, attrs_per_key=11), 4),
    (KeypadSize(numb_of_keys=10, attrs_per_key=12), 5),
])
def test_create_new_user_and_renew_keys(nkode_api, keypad_size, passocode_len):
    username = "test_username"
    nkode_policy = NKodePolicy()  # default policy
    customer_id = nkode_api.create_new_customer(keypad_size, nkode_policy)
    session_id, set_interface = nkode_api.generate_index_interface(customer_id)
    user_passcode = set_interface[:passocode_len]
    key_selection = lambda interface: [interface.index(attr) // keypad_size.numb_of_keys for attr in user_passcode]
    set_key_selection = key_selection(set_interface)

    confirm_interface = nkode_api.set_nkode(username, customer_id, set_key_selection, session_id)
    confirm_key_selection = key_selection(confirm_interface)
    successful_confirm = nkode_api.confirm_nkode(
        username,
        customer_id,
        confirm_key_selection,
        session_id
    )
    assert successful_confirm

    key_selection = lambda interface: [interface.index(attr) // keypad_size.attrs_per_key for attr in user_passcode]
    login_interface = nkode_api.get_login_index_interface(username, customer_id)
    login_key_selection = key_selection(login_interface)
    successful_login = nkode_api.login(customer_id, username, login_key_selection)
    assert successful_login

    successful_renew = nkode_api.renew_keys(customer_id)
    assert successful_renew

    login_interface = nkode_api.get_login_index_interface(username, customer_id)
    login_key_selection = key_selection(login_interface)
    successful_login = nkode_api.login(customer_id, username, login_key_selection)
    assert successful_login

    login_interface = nkode_api.get_login_index_interface(username, customer_id)
    login_key_selection = key_selection(login_interface)
    successful_login = nkode_api.login(customer_id, username, login_key_selection)
    assert successful_login