from genesis.data_models import SheetType
from genesis.parsers.sheet_detector import detect_sheet_type


def test_detect_by_sheet_name():
    assert detect_sheet_type("機能一覧", []) == SheetType.FUNCTION
    assert detect_sheet_type("画面一覧", []) == SheetType.SCREEN
    assert detect_sheet_type("帳票一覧", []) == SheetType.REPORT
    assert detect_sheet_type("DB定義", []) == SheetType.DATABASE
    assert detect_sheet_type("IF定義", []) == SheetType.INTERFACE
    assert detect_sheet_type("バッチ一覧", []) == SheetType.BATCH
    assert detect_sheet_type("コード管理", []) == SheetType.MASTER


def test_detect_by_header_keyword():
    matrix = [["帳票ID", "帳票名"], ["TB1", "月次"]]
    assert detect_sheet_type("補充シート", matrix) == SheetType.REPORT


def test_unknown_is_generic():
    assert detect_sheet_type("メモ", [["随筆"]]) == SheetType.GENERIC