from core.settings_objects import BudgetTrackerSettings


def test_i_a_can_as_dict():
    settings = BudgetTrackerSettings()
    assert settings.as_dict() == {'col_actual_amt': 'AO',
                                  'col_budget_amt': 'BQ',
                                  'col_capex': 'G',
                                  'col_details': 'H',
                                  'col_forecast5_7_amt': 'BC',
                                  'col_owner': 'E',
                                  'col_project': 'D',
                                  'col_row_num': 'A',
                                  'col_supplier': 'I',
                                  'spread_sheet': 'full charges'}


def test_i_can_load_from_dict():
    as_dict = {'spread_sheet': 'spread_sheet',
               'col_actual_amt': 'actual',
               'col_budget_amt': 'budget',
               'col_capex': 'capex',
               'col_details': 'details',
               'col_forecast5_7_amt': 'forecast5_7',
               'col_index': 'index',
               'col_owner': 'owner',
               'col_project': 'project',
               'col_supplier': 'supplier'}

    settings = BudgetTrackerSettings().from_dict(as_dict)

    assert settings.spread_sheet == "spread_sheet"
    assert settings.col_actual_amt == "actual"
    assert settings.col_budget_amt == "budget"
    assert settings.col_capex == "capex"
    assert settings.col_details == "details"
    assert settings.col_forecast5_7_amt == "forecast5_7"
    assert settings.col_index == "index"
    assert settings.col_owner == "owner"
    assert settings.col_project == "project"
    assert settings.col_supplier == "supplier"


def test_i_can_as_formatted_dict():
    settings = BudgetTrackerSettings()
    assert settings.get_display_name("spread_sheet") == 'Spread Sheet'
    assert settings.get_display_name("col_forecast5_7_amt") == 'Forecast 5+7'