"""Tests for schemas.columns — CsvColumn enum and REQUIRED_COLUMNS."""

from schemas import REQUIRED_COLUMNS, CsvColumn


class TestCsvColumn:
    def test_all_columns_are_strings(self):
        for col in CsvColumn:
            assert isinstance(col.value, str)

    def test_required_columns_are_subset(self):
        assert REQUIRED_COLUMNS.issubset(set(CsvColumn))

    def test_required_includes_account_id(self):
        assert CsvColumn.ACCOUNT_ID in REQUIRED_COLUMNS

    def test_required_includes_contract_name(self):
        assert CsvColumn.CONTRACT_NAME in REQUIRED_COLUMNS

    def test_required_includes_contract_type(self):
        assert CsvColumn.CONTRACT_TYPE in REQUIRED_COLUMNS

    def test_optional_not_in_required(self):
        assert CsvColumn.SIGNING_ENTITY not in REQUIRED_COLUMNS
        assert CsvColumn.RUN_CONTROLLER not in REQUIRED_COLUMNS
        assert CsvColumn.EXECUTION_DATE not in REQUIRED_COLUMNS