109 lines
2.7 KiB
Python
109 lines
2.7 KiB
Python
import asyncio
|
|
import os
|
|
from unittest.mock import patch
|
|
from urllib.parse import urljoin
|
|
|
|
from aioresponses import aioresponses
|
|
|
|
from resa_padel import booking
|
|
from tests import fixtures
|
|
from tests.fixtures import (
|
|
a_booking_failure_response,
|
|
a_booking_filter,
|
|
a_booking_success_response,
|
|
a_club,
|
|
a_user,
|
|
)
|
|
|
|
login = "user"
|
|
password = "password"
|
|
club_id = "88"
|
|
court_id = "11"
|
|
|
|
|
|
def mock_successful_connection(aio_mock, url):
|
|
aio_mock.get(
|
|
url,
|
|
status=200,
|
|
headers={"Set-Cookie": f"connection_called=True; Domain={url}"},
|
|
)
|
|
|
|
|
|
def mock_successful_login(aio_mock, url):
|
|
aio_mock.post(
|
|
url,
|
|
status=200,
|
|
headers={"Set-Cookie": f"login_called=True; Domain={url}"},
|
|
)
|
|
|
|
|
|
def mock_booking(aio_mock, url, response):
|
|
aio_mock.post(
|
|
url,
|
|
status=200,
|
|
headers={"Set-Cookie": f"booking_called=True; Domain={url}"},
|
|
body=response,
|
|
)
|
|
|
|
|
|
def mock_rest_api_from_connection_to_booking(
|
|
aio_mock, url, a_booking_failure_response, a_booking_success_response
|
|
):
|
|
connexion_url = urljoin(url, "/connexion.php?")
|
|
mock_successful_connection(aio_mock, connexion_url)
|
|
login_url = urljoin(url, "/connexion.php?")
|
|
mock_successful_login(aio_mock, login_url)
|
|
booking_url = urljoin(url, "/membre/reservation.html?")
|
|
mock_booking(aio_mock, booking_url, a_booking_failure_response)
|
|
mock_booking(aio_mock, booking_url, a_booking_success_response)
|
|
|
|
|
|
def test_booking_does_the_rights_calls(
|
|
a_booking_success_response,
|
|
a_booking_failure_response,
|
|
a_user,
|
|
a_club,
|
|
a_booking_filter,
|
|
):
|
|
# mock connection to the booking platform
|
|
with aioresponses() as aio_mock:
|
|
mock_rest_api_from_connection_to_booking(
|
|
aio_mock,
|
|
fixtures.url,
|
|
a_booking_failure_response,
|
|
a_booking_success_response,
|
|
)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
court_booked = loop.run_until_complete(
|
|
booking.book(a_club, a_user, a_booking_filter)
|
|
)
|
|
assert court_booked == a_club.courts_ids[1]
|
|
|
|
|
|
@patch.dict(
|
|
os.environ,
|
|
{
|
|
"LOGIN": login,
|
|
"PASSWORD": password,
|
|
"CLUB_ID": club_id,
|
|
"CLUB_URL": fixtures.url,
|
|
"COURT_IDS": "7,8,10",
|
|
"SPORT_ID": "217",
|
|
"DATE_TIME": "2024-04-23T15:00:00Z",
|
|
},
|
|
clear=True,
|
|
)
|
|
def test_main(a_booking_success_response, a_booking_failure_response):
|
|
|
|
with aioresponses() as aio_mock:
|
|
mock_rest_api_from_connection_to_booking(
|
|
aio_mock,
|
|
fixtures.url,
|
|
a_booking_failure_response,
|
|
a_booking_success_response,
|
|
)
|
|
|
|
court_booked = booking.main()
|
|
assert court_booked == 8
|