created a gestion sports services class that handles the connection while the connector is dedicated to the requests
This commit is contained in:
parent
bcd8dc0733
commit
e6023e0687
12 changed files with 513 additions and 593 deletions
|
@ -3,12 +3,10 @@ from pathlib import Path
|
|||
from unittest.mock import patch
|
||||
|
||||
import aiohttp
|
||||
import config
|
||||
import pendulum
|
||||
import pytest
|
||||
from aiohttp import ClientSession
|
||||
from connectors import GestionSportsConnector
|
||||
from models import Booking, BookingFilter, Club, User
|
||||
from models import BookingFilter, Club
|
||||
from pendulum import DateTime
|
||||
from yarl import URL
|
||||
|
||||
|
@ -38,9 +36,7 @@ def retrieve_booking_datetime(
|
|||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
def test_urls():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
def test_urls(connector):
|
||||
assert (
|
||||
connector.landing_url
|
||||
== "https://toulousepadelclub.gestion-sports.com/connexion.php"
|
||||
|
@ -65,12 +61,10 @@ def test_urls():
|
|||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc", "RESOURCES_FOLDER": "/some/path"},
|
||||
{"RESOURCES_FOLDER": "/some/path"},
|
||||
clear=True,
|
||||
)
|
||||
def test_urls_payload_templates():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
def test_urls_payload_templates(connector):
|
||||
resources_folder = Path("/some", "path", "gestion-sports")
|
||||
assert connector.login_template == resources_folder / "login-payload.txt"
|
||||
assert connector.booking_template == resources_folder / "booking-payload.txt"
|
||||
|
@ -84,15 +78,8 @@ def test_urls_payload_templates():
|
|||
)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_landing_page():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
async def test_landing_page(connector):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
response = await connector.land(session)
|
||||
|
||||
|
@ -104,17 +91,8 @@ async def test_landing_page():
|
|||
assert response.cookies.get("PHPSESSID") is not None
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_login():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_login(connector, user):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
|
||||
|
@ -130,90 +108,46 @@ async def test_login():
|
|||
assert response.cookies.get("COOK_ID_USER").value == "232382"
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
def test_get_booked_court():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
|
||||
session = ClientSession()
|
||||
def test_get_booked_court(
|
||||
connector, booking_success_response, booking_failure_response
|
||||
):
|
||||
bookings = [
|
||||
(session, 601, False),
|
||||
(session, 602, False),
|
||||
(session, 603, False),
|
||||
(session, 614, False),
|
||||
(session, 605, False),
|
||||
(session, 606, True),
|
||||
(session, 607, False),
|
||||
(session, 608, False),
|
||||
(601, booking_failure_response),
|
||||
(602, booking_failure_response),
|
||||
(603, booking_failure_response),
|
||||
(614, booking_failure_response),
|
||||
(605, booking_failure_response),
|
||||
(606, booking_success_response),
|
||||
(607, booking_failure_response),
|
||||
(608, booking_failure_response),
|
||||
]
|
||||
|
||||
court = connector.get_booked_court(bookings, "padel")
|
||||
assert court.number == 9
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_book_one_court():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_book_one_court(connector, user, booking_filter):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
await connector.login(session, user)
|
||||
response, court_id, ok = await connector.send_booking_request(
|
||||
session, pendulum.parse("2024-03-21T13:30:00Z"), 610, 217
|
||||
|
||||
court_id, response = await connector.send_booking_request(
|
||||
session, pendulum.parse("2024-03-21T13:30:00+01:00"), 610, 217
|
||||
)
|
||||
|
||||
assert response.status == 200
|
||||
assert response.request_info.method == "POST"
|
||||
assert response.content_type == "text/html"
|
||||
assert response.request_info.url == URL(connector.booking_url)
|
||||
assert response.charset == "UTF-8"
|
||||
assert response.text is not None
|
||||
assert court_id == 610
|
||||
assert ok is True
|
||||
assert court_id == 610
|
||||
assert response.get("status") == "ok"
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_book():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
booking_filter = BookingFilter(
|
||||
sport_name="Padel", date=pendulum.parse("2024-03-21T13:30:00Z")
|
||||
)
|
||||
|
||||
async def test_book(connector, user, booking_filter):
|
||||
booked_court = await connector.book(user, booking_filter)
|
||||
|
||||
assert booked_court is not None
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
def test_build_booking_datetime():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
booking_filter = BookingFilter(
|
||||
sport_name="Padel", date=pendulum.parse("2024-03-21T13:30:00Z")
|
||||
)
|
||||
|
||||
def test_build_booking_datetime(connector, booking_filter):
|
||||
opening_datetime = connector.build_booking_datetime(booking_filter)
|
||||
assert opening_datetime.year == 2024
|
||||
assert opening_datetime.month == 3
|
||||
|
@ -222,19 +156,8 @@ def test_build_booking_datetime():
|
|||
assert opening_datetime.minute == 0
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@patch("pendulum.now")
|
||||
def test_wait_until_booking_time(mock_now):
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
booking_filter = BookingFilter(
|
||||
sport_name="Padel", date=pendulum.parse("2024-03-21T13:30:00Z")
|
||||
)
|
||||
|
||||
def test_wait_until_booking_time(mock_now, connector, booking_filter, club):
|
||||
booking_datetime = retrieve_booking_datetime(booking_filter, club)
|
||||
|
||||
seconds = [
|
||||
|
@ -252,17 +175,8 @@ def test_wait_until_booking_time(mock_now):
|
|||
assert pendulum.now() == booking_datetime.add(microseconds=1)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_hash():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_get_hash(connector, user):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
await connector.login(session, user)
|
||||
|
@ -281,17 +195,8 @@ def test_get_hash_input():
|
|||
assert hash_value == "63470fa38e300fd503de1ee21a71b3bdb6fb206b"
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_bookings():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_get_bookings(connector, user):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
await connector.login(session, user)
|
||||
|
@ -303,17 +208,8 @@ async def test_get_bookings():
|
|||
print(bookings)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_ongoing_bookings():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_get_ongoing_bookings(connector, user):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
await connector.login(session, user)
|
||||
|
@ -322,31 +218,13 @@ async def test_get_ongoing_bookings():
|
|||
print(bookings)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_has_user_ongoing_bookings():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_has_user_ongoing_bookings(connector, user):
|
||||
assert await connector.has_user_ongoing_booking(user)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_cancel_booking_id():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
|
||||
async def test_cancel_booking_id(connector, user):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
await connector.login(session, user)
|
||||
|
@ -358,73 +236,16 @@ async def test_cancel_booking_id():
|
|||
assert len(await connector.get_ongoing_bookings(session)) == 0
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
def test_is_booking_matching_filter():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
filter_date = pendulum.parse("2024-03-02T15:00:00+01:00")
|
||||
booking = Booking(
|
||||
id=1,
|
||||
dateResa="02/03/2024",
|
||||
startTime="15:00",
|
||||
sport="Padel",
|
||||
court="10",
|
||||
)
|
||||
booking_filter = BookingFilter(date=filter_date, sport_name="Padel")
|
||||
|
||||
assert connector.is_booking_matching_filter(booking, booking_filter)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
def test_is_booking_not_matching_filter():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
filter_date = pendulum.parse("2024-03-02T15:00:00+01:00")
|
||||
booking = Booking(
|
||||
id=1,
|
||||
dateResa="02/03/2024",
|
||||
startTime="16:00",
|
||||
sport="Padel",
|
||||
court="10",
|
||||
)
|
||||
booking_filter = BookingFilter(date=filter_date, sport_name="Padel")
|
||||
|
||||
assert not connector.is_booking_matching_filter(booking, booking_filter)
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
def test_find_court():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
|
||||
def test_find_court(connector):
|
||||
court = connector.find_court(603, "Padel")
|
||||
|
||||
assert court.number == 6
|
||||
|
||||
|
||||
@patch.dict(
|
||||
os.environ,
|
||||
{"CLUB_ID": "tpc"},
|
||||
clear=True,
|
||||
)
|
||||
@pytest.mark.asyncio
|
||||
async def test_cancel_booking():
|
||||
club = config.get_club()
|
||||
connector = GestionSportsConnector(club)
|
||||
user = User(login="padel.testing@jouf.fr", password="ridicule")
|
||||
filter_date = pendulum.parse("2024-03-21T13:30:00+01:00")
|
||||
booking_filter = BookingFilter(date=filter_date, sport_name="Padel")
|
||||
await connector.cancel_booking(user, booking_filter)
|
||||
async def test_cancel_booking(connector, user, booking_filter):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
await connector.land(session)
|
||||
await connector.login(session, user)
|
||||
await connector.cancel_booking(session, booking_filter)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue