Skip to content

Commit

Permalink
Merge pull request #68 from MErenTalan/main
Browse files Browse the repository at this point in the history
Mocking input for verification code
  • Loading branch information
subzeroid authored Jun 6, 2024
2 parents 40c34d4 + c3195e7 commit 8fd8c2c
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions routers/auth.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import json
from unittest.mock import patch
from typing import Optional, Dict
from fastapi import APIRouter, Depends, Form
from dependencies import ClientStorage, get_clients
Expand Down Expand Up @@ -30,11 +31,10 @@ async def auth_login(username: str = Form(...),
if timezone != "":
cl.set_timezone_offset(timezone)

result = cl.login(
username,
password,
verification_code=verification_code
)
# We're mocking the input
with patch('builtins.input', return_value=verification_code):
result = cl.login(username, password)

if result:
clients.set(cl)
return cl.sessionid
Expand Down

0 comments on commit 8fd8c2c

Please sign in to comment.