Arch/tests/test_integration.py

import pytest
import os
import io
from werkzeug.datastructures import FileStorage

class TestIntegration:
    """Test integrations between different components."""
    
    def test_complete_workflow(self, logged_in_client, app):
        """Test a complete workflow from project creation to document download."""
        # 1. Create a new project
        logged_in_client.post('/projects/create', data={
            'codigo': 'WORKFLOW',
            'descripcion': 'Proyecto de flujo completo',
            'cliente': 'Cliente Integración',
            'esquema': 'TEST001',
            'destinacion': 'Pruebas de integración',
            'notas': 'Notas de proyecto de prueba'
        }, follow_redirects=True)
        
        # 2. Find the project ID
        project_id = None
        with app.app_context():
            projects_dir = os.path.join(app.config['STORAGE_PATH'], 'projects')
            project_dirs = [d for d in os.listdir(projects_dir) if 'WORKFLOW' in d]
            assert len(project_dirs) > 0
            project_dir = project_dirs[0]
            project_id = project_dir.split('_')[0].replace('@', '')
        
        assert project_id is not None
        
        # 3. Upload a document to the project
        test_file = FileStorage(
            stream=io.BytesIO(b'Content for integration test document'),
            filename='integration_doc.txt',
            content_type='text/plain',
        )
        
        response = logged_in_client.post(
            f'/projects/{project_id}/documents/upload',
            data={
                'tipo_doc': 'Documento de Prueba',
                'descripcion': 'Documento de flujo de integración',
                'file': test_file
            },
            content_type='multipart/form-data',
            follow_redirects=True
        )
        
        assert response.status_code == 200
        
        # 4. Find the document ID
        doc_id = None
        with app.app_context():
            projects_dir = os.path.join(app.config['STORAGE_PATH'], 'projects')
            for dir_name in os.listdir(projects_dir):
                if dir_name.startswith(f'@{project_id}_'):
                    project_path = os.path.join(projects_dir, dir_name)
                    docs_path = os.path.join(project_path, 'documents')
                    
                    doc_dirs = os.listdir(docs_path)
                    if doc_dirs:
                        doc_id = doc_dirs[0].split('_')[0].replace('@', '')
                        break
        
        assert doc_id is not None
        
        # 5. View document details
        response = logged_in_client.get(f'/projects/{project_id}/documents/{doc_id}')
        assert response.status_code == 200
        assert b'Documento de flujo de integraci' in response.data
        
        # 6. Upload a new version of the document
        test_file2 = FileStorage(
            stream=io.BytesIO(b'Updated content for version 2'),
            filename='integration_doc_v2.txt',
            content_type='text/plain',
        )
        
        response = logged_in_client.post(
            f'/projects/{project_id}/documents/{doc_id}/upload',
            data={
                'descripcion': 'Segunda versión del documento de integración',
                'file': test_file2
            },
            content_type='multipart/form-data',
            follow_redirects=True
        )
        
        assert response.status_code == 200
        
        # 7. Download the document
        response = logged_in_client.get(f'/projects/{project_id}/documents/{doc_id}/download/latest')
        assert response.status_code == 200
        assert b'Updated content for version 2' in response.data
        
        # 8. Download a specific version (the first one)
        response = logged_in_client.get(f'/projects/{project_id}/documents/{doc_id}/download/1')
        assert response.status_code == 200
        assert b'Content for integration test document' in response.data
    
    def test_schema_project_document_integration(self, logged_in_client, app):
        """Test integration between schemas, projects and documents."""
        # 1. Create a custom schema
        logged_in_client.post('/schemas/create', data={
            'codigo': 'CUSTOM',
            'descripcion': 'Esquema personalizado para integración',
            'documentos-0-tipo': 'pdf',
            'documentos-0-nombre': 'Informe Principal',
            'documentos-0-nivel_ver': 0,
            'documentos-0-nivel_editar': 5000,
            'documentos-1-tipo': 'txt',
            'documentos-1-nombre': 'Notas Adicionales',
            'documentos-1-nivel_ver': 0,
            'documentos-1-nivel_editar': 1000
        }, follow_redirects=True)
        
        # 2. Create a project with the custom schema
        logged_in_client.post('/projects/create', data={
            'codigo': 'PROJ_SCHEMA',
            'descripcion': 'Proyecto con esquema personalizado',
            'cliente': 'Cliente Test',
            'esquema': 'CUSTOM',
            'destinacion': 'Pruebas',
            'notas': 'Proyecto para probar integración de esquemas'
        }, follow_redirects=True)
        
        # 3. Find the project ID
        project_id = None
        with app.app_context():
            projects_dir = os.path.join(app.config['STORAGE_PATH'], 'projects')
            project_dirs = [d for d in os.listdir(projects_dir) if 'PROJ_SCHEMA' in d]
            assert len(project_dirs) > 0
            project_dir = project_dirs[0]
            project_id = project_dir.split('_')[0].replace('@', '')
        
        # 4. Verify project uses the custom schema
        response = logged_in_client.get(f'/projects/{project_id}')
        assert response.status_code == 200
        assert b'CUSTOM' in response.data
        assert b'Informe Principal' in response.data
        assert b'Notas Adicionales' in response.data
        
        # 5. Upload a document of the specified type
        test_file = FileStorage(
            stream=io.BytesIO(b'Notes content for schema integration'),
            filename='notes.txt',
            content_type='text/plain',
        )
        
        response = logged_in_client.post(
            f'/projects/{project_id}/documents/upload',
            data={
                'tipo_doc': 'Notas Adicionales',  # This should match schema document type
                'descripcion': 'Notas para prueba de esquema',
                'file': test_file
            },
            content_type='multipart/form-data',
            follow_redirects=True
        )
        
        assert response.status_code == 200