refactored code to place the booking specific actions inside a gestion-sports element

tests/utils.py

@@ -0,0 +1,91 @@
+from urllib.parse import urljoin
+
+from models import BookingFilter, Club
+from pendulum import DateTime
+
+
+def mock_successful_connection(aio_mock, url):
+    """
+    Mock a call to the connection endpoint
+
+    :param aio_mock: the aioresponses mock object
+    :param url: the URL of the connection endpoint
+    """
+    aio_mock.get(
+        url,
+        status=200,
+        headers={"Set-Cookie": f"connection_called=True; Domain={url}"},
+    )
+
+
+def mock_successful_login(aio_mock, url):
+    """
+    Mock a call to the login endpoint
+
+    :param aio_mock: the aioresponses mock object
+    :param url: the URL of the login endpoint
+    """
+    aio_mock.post(
+        url,
+        status=200,
+        headers={"Set-Cookie": f"login_called=True; Domain={url}"},
+    )
+
+
+def mock_booking(aio_mock, url, response):
+    """
+    Mock a call to the booking endpoint
+
+    :param aio_mock: the aioresponses mock object
+    :param url: the URL of the booking endpoint
+    :param response: the response from the booking endpoint
+    """
+    aio_mock.post(
+        url,
+        status=200,
+        headers={"Set-Cookie": f"booking_called=True; Domain={url}"},
+        body=response,
+    )
+
+
+def retrieve_booking_datetime(
+    a_booking_filter: BookingFilter, a_club: Club
+) -> DateTime:
+    """
+    Utility to retrieve the booking datetime from the booking filter and the club
+
+    :param a_booking_filter: the booking filter that contains the date to book
+    :param a_club: the club which has the number of days before the date and the booking time
+    """
+    booking_hour = a_club.booking_opening_time.hour
+    booking_minute = a_club.booking_opening_time.minute
+
+    date_to_book = a_booking_filter.date
+    return date_to_book.subtract(days=a_club.booking_open_days_before).at(
+        booking_hour, booking_minute
+    )
+
+
+def mock_rest_api_from_connection_to_booking(
+    aio_mock, url: str, a_booking_failure_response: str, a_booking_success_response: str
+):
+    """
+    Mock a REST API from a club.
+    It mocks the calls to the connexion to the website, a call to log in the user
+    and 2 calls to the booking endpoint
+
+    :param aio_mock: the pendulum.now() mock
+    :param url: the API root URL
+    :param a_booking_success_response: the success json response
+    :param a_booking_failure_response: the failure json response
+    :return:
+    """
+    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)