import sys, os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
from comparator.aligner import align_records


def test_align_large_key_set():
    cobol = [{"ID": f"K{i:04d}", "V": i} for i in range(100)]
    spark = [{"ID": f"K{i:04d}", "V": i} for i in range(100)]
    result = align_records(cobol, spark, key_field="ID")
    assert len(result) == 100
    assert all(st == "MATCHED" for _, _, st in result)


def test_align_empty_key_value():
    cobol = [{"ID": "", "V": 1}]
    spark = [{"ID": "", "V": 1}]
    result = align_records(cobol, spark, key_field="ID")
    assert len(result) == 1