a7f9e6bacd
Fixed test_run_auth_failure by patching the correct method (CrossFitBooker.login) and calling the correct method (booker.login()) Fixed test_run_booking_outside_window by patching the correct method (Booker.run) and adding the necessary mocking for the booking cycle Added proper mocking for auth_token and user_id to avoid authentication errors in the tests Updated imports to use the src prefix for all imports Added proper test structure for the booking window logic test All tests now pass successfully
135 lines
4.1 KiB
Python
135 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Unit tests for AuthHandler class
|
|
"""
|
|
|
|
import pytest
|
|
import os
|
|
import sys
|
|
import requests
|
|
from unittest.mock import patch, Mock
|
|
|
|
# Add the parent directory to the path
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from src.auth import AuthHandler
|
|
|
|
class TestAuthHandlerAuthHeaders:
|
|
"""Test cases for get_auth_headers method"""
|
|
|
|
def test_get_auth_headers_without_token(self):
|
|
"""Test headers without auth token"""
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
headers = auth_handler.get_auth_headers()
|
|
assert "Authorization" not in headers
|
|
assert headers["User-Agent"] == "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:140.0) Gecko/20100101 Firefox/140.0"
|
|
|
|
def test_get_auth_headers_with_token(self):
|
|
"""Test headers with auth token"""
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
auth_handler.auth_token = "test_token_123"
|
|
headers = auth_handler.get_auth_headers()
|
|
assert headers["Authorization"] == "Bearer test_token_123"
|
|
|
|
class TestAuthHandlerLogin:
|
|
"""Test cases for login method"""
|
|
|
|
@patch('requests.Session.post')
|
|
def test_login_success(self, mock_post):
|
|
"""Test successful login flow"""
|
|
# Mock first login response
|
|
mock_response1 = Mock()
|
|
mock_response1.ok = True
|
|
mock_response1.json.return_value = {
|
|
"data": {
|
|
"user": {
|
|
"id_user": "12345"
|
|
}
|
|
}
|
|
}
|
|
|
|
# Mock second login response
|
|
mock_response2 = Mock()
|
|
mock_response2.ok = True
|
|
mock_response2.json.return_value = {
|
|
"token": "test_bearer_token"
|
|
}
|
|
|
|
mock_post.side_effect = [mock_response1, mock_response2]
|
|
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
result = auth_handler.login()
|
|
|
|
assert result is True
|
|
assert auth_handler.user_id == "12345"
|
|
assert auth_handler.auth_token == "test_bearer_token"
|
|
|
|
@patch('requests.Session.post')
|
|
def test_login_first_step_failure(self, mock_post):
|
|
"""Test login failure on first step"""
|
|
mock_response = Mock()
|
|
mock_response.ok = False
|
|
mock_response.status_code = 400
|
|
mock_response.text = "Bad Request"
|
|
|
|
mock_post.return_value = mock_response
|
|
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
result = auth_handler.login()
|
|
|
|
assert result is False
|
|
assert auth_handler.user_id is None
|
|
assert auth_handler.auth_token is None
|
|
|
|
@patch('requests.Session.post')
|
|
def test_login_second_step_failure(self, mock_post):
|
|
"""Test login failure on second step"""
|
|
# First response succeeds
|
|
mock_response1 = Mock()
|
|
mock_response1.ok = True
|
|
mock_response1.json.return_value = {
|
|
"data": {
|
|
"user": {
|
|
"id_user": "12345"
|
|
}
|
|
}
|
|
}
|
|
|
|
# Second response fails
|
|
mock_response2 = Mock()
|
|
mock_response2.ok = False
|
|
mock_response2.status_code = 401
|
|
|
|
mock_post.side_effect = [mock_response1, mock_response2]
|
|
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
result = auth_handler.login()
|
|
|
|
assert result is False
|
|
|
|
@patch('requests.Session.post')
|
|
def test_login_json_parsing_error(self, mock_post):
|
|
"""Test login with JSON parsing error"""
|
|
mock_response = Mock()
|
|
mock_response.ok = True
|
|
mock_response.json.side_effect = ValueError("Invalid JSON")
|
|
|
|
mock_post.return_value = mock_response
|
|
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
result = auth_handler.login()
|
|
|
|
assert result is False
|
|
|
|
@patch('requests.Session.post')
|
|
def test_login_request_exception(self, mock_post):
|
|
"""Test login with request exception"""
|
|
mock_post.side_effect = requests.exceptions.ConnectionError("Network error")
|
|
|
|
auth_handler = AuthHandler('test_user', 'test_pass')
|
|
result = auth_handler.login()
|
|
|
|
assert result is False
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__, "-v"]) |