366 lines
14 KiB
Python
366 lines
14 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Unit tests for CrossFitBooker session-related methods
|
|
"""
|
|
|
|
import pytest
|
|
import os
|
|
import sys
|
|
from unittest.mock import patch, Mock
|
|
from datetime import datetime, date
|
|
import pytz
|
|
|
|
# Add the parent directory to the path
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from crossfit_booker import CrossFitBooker
|
|
|
|
|
|
class TestCrossFitBookerGetAvailableSessions:
|
|
"""Test cases for get_available_sessions method"""
|
|
|
|
def test_get_available_sessions_no_auth(self):
|
|
"""Test get_available_sessions without authentication"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
result = booker.get_available_sessions(date(2025, 7, 24), date(2025, 7, 25))
|
|
assert result is None
|
|
|
|
@patch('requests.Session.post')
|
|
def test_get_available_sessions_success(self, mock_post):
|
|
"""Test successful get_available_sessions"""
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"success": True,
|
|
"data": {
|
|
"activities_calendar": [
|
|
{"id": "1", "name": "Test Session"}
|
|
]
|
|
}
|
|
}
|
|
|
|
mock_post.return_value = mock_response
|
|
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
booker.auth_token = "test_token"
|
|
booker.user_id = "12345"
|
|
|
|
result = booker.get_available_sessions(date(2025, 7, 24), date(2025, 7, 25))
|
|
|
|
assert result is not None
|
|
assert result["success"] is True
|
|
|
|
@patch('requests.Session.post')
|
|
def test_get_available_sessions_401_error(self, mock_post):
|
|
"""Test get_available_sessions with 401 error"""
|
|
mock_response = Mock()
|
|
mock_response.status_code = 401
|
|
|
|
mock_post.return_value = mock_response
|
|
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
booker.auth_token = "test_token"
|
|
booker.user_id = "12345"
|
|
|
|
result = booker.get_available_sessions(date(2025, 7, 24), date(2025, 7, 25))
|
|
|
|
assert result is None
|
|
|
|
|
|
class TestCrossFitBookerBookSession:
|
|
"""Test cases for book_session method"""
|
|
|
|
def test_book_session_no_auth(self):
|
|
"""Test book_session without authentication"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
result = booker.book_session("session_123")
|
|
assert result is False
|
|
|
|
@patch('requests.Session.post')
|
|
def test_book_session_success(self, mock_post):
|
|
"""Test successful book_session"""
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {"success": True}
|
|
|
|
mock_post.return_value = mock_response
|
|
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
booker.auth_token = "test_token"
|
|
booker.user_id = "12345"
|
|
|
|
result = booker.book_session("session_123")
|
|
|
|
assert result is True
|
|
|
|
@patch('requests.Session.post')
|
|
def test_book_session_api_failure(self, mock_post):
|
|
"""Test book_session with API failure"""
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {"success": False, "error": "Session full"}
|
|
|
|
mock_post.return_value = mock_response
|
|
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
booker.auth_token = "test_token"
|
|
booker.user_id = "12345"
|
|
|
|
result = booker.book_session("session_123")
|
|
|
|
assert result is False
|
|
|
|
|
|
class TestCrossFitBookerIsSessionBookable:
|
|
"""Test cases for is_session_bookable method"""
|
|
|
|
def test_is_session_bookable_can_join_true(self):
|
|
"""Test session bookable with can_join=True"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
session = {"user_info": {"can_join": True}}
|
|
current_time = datetime.now(pytz.timezone("Europe/Paris"))
|
|
|
|
result = booker.is_session_bookable(session, current_time)
|
|
assert result is True
|
|
|
|
def test_is_session_bookable_booking_window_past(self):
|
|
"""Test session bookable with booking window in past"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
session = {
|
|
"user_info": {
|
|
"can_join": False,
|
|
"unableToBookUntilDate": "01-01-2020",
|
|
"unableToBookUntilTime": "10:00"
|
|
}
|
|
}
|
|
current_time = datetime.now(pytz.timezone("Europe/Paris"))
|
|
|
|
result = booker.is_session_bookable(session, current_time)
|
|
assert result is True
|
|
|
|
def test_is_session_bookable_booking_window_future(self):
|
|
"""Test session not bookable with booking window in future"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
session = {
|
|
"user_info": {
|
|
"can_join": False,
|
|
"unableToBookUntilDate": "01-01-2030",
|
|
"unableToBookUntilTime": "10:00"
|
|
}
|
|
}
|
|
current_time = datetime.now(pytz.timezone("Europe/Paris"))
|
|
|
|
result = booker.is_session_bookable(session, current_time)
|
|
assert result is False
|
|
|
|
|
|
class TestCrossFitBookerExcuteCycle:
|
|
"""Test cases for execute_cycle method"""
|
|
|
|
@patch('crossfit_booker.CrossFitBooker.get_available_sessions')
|
|
@patch('crossfit_booker.CrossFitBooker.is_session_bookable')
|
|
@patch('crossfit_booker.CrossFitBooker.matches_preferred_session')
|
|
@patch('crossfit_booker.CrossFitBooker.book_session')
|
|
async def test_execute_cycle_no_sessions(self, mock_book_session, mock_matches_preferred, mock_is_bookable, mock_get_sessions):
|
|
"""Test execute_cycle with no available sessions"""
|
|
mock_get_sessions.return_value = {"success": False}
|
|
booker = CrossFitBooker()
|
|
await booker.execute_cycle(datetime.now(pytz.timezone("Europe/Paris")))
|
|
mock_get_sessions.assert_called_once()
|
|
mock_book_session.assert_not_called()
|
|
|
|
@patch('crossfit_booker.CrossFitBooker.get_available_sessions')
|
|
@patch('crossfit_booker.CrossFitBooker.is_session_bookable')
|
|
@patch('crossfit_booker.CrossFitBooker.matches_preferred_session')
|
|
@patch('crossfit_booker.CrossFitBooker.book_session')
|
|
async def test_execute_cycle_with_sessions(self, mock_book_session, mock_matches_preferred, mock_is_bookable, mock_get_sessions):
|
|
"""Test execute_cycle with available sessions"""
|
|
# Use current date for the session to ensure it falls within 0-2 day window
|
|
current_time = datetime.now(pytz.timezone("Europe/Paris"))
|
|
session_date = current_time.date()
|
|
|
|
mock_get_sessions.return_value = {
|
|
"success": True,
|
|
"data": {
|
|
"activities_calendar": [
|
|
{
|
|
"id_activity_calendar": "1",
|
|
"name_activity": "CONDITIONING",
|
|
"start_timestamp": session_date.strftime("%Y-%m-%d") + " 18:30:00",
|
|
"user_info": {"can_join": True}
|
|
}
|
|
]
|
|
}
|
|
}
|
|
mock_is_bookable.return_value = True
|
|
mock_matches_preferred.return_value = True
|
|
mock_book_session.return_value = True
|
|
|
|
booker = CrossFitBooker()
|
|
await booker.execute_cycle(current_time)
|
|
|
|
mock_get_sessions.assert_called_once()
|
|
mock_is_bookable.assert_called_once()
|
|
mock_matches_preferred.assert_called_once()
|
|
mock_book_session.assert_called_once()
|
|
assert mock_book_session.call_count == 1
|
|
|
|
class TestCrossFitBookerRun:
|
|
"""Test cases for run method"""
|
|
|
|
@patch('crossfit_booker.CrossFitBooker.login')
|
|
@patch('crossfit_booker.CrossFitBooker.execute_cycle')
|
|
async def test_run_auth_failure(self, mock_execute_cycle, mock_login):
|
|
"""Test run with authentication failure"""
|
|
mock_login.return_value = False
|
|
booker = CrossFitBooker()
|
|
with patch.object(booker, 'run', new=booker.run) as mock_run:
|
|
await booker.run()
|
|
mock_login.assert_called_once()
|
|
mock_execute_cycle.assert_not_called()
|
|
|
|
@patch('crossfit_booker.CrossFitBooker.login')
|
|
@patch('crossfit_booker.CrossFitBooker.execute_cycle')
|
|
@patch('crossfit_booker.CrossFitBooker.quit')
|
|
@patch('time.sleep')
|
|
@patch('datetime.datetime')
|
|
async def test_run_booking_outside_window(self, mock_datetime, mock_sleep, mock_quit, mock_execute_cycle, mock_login):
|
|
"""Test run with booking outside window"""
|
|
mock_login.return_value = True
|
|
mock_quit.return_value = None # Prevent actual exit
|
|
|
|
# Create a time outside the booking window (19:00)
|
|
tz = pytz.timezone("Europe/Paris")
|
|
mock_now = datetime(2025, 7, 25, 19, 0, tzinfo=tz)
|
|
mock_datetime.now.return_value = mock_now
|
|
|
|
# Make sleep return immediately to allow one iteration, then break
|
|
call_count = 0
|
|
def sleep_side_effect(seconds):
|
|
nonlocal call_count
|
|
call_count += 1
|
|
if call_count >= 1:
|
|
# Break the loop after first sleep
|
|
raise KeyboardInterrupt("Test complete")
|
|
return None
|
|
|
|
mock_sleep.side_effect = sleep_side_effect
|
|
|
|
booker = CrossFitBooker()
|
|
|
|
try:
|
|
await booker.run()
|
|
except KeyboardInterrupt:
|
|
pass # Expected to break the loop
|
|
|
|
# Verify login was called
|
|
mock_login.assert_called_once()
|
|
|
|
# Verify execute_cycle was NOT called since we're outside the booking window
|
|
mock_execute_cycle.assert_not_called()
|
|
|
|
# Verify quit was called (due to KeyboardInterrupt)
|
|
mock_quit.assert_called_once()
|
|
|
|
class TestCrossFitBookerQuit:
|
|
"""Test cases for quit method"""
|
|
|
|
def test_quit(self):
|
|
"""Test quit method"""
|
|
booker = CrossFitBooker()
|
|
with patch('sys.exit') as mock_exit:
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
booker.quit()
|
|
assert excinfo.value.code == 0
|
|
class TestCrossFitBookerMatchesPreferredSession:
|
|
"""Test cases for matches_preferred_session method"""
|
|
|
|
def test_matches_preferred_session_exact_match(self):
|
|
"""Test exact match with preferred session"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
session = {
|
|
"start_timestamp": "2025-07-30 18:30:00",
|
|
"name_activity": "CONDITIONING"
|
|
}
|
|
current_time = datetime(2025, 7, 30, 12, 0, 0, tzinfo=pytz.timezone("Europe/Paris"))
|
|
|
|
# Mock PREFERRED_SESSIONS
|
|
with patch('crossfit_booker.PREFERRED_SESSIONS', [(2, "18:30", "CONDITIONING")]):
|
|
result = booker.matches_preferred_session(session, current_time)
|
|
assert result is True
|
|
|
|
def test_matches_preferred_session_fuzzy_match(self):
|
|
"""Test fuzzy match with preferred session"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
session = {
|
|
"start_timestamp": "2025-07-30 18:30:00",
|
|
"name_activity": "CONDITIONING WORKOUT"
|
|
}
|
|
current_time = datetime(2025, 7, 30, 12, 0, 0, tzinfo=pytz.timezone("Europe/Paris"))
|
|
|
|
# Mock PREFERRED_SESSIONS
|
|
with patch('crossfit_booker.PREFERRED_SESSIONS', [(2, "18:30", "CONDITIONING")]):
|
|
result = booker.matches_preferred_session(session, current_time)
|
|
assert result is True
|
|
|
|
def test_matches_preferred_session_no_match(self):
|
|
"""Test no match with preferred session"""
|
|
with patch.dict(os.environ, {
|
|
'CROSSFIT_USERNAME': 'test_user',
|
|
'CROSSFIT_PASSWORD': 'test_pass'
|
|
}):
|
|
booker = CrossFitBooker()
|
|
session = {
|
|
"start_timestamp": "2025-07-30 18:30:00",
|
|
"name_activity": "YOGA"
|
|
}
|
|
current_time = datetime(2025, 7, 30, 12, 0, 0, tzinfo=pytz.timezone("Europe/Paris"))
|
|
|
|
# Mock PREFERRED_SESSIONS
|
|
with patch('crossfit_booker.PREFERRED_SESSIONS', [(2, "18:30", "CONDITIONING")]):
|
|
result = booker.matches_preferred_session(session, current_time)
|
|
assert result is False |