def test_config_folder_nofile(self, mock_path, mock_conf_parse):
settings_path = os.path.join(armory.CONFIG_FOLDER, armory.CONFIG_FILE)
mock_path.exists.return_value = False
mock_path.join.return_value = settings_path
with self.assertRaises(ValueError):
armory.get_config_options()
self.assertTrue(mock_path.join.called,
"os.path.join should have been called.")
self.assertTrue(mock_path.exists.called,
"os.path.exists should have been called.")
self.assertFalse(mock_conf_parse.called,
"ConfigParser() should not have been called.")
def test_custom_config_no_exist(self, mock_os, mock_path):
settings_path = "some/path/to/a/config/file.ini"
config = """
[PROJECT]
base_path=some/path
"""
config_parser_mock = mock.create_autospec(ConfigParser, instance=True)
conf_parser_class_mock = MagicMock(return_value=config_parser_mock)
mock_path.exists = MagicMock(return_value=False)
mock_path.join.return_value = settings_path
with mock.patch("builtins.open",
new_callable=mock_open,
read_data=config) as mopen:
with mock.patch("configparser.ConfigParser",
conf_parser_class_mock):
with self.assertRaises(ValueError):
self.assertIsNotNone(
armory.get_config_options(settings_path))
print("open() calls: {}".format(mopen.call_args_list))
self.assertFalse(
mopen.called,
"open() was called when it should not have been.")
self.assertFalse(mock_path.join.called,
"os.path.join should not have been called.")
self.assertEqual(mock_path.exists.call_count, 1,
"os.path.exists call_count should be 2.")
mock_path.exists.assert_called_once_with(settings_path)
self.assertFalse(
mock_os.makedirs.called,
"os.makedirs was called when it shouldn't have been.",
)
def test_config_folder_nobasepath(self, mock_os, mock_path):
settings_path = os.path.join(armory.CONFIG_FOLDER, armory.CONFIG_FILE)
config = {"PROJECT": {"base_path": "."}}
def dir_exists(value):
if value == config["PROJECT"]["base_path"]:
return False
else:
return True
config_parser_mock = mock.create_autospec(ConfigParser, instance=True)
conf_parser_class_mock = MagicMock(return_value=config_parser_mock)
mock_path.exists = MagicMock(side_effect=dir_exists)
mock_path.join.return_value = settings_path
with mock.patch("configparser.ConfigParser", conf_parser_class_mock):
self.assertIsNotNone(armory.get_config_options())
self.assertTrue(mock_path.join.called,
"os.path.join should have been called.")
self.assertEqual(mock_path.exists.call_count, 2,
"os.path.exists call_count should be 2.")
mock_path.exists.assert_any_call(settings_path)
mock_path.exists.assert_any_call(config["PROJECT"]["base_path"])
mock_os.makedirs.assert_called_once_with(
config["PROJECT"]["base_path"])