Skip to content

Commit

Permalink
fixing tests
Browse files Browse the repository at this point in the history
  • Loading branch information
amoghrajesh committed Aug 23, 2024
1 parent f8fb71e commit 3c4fd3b
Showing 1 changed file with 19 additions and 11 deletions.
30 changes: 19 additions & 11 deletions tests/core/test_settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,30 +119,38 @@ def teardown_method(self):
@mock.patch("airflow.settings.import_local_settings")
@mock.patch("airflow.settings.prepare_syspath_for_dags_folder")
def test_initialize_order(
self, prepare_syspath_for_config_and_plugins, import_local_settings, prepare_syspath_for_dags_folder
self,
mock_prepare_syspath_for_dags_folder,
mock_import_local_settings,
mock_prepare_syspath_for_config_and_plugins,
):
"""
Tests that import_local_settings is called between prepare_syspath_for_config_and_plugins
and prepare_syspath_for_dags_folder
"""
mock_local_settings = mock.Mock()

mock_local_settings.attach_mock(
mock_prepare_syspath_for_config_and_plugins, "prepare_syspath_for_config_and_plugins"
)
mock_local_settings.attach_mock(mock_import_local_settings, "import_local_settings")
mock_local_settings.attach_mock(
prepare_syspath_for_config_and_plugins, "prepare_syspath_for_config_and_plugins"
mock_prepare_syspath_for_dags_folder, "prepare_syspath_for_dags_folder"
)
mock_local_settings.attach_mock(import_local_settings, "import_local_settings")
mock_local_settings.attach_mock(prepare_syspath_for_dags_folder, "prepare_syspath_for_dags_folder")

import airflow.settings

airflow.settings.initialize()

mock_local_settings.assert_has_calls(
[
call.prepare_syspath_for_config_and_plugins(),
call.import_local_settings(),
call.prepare_syspath_for_dags_folder(),
]
)
expected_calls = [
call.prepare_syspath_for_config_and_plugins(),
call.import_local_settings(),
call.prepare_syspath_for_dags_folder(),
]

mock_local_settings.assert_has_calls(expected_calls)

assert mock_local_settings.mock_calls == expected_calls

def test_import_with_dunder_all_not_specified(self):
"""
Expand Down

0 comments on commit 3c4fd3b

Please sign in to comment.