"""Basic tests for CAD-Documenter pipeline."""

import pytest
from pathlib import Path


def test_imports():
    """Test that all modules can be imported."""
    from cad_documenter import __version__
    from cad_documenter.cli import main
    from cad_documenter.pipeline import DocumentationPipeline
    from cad_documenter.video_processor import VideoProcessor
    from cad_documenter.audio_analyzer import AudioAnalyzer
    from cad_documenter.vision_analyzer import VisionAnalyzer
    from cad_documenter.doc_generator import DocGenerator

    assert __version__ == "0.1.0"


def test_transcript_dataclass():
    """Test Transcript dataclass functionality."""
    from cad_documenter.audio_analyzer import Transcript, TranscriptSegment

    segments = [
        TranscriptSegment(start=0.0, end=5.0, text="This is the main bracket"),
        TranscriptSegment(start=5.0, end=10.0, text="It holds the motor"),
        TranscriptSegment(start=10.0, end=15.0, text="Made of aluminum"),
    ]

    transcript = Transcript(segments=segments, full_text="This is the main bracket. It holds the motor. Made of aluminum.")

    # Test get_text_at
    text = transcript.get_text_at(7.0, window=3.0)
    assert "holds the motor" in text
    assert "main bracket" in text


def test_component_dataclass():
    """Test Component dataclass."""
    from cad_documenter.vision_analyzer import Component

    component = Component(
        name="Main Bracket",
        description="Primary structural member",
        function="Holds the motor",
        material="Aluminum 6061-T6",
        features=["4x M6 holes", "Fillet radii"],
    )

    assert component.name == "Main Bracket"
    assert len(component.features) == 2


# TODO: Add integration tests with sample videos