"""Tests for app/views/onedrive.py module.""" import json from unittest.mock import patch import pytest from app.models import UserIntegration @pytest.mark.integration class TestOnedriveViews: """Tests for OneDrive view routes.""" def test_onedrive_setup_page(self, client): """Test the OneDrive setup page.""" response = client.get("/onedrive-setup") assert response.status_code == 200 def test_onedrive_callback_no_code(self, client): """Test the OneDrive OAuth callback without code.""" response = client.get("/onedrive-callback", follow_redirects=False) assert response.status_code == 200 def test_onedrive_callback_with_error(self, client): """Test the OneDrive OAuth callback with error.""" response = client.get("/onedrive-callback?error=access_denied") assert response.status_code == 200 def test_onedrive_callback_with_code(self, client): """Test the OneDrive OAuth callback with auth code.""" response = client.get("/onedrive-callback?code=test_code") assert response.status_code == 200 def test_onedrive_setup_page_with_integration_id(self, client): """Test the OneDrive setup page accepts integration_id query param.""" response = client.get("/onedrive-setup?integration_id=77") assert response.status_code == 200 # The template should store the integration_id for per-user OAuth flow assert b"oauth_integration_id" in response.content assert b"77" in response.content def test_onedrive_setup_page_without_integration_id(self, client): """Test the OneDrive setup page works without integration_id (global flow).""" response = client.get("/onedrive-setup") assert response.status_code == 200 body = response.text assert "oauth_integration_id" in body assert 'const integrationId = ""' in body def test_onedrive_setup_user_mode_invalid_json_config(self, client, db_session): """Test user-mode renders correctly when integration.config is invalid JSON.""" owner_id = "user_od_invalid_json@example.com" integration = UserIntegration( owner_id=owner_id, direction="DESTINATION", integration_type="ONEDRIVE", name="My OneDrive (bad cfg)", config="{INVALID JSON}", is_active=True, ) db_session.add(integration) db_session.commit() db_session.refresh(integration) with patch("app.views.onedrive.get_current_owner_id", return_value=owner_id): response = client.get(f"/onedrive-setup?integration_id={integration.id}") assert response.status_code == 200 # Should render user mode without errors despite the bad config assert b"Back to Integrations" in response.content def test_onedrive_setup_user_mode_none_config(self, client, db_session): """Test user-mode renders correctly when integration.config is None (no folder path).""" owner_id = "user_od_none_cfg@example.com" integration = UserIntegration( owner_id=owner_id, direction="DESTINATION", integration_type="ONEDRIVE", name="My OneDrive (no cfg)", config=None, is_active=True, ) db_session.add(integration) db_session.commit() db_session.refresh(integration) with patch("app.views.onedrive.get_current_owner_id", return_value=owner_id): response = client.get(f"/onedrive-setup?integration_id={integration.id}") assert response.status_code == 200 # Should render user mode without errors, with empty folder_path assert b"Back to Integrations" in response.content def test_onedrive_setup_user_mode_watchfolder_config(self, client, db_session): """Test user-mode correctly loads folder_path from WATCH_FOLDER source config.""" owner_id = "user_od_wf_cfg@example.com" integration = UserIntegration( owner_id=owner_id, direction="SOURCE", integration_type="WATCH_FOLDER", name="My OneDrive Watch", config=json.dumps({"source_type": "onedrive", "folder_path": "Work/Inbox"}), is_active=True, ) db_session.add(integration) db_session.commit() db_session.refresh(integration) with patch("app.views.onedrive.get_current_owner_id", return_value=owner_id): response = client.get(f"/onedrive-setup?integration_id={integration.id}") assert response.status_code == 200 assert b"Work/Inbox" in response.content assert b"Back to Integrations" in response.content def test_onedrive_setup_user_mode_integration_not_found(self, client, db_session): """Test user-mode falls back to admin mode when integration not owned by user.""" with patch("app.views.onedrive.get_current_owner_id", return_value="other_user@example.com"): response = client.get("/onedrive-setup?integration_id=999999") assert response.status_code == 200 # Falls back to admin mode (no "Back to Integrations" link) assert b"OneDrive Integration Setup" in response.content def test_onedrive_setup_user_mode_valid_config(self, client, db_session): """Test user-mode correctly loads folder path from integration config.""" owner_id = "user_od_valid_cfg@example.com" integration = UserIntegration( owner_id=owner_id, direction="DESTINATION", integration_type="ONEDRIVE", name="My OneDrive", config=json.dumps({"folder_path": "Documents/Archive"}), is_active=True, ) db_session.add(integration) db_session.commit() db_session.refresh(integration) with patch("app.views.onedrive.get_current_owner_id", return_value=owner_id): response = client.get(f"/onedrive-setup?integration_id={integration.id}") assert response.status_code == 200 assert b"Documents/Archive" in response.content assert b"Back to Integrations" in response.content