refactor: Split code into many files

test/test_crossfit_booker_auth.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 """
-Unit tests for CrossFitBooker authentication methods
+Unit tests for CrossFitBooker authentication methods using AuthHandler
 """
 
 import pytest
@@ -13,11 +13,11 @@ from unittest.mock import patch, Mock
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from crossfit_booker import CrossFitBooker
-
+from auth import AuthHandler
 
 class TestCrossFitBookerAuthHeaders:
     """Test cases for get_auth_headers method"""
-    
+
     def test_get_auth_headers_without_token(self):
         """Test headers without auth token"""
         with patch.dict(os.environ, {
@@ -28,7 +28,7 @@ class TestCrossFitBookerAuthHeaders:
             headers = booker.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"""
         with patch.dict(os.environ, {
@@ -36,14 +36,13 @@ class TestCrossFitBookerAuthHeaders:
             'CROSSFIT_PASSWORD': 'test_pass'
         }):
             booker = CrossFitBooker()
-            booker.auth_token = "test_token_123"
+            booker.auth_handler.auth_token = "test_token_123"
             headers = booker.get_auth_headers()
             assert headers["Authorization"] == "Bearer test_token_123"
 
-
 class TestCrossFitBookerLogin:
     """Test cases for login method"""
-    
+
     @patch('requests.Session.post')
     def test_login_success(self, mock_post):
         """Test successful login flow"""
@@ -57,27 +56,27 @@ class TestCrossFitBookerLogin:
                 }
             }
         }
-        
+
         # 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]
-        
+
         with patch.dict(os.environ, {
             'CROSSFIT_USERNAME': 'test_user',
             'CROSSFIT_PASSWORD': 'test_pass'
         }):
             booker = CrossFitBooker()
             result = booker.login()
-            
+
             assert result is True
-            assert booker.user_id == "12345"
-            assert booker.auth_token == "test_bearer_token"
-    
+            assert booker.auth_handler.user_id == "12345"
+            assert booker.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"""
@@ -85,20 +84,20 @@ class TestCrossFitBookerLogin:
         mock_response.ok = False
         mock_response.status_code = 400
         mock_response.text = "Bad Request"
-        
+
         mock_post.return_value = mock_response
-        
+
         with patch.dict(os.environ, {
             'CROSSFIT_USERNAME': 'test_user',
             'CROSSFIT_PASSWORD': 'test_pass'
         }):
             booker = CrossFitBooker()
             result = booker.login()
-            
+
             assert result is False
-            assert booker.user_id is None
-            assert booker.auth_token is None
-    
+            assert booker.auth_handler.user_id is None
+            assert booker.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"""
@@ -112,55 +111,54 @@ class TestCrossFitBookerLogin:
                 }
             }
         }
-        
+
         # Second response fails
         mock_response2 = Mock()
         mock_response2.ok = False
         mock_response2.status_code = 401
-        
+
         mock_post.side_effect = [mock_response1, mock_response2]
-        
+
         with patch.dict(os.environ, {
             'CROSSFIT_USERNAME': 'test_user',
             'CROSSFIT_PASSWORD': 'test_pass'
         }):
             booker = CrossFitBooker()
             result = booker.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
-        
+
         with patch.dict(os.environ, {
             'CROSSFIT_USERNAME': 'test_user',
             'CROSSFIT_PASSWORD': 'test_pass'
         }):
             booker = CrossFitBooker()
             result = booker.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")
-        
+
         with patch.dict(os.environ, {
             'CROSSFIT_USERNAME': 'test_user',
             'CROSSFIT_PASSWORD': 'test_pass'
         }):
             booker = CrossFitBooker()
             result = booker.login()
-            
-            assert result is False
 
+            assert result is False
 
 if __name__ == "__main__":
     pytest.main([__file__, "-v"])