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


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


@pytest.mark.parametrize("numb_of_keys,attrs_per_key,user_passcode", [
    (10, 7, [3, 10, 27, 68]),
    (12, 6, [3, 10, 27, 68, 32]),
])
def test_create_new_user_and_renew_keys(nkode_api, numb_of_keys, attrs_per_key, user_passcode):
    username = "test_username"
    nkode_policy = NKodePolicy()  # default policy
    customer_id = nkode_api.create_new_customer(numb_of_keys, attrs_per_key, nkode_policy)
    session_id, set_interface = nkode_api.generate_index_interface(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 = 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

    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