def test_policy_function(self): """ Tests that task instances are mutated by the policy function in airflow_local_settings. """ with SettingsContext(SETTINGS_FILE_POLICY, "airflow_local_settings"): from airflow import settings settings.import_local_settings() task_instance = MagicMock() settings.test_policy(task_instance) # pylint: disable=no-member assert task_instance.run_as_user == "myself"
def test_import_with_dunder_all(self): """ Tests that if __all__ is specified in airflow_local_settings, only module attributes specified within are imported. """ with SettingsContext(SETTINGS_FILE_POLICY_WITH_DUNDER_ALL, "airflow_local_settings"): from airflow import settings settings.import_local_settings() # pylint: ignore task_instance = MagicMock() settings.test_policy(task_instance) assert task_instance.run_as_user == "myself"