wehub-resource-sync
90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
# Copyright 2025-present the zvec project
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from zvec import require_module
|
|
|
|
|
|
# ----------------------------
|
|
# require_module func Test Case
|
|
# ----------------------------
|
|
def test_require_module_success():
|
|
module = require_module("os")
|
|
assert module is not None
|
|
assert hasattr(module, "path")
|
|
|
|
|
|
def test_require_module_with_submodule_success():
|
|
module = require_module("os.path")
|
|
assert module is not None
|
|
assert hasattr(module, "join")
|
|
|
|
|
|
def test_require_module_import_error():
|
|
with pytest.raises(ImportError) as exc_info:
|
|
require_module("nonexistent_module")
|
|
|
|
exception_msg = str(exc_info.value)
|
|
assert "Required package 'nonexistent_module' is not installed." in exception_msg
|
|
|
|
|
|
def test_require_module_with_mitigation_import_error():
|
|
with pytest.raises(ImportError) as exc_info:
|
|
require_module("nonexistent_module.submodule", mitigation="custom_package")
|
|
|
|
exception_msg = str(exc_info.value)
|
|
assert "Required package 'custom_package' is not installed." in exception_msg
|
|
assert (
|
|
"Module 'nonexistent_module.submodule' is part of 'nonexistent_module'"
|
|
in exception_msg
|
|
)
|
|
assert "please pip install 'custom_package'." in exception_msg
|
|
|
|
|
|
def test_require_module_submodule_import_error():
|
|
with pytest.raises(ImportError) as exc_info:
|
|
require_module("os.nonexistent_submodule")
|
|
|
|
exception_msg = str(exc_info.value)
|
|
assert (
|
|
"Required package 'os.nonexistent_submodule' is not installed." in exception_msg
|
|
)
|
|
assert "Module 'os.nonexistent_submodule' is part of 'os'" in exception_msg
|
|
assert "please pip install 'os'." in exception_msg
|
|
|
|
|
|
@patch("importlib.import_module")
|
|
def test_require_module_wraps_original_exception(mock_import_module):
|
|
original_exception = ImportError("Original error")
|
|
mock_import_module.side_effect = original_exception
|
|
|
|
with pytest.raises(ImportError) as exc_info:
|
|
require_module("some_module")
|
|
|
|
assert exc_info.value.__cause__ is original_exception
|
|
|
|
|
|
@patch("importlib.import_module")
|
|
def test_require_module_calls_importlib(mock_import_module):
|
|
mock_module = MagicMock()
|
|
mock_import_module.return_value = mock_module
|
|
|
|
result = require_module("test_module")
|
|
|
|
mock_import_module.assert_called_once_with("test_module")
|
|
assert result is mock_module
|