import numpy as np
import pytest
from src.nkode_api import NKodeAPI
from src.models import NKodePolicy, KeypadSize


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


@pytest.mark.parametrize("keypad_size,passcode_len", [
    (KeypadSize(numb_of_keys=10, props_per_key=11), 4),
    (KeypadSize(numb_of_keys=10, props_per_key=12), 5),
])
def test_create_new_user_and_renew_keys(nkode_api, keypad_size, passcode_len):
    username = "test_username"
    nkode_policy = NKodePolicy()  # default policy
    customer_id = nkode_api.create_new_customer(keypad_size, nkode_policy)
    session_id, set_keypad = nkode_api.generate_signup_keypad(customer_id)
    user_passcode = set_keypad[:passcode_len]

    signup_key_selection = lambda keypad: [int(np.where(keypad == prop)[0][0]) // keypad_size.numb_of_keys for prop in user_passcode]
    set_key_selection = signup_key_selection(set_keypad)

    confirm_keypad = nkode_api.set_nkode(username, customer_id, set_key_selection, session_id)
    confirm_key_selection = signup_key_selection(confirm_keypad)
    successful_confirm = nkode_api.confirm_nkode(
        username,
        customer_id,
        confirm_key_selection,
        session_id
    )
    assert successful_confirm

    sign_in_key_selection = lambda keypad: [int(np.where(keypad ==prop)[0][0]) // keypad_size.props_per_key for prop in user_passcode]
    login_keypad = nkode_api.get_login_keypad(username, customer_id)
    login_key_selection = sign_in_key_selection(login_keypad)
    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_keypad = nkode_api.get_login_keypad(username, customer_id)
    login_key_selection = sign_in_key_selection(login_keypad)
    successful_login = nkode_api.login(customer_id, username, login_key_selection)
    assert successful_login

    login_keypad = nkode_api.get_login_keypad(username, customer_id)
    login_key_selection = sign_in_key_selection(login_keypad)
    successful_login = nkode_api.login(customer_id, username, login_key_selection)
    assert successful_login