|
from unittest.mock import mock_open, patch
|
|
|
|
from pr.core.config_loader import (
|
|
_load_config_file,
|
|
_parse_value,
|
|
create_default_config,
|
|
load_config,
|
|
)
|
|
|
|
|
|
def test_parse_value_string():
|
|
assert _parse_value("hello") == "hello"
|
|
|
|
|
|
def test_parse_value_int():
|
|
assert _parse_value("123") == 123
|
|
|
|
|
|
def test_parse_value_float():
|
|
assert _parse_value("1.23") == 1.23
|
|
|
|
|
|
def test_parse_value_bool_true():
|
|
assert _parse_value("true") == True
|
|
|
|
|
|
def test_parse_value_bool_false():
|
|
assert _parse_value("false") == False
|
|
|
|
|
|
def test_parse_value_bool_upper():
|
|
assert _parse_value("TRUE") == True
|
|
|
|
|
|
@patch("os.path.exists", return_value=False)
|
|
def test_load_config_file_not_exists(mock_exists):
|
|
config = _load_config_file("test.ini")
|
|
assert config == {}
|
|
|
|
|
|
@patch("os.path.exists", return_value=True)
|
|
@patch("configparser.ConfigParser")
|
|
def test_load_config_file_exists(mock_parser_class, mock_exists):
|
|
mock_parser = mock_parser_class.return_value
|
|
mock_parser.sections.return_value = ["api"]
|
|
mock_parser.items.return_value = [("key", "value")]
|
|
config = _load_config_file("test.ini")
|
|
assert "api" in config
|
|
assert config["api"]["key"] == "value"
|
|
|
|
|
|
@patch("pr.core.config_loader._load_config_file")
|
|
def test_load_config(mock_load):
|
|
mock_load.side_effect = [{"api": {"key": "global"}}, {"api": {"key": "local"}}]
|
|
config = load_config()
|
|
assert config["api"]["key"] == "local"
|
|
|
|
|
|
@patch("builtins.open", new_callable=mock_open)
|
|
def test_create_default_config(mock_file):
|
|
result = create_default_config("test.ini")
|
|
assert result == True
|
|
mock_file.assert_called_once_with("test.ini", "w")
|
|
handle = mock_file()
|
|
handle.write.assert_called_once()
|
|
|
|
|
|
@patch("builtins.open", side_effect=Exception("error"))
|
|
def test_create_default_config_error(mock_file):
|
|
result = create_default_config("test.ini")
|
|
assert result == False
|