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()