diff --git a/tests/component_tests/external_components/test_init.py b/tests/component_tests/external_components/test_init.py index efc81c5475..d1c328b3cb 100644 --- a/tests/component_tests/external_components/test_init.py +++ b/tests/component_tests/external_components/test_init.py @@ -46,10 +46,12 @@ def test_external_components_skip_update_true( # Call with skip_update=True do_external_components_pass(config, skip_update=True) - # Verify clone_or_update was called with refresh=None + # Verify clone_or_update was called with NEVER_REFRESH mock_clone_or_update.assert_called_once() call_args = mock_clone_or_update.call_args - assert call_args.kwargs["refresh"] is None + from esphome import git + + assert call_args.kwargs["refresh"] == git.NEVER_REFRESH def test_external_components_skip_update_false( diff --git a/tests/component_tests/packages/test_init.py b/tests/component_tests/packages/test_init.py index 4f66f83e84..dfd4147c14 100644 --- a/tests/component_tests/packages/test_init.py +++ b/tests/component_tests/packages/test_init.py @@ -36,10 +36,12 @@ def test_packages_skip_update_true( # Call with skip_update=True do_packages_pass(config, skip_update=True) - # Verify clone_or_update was called with refresh=None + # Verify clone_or_update was called with NEVER_REFRESH mock_clone_or_update.assert_called_once() call_args = mock_clone_or_update.call_args - assert call_args.kwargs["refresh"] is None + from esphome import git + + assert call_args.kwargs["refresh"] == git.NEVER_REFRESH def test_packages_skip_update_false(