import os 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, User from yarl import URL from tests import utils @patch.dict( os.environ, {"CLUB_ID": "tpc"}, clear=True, ) def test_urls(): club = config.get_club() connector = GestionSportsConnector(club) assert ( connector.landing_url == "https://toulousepadelclub.gestion-sports.com/connexion.php" ) assert ( connector.login_url == "https://toulousepadelclub.gestion-sports.com/connexion.php" ) assert ( connector.booking_url == "https://toulousepadelclub.gestion-sports.com/membre/reservation.html" ) assert ( connector.user_bookings_url == "https://toulousepadelclub.gestion-sports.com/membre/mesresas.html" ) assert ( connector.booking_cancellation_url == "https://toulousepadelclub.gestion-sports.com/membre/mesresas.html" ) @patch.dict( os.environ, {"CLUB_ID": "tpc", "RESOURCES_FOLDER": "/some/path"}, clear=True, ) def test_urls_payload_templates(): club = config.get_club() connector = GestionSportsConnector(club) 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" assert ( connector.user_bookings_template == resources_folder / "user-bookings-payload.txt" ) assert ( connector.booking_cancellation_template == resources_folder / "booking-cancellation-payload.txt" ) @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 with aiohttp.ClientSession() as session: response = await connector.land(session) assert response.status == 200 assert response.request_info.method == "GET" assert response.content_type == "text/html" assert response.request_info.url == URL(connector.landing_url) assert response.charset == "UTF-8" 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 with aiohttp.ClientSession() as session: await connector.land(session) response = await connector.login(session, user) assert response.status == 200 assert response.request_info.method == "POST" assert response.content_type == "text/html" assert response.request_info.url == URL(connector.landing_url) assert response.charset == "UTF-8" assert response.cookies.get("COOK_COMPTE") is not None assert response.cookies.get("COOK_ID_CLUB").value == "88" 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() 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), ] 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 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 ) 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 @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") ) 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") ) opening_datetime = connector.build_booking_datetime(booking_filter) assert opening_datetime.year == 2024 assert opening_datetime.month == 3 assert opening_datetime.day == 14 assert opening_datetime.hour == 0 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") ) booking_datetime = utils.retrieve_booking_datetime(booking_filter, club) seconds = [ booking_datetime.subtract(seconds=3), booking_datetime.subtract(seconds=2), booking_datetime.subtract(seconds=1), booking_datetime, booking_datetime.add(microseconds=1), booking_datetime.add(microseconds=2), ] mock_now.side_effect = seconds connector.wait_until_booking_time(booking_filter) 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 with aiohttp.ClientSession() as session: await connector.land(session) await connector.login(session, user) hash_value = await connector.send_hash_request(session) assert hash_value is not None def test_get_hash_input(): resources_folder = Path(__file__).parent / "data" html_file = resources_folder / "user_bookings_html_response.html" html = html_file.read_text(encoding="utf-8") hash_value = GestionSportsConnector.get_hash_input(html) 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 with aiohttp.ClientSession() as session: await connector.land(session) await connector.login(session, user) hash_value = await connector.send_hash_request(session) payload = f"ajax=loadResa&hash={hash_value}" bookings = await connector.send_user_bookings_request(session, payload) 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 with aiohttp.ClientSession() as session: await connector.land(session) await connector.login(session, user) bookings = await connector.get_ongoing_bookings(session) 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") 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 with aiohttp.ClientSession() as session: await connector.land(session) await connector.login(session, user) ongoing_bookings = await connector.get_ongoing_bookings(session) booking_id = ongoing_bookings[0].id response = await connector.cancel_booking_id(user, 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) 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)