clean up code

This commit is contained in:
2024-07-15 15:43:06 -05:00
parent 1fea44c9ba
commit 2da3e7a443
3 changed files with 4 additions and 4 deletions

View File

@@ -45,4 +45,3 @@ class CustomerInterface(BaseModel):
def get_set_index(self, set_val: int) -> int:
assert(set_val in self.set_vals)
return self.set_vals.index(set_val)

View File

@@ -50,6 +50,7 @@ class CustomerDBModel(BaseModel):
return False
class SessionCacheModel(BaseModel):
session_id: UUID
set_interface: list[int] | None = None

View File

@@ -1,5 +1,5 @@
import pytest
from src.pseudo_nkode_api import PseudoNKodeAPI, CustomerDBModel
from src.pseudo_nkode_api import PseudoNKodeAPI
@pytest.fixture()
@@ -31,4 +31,4 @@ def test_create_new_user(pseudo_nkode_api, numb_keys, attrs_per_key, user_passco
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)
assert successful_login