refactor code

This commit is contained in:
2024-07-16 13:20:52 -05:00
parent ec398c064a
commit d23ad7ac2b
4 changed files with 49 additions and 48 deletions

1
.gitignore vendored
View File

@@ -1 +1,2 @@
.idea
__pycache__

View File

@@ -9,7 +9,7 @@ from src.user_interface import UserInterface
from src.customer_interface import CustomerInterface
class PseudoNKodeAPI(BaseModel):
class NKodeAPI(BaseModel):
customers: dict[UUID, CustomerDBModel] = {}
sessions: dict[UUID, SessionCacheModel] = {}

47
test/test_nkode_api.py Normal file
View File

@@ -0,0 +1,47 @@
import pytest
from nkode_api import NKodeAPI
@pytest.fixture()
def nkode_api() -> NKodeAPI:
return NKodeAPI()
@pytest.mark.parametrize("numb_keys,attrs_per_key,user_passcode", [
(10, 7, [3, 10, 27, 68]),
])
def test_create_new_user_and_renew_keys(nkode_api, numb_keys, attrs_per_key, user_passcode):
username = "test_username"
customer = nkode_api.create_new_customer(numb_keys, attrs_per_key)
session_id, set_interface = 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 = nkode_api.set_nkode(username, customer.customer_id, set_key_selection, session_id)
confirm_key_selection = key_selection(confirm_interface)
response = nkode_api.confirm_nkode(
username,
customer.customer_id,
confirm_key_selection,
session_id
)
assert ("success" == response)
login_interface = nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface)
successful_login = nkode_api.login(customer.customer_id, username, login_key_selection)
assert successful_login
successful_renew = nkode_api.renew_keys(customer.customer_id)
assert successful_renew
login_interface = nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface)
successful_login = nkode_api.login(customer.customer_id, username, login_key_selection)
assert successful_login
login_interface = nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface)
successful_login = nkode_api.login(customer.customer_id, username, login_key_selection)
assert successful_login

View File

@@ -1,47 +0,0 @@
import pytest
from src.pseudo_nkode_api import PseudoNKodeAPI
@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_and_renew_keys(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)
login_interface = pseudo_nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface)
successful_login = pseudo_nkode_api.login(customer.customer_id, username, login_key_selection)
assert successful_login
successful_renew = pseudo_nkode_api.renew_keys(customer.customer_id)
assert successful_renew
login_interface = pseudo_nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface)
successful_login = pseudo_nkode_api.login(customer.customer_id, username, login_key_selection)
assert successful_login
login_interface = pseudo_nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface)
successful_login = pseudo_nkode_api.login(customer.customer_id, username, login_key_selection)
assert successful_login