"""
Cross-Tabulated CSV 내보내기 개선 테스트 스크립트 (Simple 버전)
키 통합 기능 테스트

Author: Claude Assistant  
Created: 2025-07-16
Version: 1.0.0
"""

import sys
import os
from datetime import datetime
import json

# 프로젝트 루트 디렉토리를 Python 경로에 추가
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from cross_tabulated_csv_exporter import CrossTabulatedCSVExporter

class MockResult:
    """테스트용 모의 결과 객체"""
    
    def __init__(self, file_name, file_type, success=True, **kwargs):
        self.file_name = file_name
        self.file_type = file_type
        self.success = success
        
        # PDF 관련 속성
        if 'pdf_analysis_result' in kwargs:
            self.pdf_analysis_result = kwargs['pdf_analysis_result']
        
        # DXF 관련 속성
        if 'dxf_title_blocks' in kwargs:
            self.dxf_title_blocks = kwargs['dxf_title_blocks']
        
        # 오류 메시지
        if 'error_message' in kwargs:
            self.error_message = kwargs['error_message']


def test_key_integration():
    """키 통합 기능 테스트"""
    print("=== Cross-Tabulated CSV 키 통합 기능 테스트 ===")
    
    # 테스트용 CSV 내보내기 객체 생성
    exporter = CrossTabulatedCSVExporter()
    
    # 테스트 케이스 1: PDF 분석 결과 (키 분리 형태)
    print("\n테스트 케이스 1: PDF 분석 결과 (키 분리 -> 통합)")
    
    pdf_analysis_result = {
        "사업명_value": "고속국도 제30호선 대산~당진 고속도로 건설공사",
        "사업명_x": "40",
        "사업명_y": "130",
        "시설_공구_value": "제2공구 : 온산~사성",
        "시설_공구_x": "41", 
        "시설_공구_y": "139",
        "건설분야_value": "토목",
        "건설분야_x": "199",
        "건설분야_y": "1069",
        "건설단계_value": "실시설계",
        "건설단계_x": "263",
        "건설단계_y": "1069"
    }
    
    mock_pdf_result = MockResult(
        "황단면도.pdf",
        "PDF",
        success=True,
        pdf_analysis_result=json.dumps(pdf_analysis_result)
    )
    
    # 테스트 케이스 2: DXF 분석 결과
    print("테스트 케이스 2: DXF 분석 결과")
    
    dxf_title_blocks = [
        {
            "block_name": "TITLE_BLOCK",
            "attributes": [
                {
                    "tag": "DWG_NO",
                    "text": "A-001",
                    "insert_x": 150,
                    "insert_y": 25
                },
                {
                    "tag": "TITLE",
                    "text": "도면제목",
                    "insert_x": 200,
                    "insert_y": 50
                }
            ]
        }
    ]
    
    mock_dxf_result = MockResult(
        "도면001.dxf",
        "DXF",
        success=True,
        dxf_title_blocks=dxf_title_blocks
    )
    
    # 모든 테스트 결과를 리스트로 구성
    test_results = [mock_pdf_result, mock_dxf_result]
    
    # 출력 디렉토리 생성
    output_dir = "test_results_integrated"
    os.makedirs(output_dir, exist_ok=True)
    
    # 테스트 실행: 통합된 CSV 저장
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    output_path = os.path.join(output_dir, f"integrated_test_results_{timestamp}.csv")
    
    print(f"\nCSV 저장 위치: {output_path}")
    
    success = exporter.export_cross_tabulated_csv(
        test_results,
        output_path,
        include_coordinates=True,
        coordinate_source="auto"
    )
    
    if success:
        print(f"SUCCESS: 통합된 CSV 파일이 저장되었습니다.")
        print(f"파일 위치: {output_path}")
        
        # 저장된 CSV 파일 내용 미리보기
        try:
            with open(output_path, 'r', encoding='utf-8-sig') as f:
                lines = f.readlines()
                print(f"\nCSV 파일 미리보기 (상위 10행):")
                for i, line in enumerate(lines[:10]):
                    print(f"{i+1:2d}: {line.rstrip()}")
                
                if len(lines) > 10:
                    print(f"    ... 총 {len(lines)}행")
                    
        except Exception as e:
            print(f"WARNING: 파일 읽기 오류: {e}")
            
    else:
        print("ERROR: CSV 저장에 실패했습니다.")
    
    return success


def test_key_extraction():
    """키 추출 및 그룹화 기능 단위 테스트"""
    print("\n=== 키 추출 및 그룹화 단위 테스트 ===")
    
    exporter = CrossTabulatedCSVExporter()
    
    # 테스트 케이스: 다양한 키 형태
    test_keys = [
        "사업명_value",
        "사업명_x", 
        "사업명_y",
        "시설_공구_val",
        "시설_공구_x_coord",
        "시설_공구_y_coord",
        "건설분야_text",
        "건설분야_left",
        "건설분야_top",
        "일반키"
    ]
    
    print("\n키 분석 결과:")
    for key in test_keys:
        base_key = exporter._extract_base_key(key)
        key_type = exporter._determine_key_type(key)
        print(f"   {key:20s} -> 기본키: '{base_key:12s}' 타입: {key_type}")
    
    print("\n키 추출 및 그룹화 단위 테스트 완료")


def test_expected_output():
    """예상 출력 형태 테스트"""
    print("\n=== 예상 출력 형태 확인 ===")
    
    print("\n기대하는 CSV 출력 형태:")
    print("file_name,file_type,key,value,x,y")
    print("황단면도.pdf,PDF,사업명,고속국도 제30호선 대산~당진 고속도로 건설공사,40,130")
    print("황단면도.pdf,PDF,시설_공구,제2공구 : 온산~사성,41,139")
    print("황단면도.pdf,PDF,건설분야,토목,199,1069")
    print("도면001.dxf,DXF,DWG_NO,A-001,150,25")
    print("도면001.dxf,DXF,TITLE,도면제목,200,50")
    
    print("\n개선사항:")
    print("- 기존: 사업명_value, 사업명_x, 사업명_y가 별도 행으로 분리")
    print("- 개선: 사업명 하나의 행에 value, x, y 정보 통합")
    

if __name__ == "__main__":
    try:
        print("Cross-Tabulated CSV 키 통합 기능 테스트 시작")
        
        # 1. 키 추출 및 그룹화 단위 테스트
        test_key_extraction()
        
        # 2. 예상 출력 형태 확인
        test_expected_output()
        
        # 3. 실제 통합 기능 테스트
        success = test_key_integration()
        
        print(f"\n{'='*60}")
        if success:
            print("SUCCESS: 모든 테스트가 성공적으로 완료되었습니다!")
            print("키 통합 기능이 정상적으로 동작합니다.")
        else:
            print("ERROR: 테스트 중 일부가 실패했습니다.")
            print("코드를 점검해주세요.")
        print(f"{'='*60}")
        
    except Exception as e:
        print(f"ERROR: 테스트 실행 중 오류 발생: {e}")
        import traceback
        traceback.print_exc()