import pytest


@pytest.fixture
def example_file(example_root):
    return example_root / "Verbrauchsmittel-Toto-2020.xlsx"


@pytest.fixture
def example_workbook(example_file):
    import openpyxl

    yield openpyxl.open(example_file)


@pytest.fixture
def example_sheet(example_workbook):
    sheets = example_workbook.sheetnames
    first = sheets[0]
    yield example_workbook[first]


def test_check_table_header_raises_error():
    from superx_budget.budget import _check_table_header
    from superx_budget.exceptions import BudgetParserError

    row = ["not", "the", "expected", "row"]

    with pytest.raises(BudgetParserError):
        _check_table_header(row)


def test_skip_empty_lines():
    from superx_budget.budget import _skip_empty_lines

    rows = [[""], ["one"], [None], [""], ["two"], [""]]

    result = _skip_empty_lines(rows)

    assert list(result) == [(1, ["one"]), (4, ["two"])]


def test_parse_data_table():
    from superx_budget.budget import _parse_data_table

    rows = [
        list("ABCDEFG"),
        [None for i in range(7)],
        list("tuvwxyzX"),  # one item more
    ]

    result = _parse_data_table(rows)
    first, last = list(result)

    assert first.row == 2
    assert first.number == "A"
    assert first.rest == "G"
    assert last.row == 4
    assert last.number == "t"
    assert last.rest == "z"


def test_parse_budget_data(example_sheet):
    from superx_budget.budget import parse_budget_data

    result = parse_budget_data(example_sheet)
    first, last = result[0], result[-1]

    assert len(result) == 18
    assert first.row == 3
    assert first.number == 1
    assert first.project_name == "Safegurard I (neu)"
    assert first.rest == "=E3-F3"
    assert last.row == 54
    assert last.number == "=A51+1"
    assert last.project_name == "ZIM Microcoat II"
    assert last.rest == "=E54-F54"