def test_ConfigLoader_revert_to_default_creates_files_and_directories(
self):
home_to_delete = os.path.abspath('delete_me')
save_config({'home_directory': home_to_delete})
new = ConfigLoader()
home = os.path.join(get_documents_folder(), APP_NAME)
self.assertFalse(os.path.exists(home))
for filename in [
VERBS_CSV, COUNTABLE_NOUNS_CSV, UNCOUNTABLE_NOUNS_CSV
]:
self.assertTrue(
os.path.exists(os.path.join(home_to_delete, filename)))
new.revert_to_default()
for filename in [
VERBS_CSV, COUNTABLE_NOUNS_CSV, UNCOUNTABLE_NOUNS_CSV
]:
self.assertTrue(
os.path.exists(os.path.join(home_to_delete, filename)))
self.assert_default_ConfigLoader_state(new)
rmtree(home_to_delete)
def test_ConfigLoader_revert_to_default_resets_csvs_but_leaves_other_files(
self):
new = ConfigLoader()
home = os.path.join(get_documents_folder(), APP_NAME)
save = os.path.join(home, DEFAULT_SAVE_DIR)
home_files = [VERBS_CSV, COUNTABLE_NOUNS_CSV, 'foo.txt', 'bar.txt']
save_files = ['foo.txt', 'bar.txt']
write_files = [os.path.join(home, filename) for filename in home_files]
write_files += [
os.path.join(save, filename) for filename in save_files
]
files_not_reset = write_files[2:]
for filename in write_files:
with open(filename, 'w') as f:
f.write('foobar')
new.revert_to_default()
self.assert_default_ConfigLoader_state(new)
for filename in files_not_reset:
with open(filename, 'r') as f:
self.assertEqual(f.read(), 'foobar')
def revert_to_original(self):
loader = ConfigLoader()
loader.revert_to_default()
self.load_config()