fix: mypy errors

fastapi-admin
Pierre-Louis Guhur 1 year ago
parent 18e535c7be
commit 72db86e958

@ -8,10 +8,10 @@ from jose import jwe, jws
from jose.exceptions import JWEError, JWSError
from . import crud, models, schemas, errors
from .database import get_db, engine
from .database import get_db, engine, Base
from .settings import settings
models.Base.metadata.create_all(bind=engine)
Base.metadata.create_all(bind=engine)
app = FastAPI()

@ -40,17 +40,17 @@ def test_ballot_token():
Can verify ballot tokens with MANY different tokens
"""
vote_ids = list(range(1000))
election_id = 0
token = create_ballot_token(vote_ids, election_id)
election_ref = "qwertyuiop"
token = create_ballot_token(vote_ids, election_ref)
data = jws_verify(token)
assert data == {"votes": vote_ids, "election": election_id}
assert data == {"votes": vote_ids, "election": election_ref}
def test_admin_token():
"""
Can verify ballot tokens with MANY different tokens
"""
election_id = 0
token = create_admin_token(election_id)
election_ref = "qwertyuiop"
token = create_admin_token(election_ref)
data = jws_verify(token)
assert data == {"admin": True, "election": election_id}
assert data == {"admin": True, "election": election_ref}

@ -3,3 +3,5 @@ requests==2.28.1
pytest==7.2.0
mypy==0.991
black==22.10.0
alembic==1.8.1
types-python-jose==3.3.4

Loading…
Cancel
Save