Arch/tests/test_documents.py

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


class TestDocuments:
    """Test document management functionality."""

    def setup_project(self, logged_in_client, app):
        """Helper to create a test project."""
        # Create a project
        logged_in_client.post(
            "/projects/create",
            data={
                "codigo": "TESTDOC",
                "descripcion": "Proyecto para documentos",
                "cliente": "Cliente Test",
                "esquema": "TEST001",
                "destinacion": "Pruebas",
                "notas": "Proyecto para pruebas de documentos",
            },
            follow_redirects=True,
        )

        # Find the project ID
        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 "TESTDOC" in d]
            assert len(project_dirs) > 0

            project_dir = project_dirs[0]
            return project_dir.split("_")[0].replace("@", "")

    def test_upload_document(self, logged_in_client, app):
        """Test uploading a document to a project."""
        # Create a project and get its ID
        project_id = self.setup_project(logged_in_client, app)

        # Create a test file
        test_file = FileStorage(
            stream=io.BytesIO(b"This is a test document content."),
            filename="test_document.txt",
            content_type="text/plain",
        )

        # Upload the document
        response = logged_in_client.post(
            f"/projects/{project_id}/documents/upload",
            data={
                "tipo_doc": "Documento de Prueba",  # Match schema type from TEST001
                "descripcion": "Documento de prueba para tests",
                "file": test_file,
            },
            content_type="multipart/form-data",
            follow_redirects=True,
        )

        assert response.status_code == 200

        # Verify document was created
        with app.app_context():
            project_path = None

            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)
                    break

            assert project_path is not None

            # Check for documents directory with content
            docs_path = os.path.join(project_path, "documents")
            assert os.path.exists(docs_path)

            # Should have at least one document folder
            assert len(os.listdir(docs_path)) > 0

    def test_document_versions(self, logged_in_client, app):
        """Test document versioning functionality."""
        # Create a project and upload first version
        project_id = self.setup_project(logged_in_client, app)

        # Upload first version
        test_file1 = FileStorage(
            stream=io.BytesIO(b"Document content version 1"),
            filename="test_versioning.txt",
            content_type="text/plain",
        )

        logged_in_client.post(
            f"/projects/{project_id}/documents/upload",
            data={
                "tipo_doc": "Documento de Prueba",
                "descripcion": "Documento para pruebas de versiones",
                "file": test_file1,
            },
            content_type="multipart/form-data",
            follow_redirects=True,
        )

        # 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")

                    # Get first document directory
                    doc_dirs = os.listdir(docs_path)
                    if doc_dirs:
                        doc_id = doc_dirs[0].split("_")[0].replace("@", "")
                        break

        assert doc_id is not None

        # Upload second version of the same document
        test_file2 = FileStorage(
            stream=io.BytesIO(b"Document content version 2 - UPDATED"),
            filename="test_versioning_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", "file": test_file2},
            content_type="multipart/form-data",
            follow_redirects=True,
        )

        assert response.status_code == 200

        # Check for multiple versions in metadata
        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")

                    # Find document directory
                    for doc_dir in os.listdir(docs_path):
                        if doc_dir.startswith(f"@{doc_id}_"):
                            doc_path = os.path.join(docs_path, doc_dir)
                            meta_path = os.path.join(doc_path, "meta.json")

                            # Check metadata for versions
                            if os.path.exists(meta_path):
                                with open(meta_path, "r") as f:
                                    metadata = json.load(f)
                                    assert "versions" in metadata
                                    assert len(metadata["versions"]) >= 2