def __initialize(args):
repository = SettingRepository(logger=args.log, config=args.config)
file_manager = FileService(logger=args.log, config=args.config)
if not file_manager.directory_exist():
print('... Creating directory ...')
file_manager.create_directory()
print('... Creating database ...')
repository.create_settings_table()
if repository.find_one_by('default') is None:
print('... Adding default settings ...')
repository.create(Setting('default', ''))
class FileServiceTest(unittest.TestCase):
def setUp(self):
self.service = FileService(Log(False), Config())
@patch('os.listdir')
def test_list_all__when_execute__should_return_xml_only(self, listdir):
listdir.return_value = ['se1.xml', 'se2.xml', 'msa.db', 'se3.xml']
all_settings = self.service.list_all()
self.assertEqual(all_settings, ['se1.xml', 'se2.xml', 'se3.xml'])
@patch('os.remove')
def test_remove_setting__when_setting_have_file__should_be_removed(
self, remove):
setting = Setting('alias', 'file')
self.service.remove_setting(setting)
remove.assert_called()
@patch('os.remove')
def test_remove_setting__when_setting_have_not_file__should_be_removed(
self, remove):
setting = Setting('alias', '')
self.service.remove_setting(setting)
remove.assert_not_called()
@patch('os.path.exists')
def test_directory_exist__when_os_path_exists__should_return_true(
self, exists):
exists.return_value = True
result = self.service.directory_exist()
self.assertTrue(result)
@patch('os.path.exists')
def test_directory_exist__when_os_path_not_exists__should_return_false(
self, exists):
exists.return_value = False
result = self.service.directory_exist()
self.assertFalse(result)
@patch('os.mkdir')
def test_create_directory__when_executed__should_call_mkdir(self, mkdir):
self.service.create_directory()
mkdir.assert_called()
@patch('os.path.exists')
@patch('os.remove')
def test_deactivate_setting__when_setting_not_exists__should_not_call_remove(
self, mock_remove, mock_exists):
mock_exists.return_value = False
self.service.deactivate_setting()
mock_remove.assert_not_called()
@patch('os.path.exists')
@patch('os.remove')
def test_deactivate_setting__when_setting_exists__should_call_remove(
self, mock_remove, mock_exists):
mock_exists.return_value = True
self.service.deactivate_setting()
mock_remove.assert_called()