def test_unicode_character(self):
new_file = 'n\xc3\xa9w_file.txt'
self.session.add_entry(new_file)
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.process()
path = os.path.join(TEST_DIR, new_file)
self.assertTrue(os.path.exists(path))
self.assertTrue(os.path.isfile(path))
def test_copy_file_between_directories(self):
digest = self.session.get_digest('egg.txt')
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.add_entry('egg.txt', digest)
second_session.process()
path = os.path.join(DOC_DIR, 'egg.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'egg file content')
def test_cut_paste_directory(self):
digest = self.session.get_digest('docs/')
self.session.remove_entry('docs/')
self.session.process()
second_session = MockSession(HELLO_DIR)
second_session.add_entry('docs/', digest)
second_session.process()
path = os.path.join(HELLO_DIR, 'docs/')
self.assertTrue(os.path.exists(path))
contents = os.listdir(path)
self.assertEqual(contents, ['research.txt'])
def test_multiple_file_deletes(self):
self.session.remove_entry('argh.md')
self.session.process()
self.assertFalse(os.path.exists(ARGH_FILE))
second_session = MockSession(TEST_DIR)
second_session.add_entry('argh.md')
second_session.process()
self.assertTrue(os.path.exists(ARGH_FILE))
second_session.reload()
second_session.remove_entry('argh.md')
second_session.process()
self.assertFalse(os.path.exists(ARGH_FILE))
def test_file_save_outside_roamer(self):
digest = self.session.get_digest('egg.txt')
path = os.path.join(TEST_DIR, 'egg.txt')
with open(path, 'a') as egg_file:
egg_file.write(' extra content')
os.utime(path, (1330712280, 1330712292))
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.add_entry('egg.txt', digest)
second_session.process()
path = os.path.join(DOC_DIR, 'egg.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'egg file content extra content')
def test_copy_over_existing_file(self):
erased_spam_digest = self.session.get_digest('spam.txt')
egg_digest = self.session.get_digest('egg.txt')
self.session.remove_entry('spam.txt')
self.session.add_entry('spam.txt', egg_digest)
self.session.process()
path = os.path.join(TEST_DIR, 'spam.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as spam_file:
self.assertEqual(spam_file.read(), 'egg file content')
second_session = MockSession(DOC_DIR)
second_session.add_entry('spam.txt', erased_spam_digest)
second_session.process()
path = os.path.join(DOC_DIR, 'spam.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as spam_file:
self.assertEqual(spam_file.read(), 'spam file content')
def test_cut_paste_file_same_name(self):
doc_session = MockSession(DOC_DIR)
digest = doc_session.get_digest('research.txt')
doc_session.remove_entry('research.txt')
doc_session.process()
self.session.add_entry('research.txt')
self.session.process()
self.session.reload()
self.session.remove_entry('research.txt')
self.session.process()
self.session.reload()
self.session.add_entry('my_new_research.txt', digest)
self.session.process()
path = os.path.join(TEST_DIR, 'my_new_research.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as research_file:
self.assertEqual(research_file.read(), 'research file content')
class TestOperations(unittest.TestCase): #pylint: disable=too-many-public-methods
def setUp(self):
reset_dirs([ROAMER_DATA_PATH, TRASH_DIR, TEST_DIR])
build_testing_entries()
self.session = MockSession(TEST_DIR)
def test_directory_text_output(self):
self.assertTrue('hello/' in self.session.text)
self.assertTrue('docs/' in self.session.text)
self.assertTrue('spam.txt' in self.session.text)
self.assertTrue('egg.txt' in self.session.text)
self.assertTrue('argh.md' in self.session.text)
def test_no_changes_does_nothing(self):
self.session.process()
self.assertTrue('hello/' in self.session.text)
self.assertTrue('docs/' in self.session.text)
self.assertTrue('spam.txt' in self.session.text)
self.assertTrue('egg.txt' in self.session.text)
self.assertTrue('argh.md' in self.session.text)
def test_create_new_file(self):
self.session.add_entry('new_file.txt')
self.session.process()
path = os.path.join(TEST_DIR, 'new_file.txt')
self.assertTrue(os.path.exists(path))
self.assertTrue(os.path.isfile(path))
def test_create_new_directory(self):
self.session.add_entry('new_dir/')
self.session.process()
path = os.path.join(TEST_DIR, 'new_dir/')
self.assertTrue(os.path.exists(path))
self.assertTrue(os.path.isdir(path))
def test_delete_file(self):
self.session.remove_entry('argh.md')
self.session.process()
self.assertFalse(os.path.exists(ARGH_FILE))
def test_delete_directory(self):
self.session.remove_entry('hello/')
self.session.process()
self.assertFalse(os.path.exists(HELLO_DIR))
def test_copy_file(self):
digest = self.session.get_digest('egg.txt')
self.session.add_entry('egg2.txt', digest)
self.session.process()
path = os.path.join(TEST_DIR, 'egg2.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg2_file:
self.assertEqual(egg2_file.read(), 'egg file content')
def test_copy_directory(self):
digest = self.session.get_digest('docs/')
self.session.add_entry('docs2/', digest)
self.session.process()
path = os.path.join(TEST_DIR, 'docs2/')
self.assertTrue(os.path.exists(path))
contents = os.listdir(path)
self.assertEqual(contents, ['research.txt'])
def test_rename_file(self):
self.session.rename('argh.md', 'blarg.md')
self.session.process()
self.assertFalse(os.path.exists(ARGH_FILE))
path = os.path.join(TEST_DIR, 'blarg.md')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as blarg_file:
self.assertEqual(blarg_file.read(), 'argh file content')
def test_rename_directory(self):
self.session.rename('docs/', 'my-docs/')
self.session.process()
self.assertFalse(os.path.exists(DOC_DIR))
path = os.path.join(TEST_DIR, 'my-docs/')
self.assertTrue(os.path.exists(path))
contents = os.listdir(path)
self.assertEqual(contents, ['research.txt'])
def disabled_clear_persisted_file_contents(self):
self.session.remove_entry('argh.md')
self.session.add_entry('argh.md')
self.session.process()
path = ARGH_FILE
self.assertTrue(os.path.exists(path))
with open(path, 'r') as argh_file:
self.assertEqual(argh_file.read(), '')
def test_comment_line(self):
original_entry_count = len(os.listdir(TEST_DIR))
self.session.add_entry('" Comment Line')
self.session.process()
entry_count = len(os.listdir(TEST_DIR))
self.assertEqual(entry_count, original_entry_count)
def test_rename_file_to_directory(self):
self.session.rename('hello/', 'hello.txt')
with self.assertRaises(ValueError):
self.session.process()
def test_empty_lines(self):
self.session.text += '\n \n \n '
self.session.process()
self.assertTrue(os.path.exists(ARGH_FILE))
def test_swap_files(self):
spam_digest = self.session.get_digest('spam.txt')
self.session.remove_entry('spam.txt')
self.session.rename('egg.txt', 'spam.txt')
self.session.add_entry('egg.txt', spam_digest)
self.session.process()
path = os.path.join(TEST_DIR, 'egg.txt')
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'spam file content')
path = os.path.join(TEST_DIR, 'spam.txt')
with open(path, 'r') as spam_file:
self.assertEqual(spam_file.read(), 'egg file content')
def test_multiple_simple_operations(self):
self.test_create_new_file()
self.session.reload()
self.test_delete_directory()
self.session.reload()
self.test_rename_file()
self.session.reload()
self.test_delete_file()
def test_copy_file_between_directories(self):
digest = self.session.get_digest('egg.txt')
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.add_entry('egg.txt', digest)
second_session.process()
path = os.path.join(DOC_DIR, 'egg.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'egg file content')
def test_cut_paste_files_between_directories(self):
digest = self.session.get_digest('egg.txt')
self.session.remove_entry('egg.txt')
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.add_entry('egg.txt', digest)
second_session.add_entry('egg2.txt', digest)
second_session.process()
path = os.path.join(DOC_DIR, 'egg.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'egg file content')
path = os.path.join(DOC_DIR, 'egg2.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'egg file content')
def test_cut_paste_file_same_name(self):
doc_session = MockSession(DOC_DIR)
digest = doc_session.get_digest('research.txt')
doc_session.remove_entry('research.txt')
doc_session.process()
self.session.add_entry('research.txt')
self.session.process()
self.session.reload()
self.session.remove_entry('research.txt')
self.session.process()
self.session.reload()
self.session.add_entry('my_new_research.txt', digest)
self.session.process()
path = os.path.join(TEST_DIR, 'my_new_research.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as research_file:
self.assertEqual(research_file.read(), 'research file content')
def test_cut_paste_directory(self):
digest = self.session.get_digest('docs/')
self.session.remove_entry('docs/')
self.session.process()
second_session = MockSession(HELLO_DIR)
second_session.add_entry('docs/', digest)
second_session.process()
path = os.path.join(HELLO_DIR, 'docs/')
self.assertTrue(os.path.exists(path))
contents = os.listdir(path)
self.assertEqual(contents, ['research.txt'])
def test_multiple_file_deletes(self):
self.session.remove_entry('argh.md')
self.session.process()
self.assertFalse(os.path.exists(ARGH_FILE))
second_session = MockSession(TEST_DIR)
second_session.add_entry('argh.md')
second_session.process()
self.assertTrue(os.path.exists(ARGH_FILE))
second_session.reload()
second_session.remove_entry('argh.md')
second_session.process()
self.assertFalse(os.path.exists(ARGH_FILE))
def test_copy_over_existing_file(self):
erased_spam_digest = self.session.get_digest('spam.txt')
egg_digest = self.session.get_digest('egg.txt')
self.session.remove_entry('spam.txt')
self.session.add_entry('spam.txt', egg_digest)
self.session.process()
path = os.path.join(TEST_DIR, 'spam.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as spam_file:
self.assertEqual(spam_file.read(), 'egg file content')
second_session = MockSession(DOC_DIR)
second_session.add_entry('spam.txt', erased_spam_digest)
second_session.process()
path = os.path.join(DOC_DIR, 'spam.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as spam_file:
self.assertEqual(spam_file.read(), 'spam file content')
def test_copy_file_same_name(self):
digest = self.session.get_digest('egg.txt')
for _ in range(3):
self.session.add_entry('egg.txt', digest)
self.session.process()
for egg_file in ['egg.txt', 'egg_copy_1.txt', 'egg_copy_2.txt', 'egg_copy_3.txt']:
path = os.path.join(TEST_DIR, egg_file)
self.assertTrue(os.path.exists(path))
with open(path, 'r') as new_file:
self.assertEqual(new_file.read(), 'egg file content')
def test_copy_dir_same_name(self):
digest = self.session.get_digest('docs/')
self.session.add_entry('docs/', digest)
self.session.process()
for doc_dir in ['docs/', 'docs_copy_1/']:
path = os.path.join(TEST_DIR, doc_dir)
self.assertTrue(os.path.exists(path))
contents = os.listdir(path)
self.assertEqual(contents, ['research.txt'])
def test_create_new_directory_and_file(self):
self.session.add_entry('new_file.txt')
self.session.add_entry('new_dir/')
self.session.process()
path = os.path.join(TEST_DIR, 'new_dir/')
self.assertTrue(os.path.exists(path))
self.assertTrue(os.path.isdir(path))
def test_copy_file_same_name_no_extension(self):
digest = self.session.get_digest('egg.txt')
for _ in range(3):
self.session.add_entry('egg', digest)
self.session.process()
for egg_file in ['egg.txt', 'egg', 'egg_copy_1', 'egg_copy_2']:
path = os.path.join(TEST_DIR, egg_file)
self.assertTrue(os.path.exists(path))
with open(path, 'r') as new_file:
self.assertEqual(new_file.read(), 'egg file content')
def test_shell_injection(self):
self.session.add_entry('new_file.txt; rm %s' % SPAM_FILE)
self.session.process()
self.assertTrue(os.path.exists(SPAM_FILE))
def test_unicode_character(self):
new_file = 'n\xc3\xa9w_file.txt'
self.session.add_entry(new_file)
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.process()
path = os.path.join(TEST_DIR, new_file)
self.assertTrue(os.path.exists(path))
self.assertTrue(os.path.isfile(path))
def test_entries_collision(self):
digest = self.session.get_digest('egg.txt')
directory = self.session.session.directory
entry = directory.entries[digest]
entry.digest = self.session.get_digest('argh.md')
with self.assertRaises(record.DigesetCollision):
record.add_dir(directory)
def test_trash_collision(self):
digest = self.session.get_digest('egg.txt')
directory = self.session.session.directory
entry = directory.entries[digest]
with self.assertRaises(record.DigesetCollision):
record.add_trash(entry.digest, entry.name, 'irrelevant-param')
record.add_trash(entry.digest, entry.name, 'irrelevant-param')
def test_missing_file_for_symbolic_link(self):
spam_link = os.path.join(TEST_DIR, 'spam_link')
os.symlink(SPAM_FILE, spam_link)
os.remove(SPAM_FILE)
self.session.reload()
self.session.remove_entry('spam_link')
self.session.process()
self.assertFalse(os.path.exists(spam_link))
def test_copy_symbolic_link(self):
spam_link = os.path.join(TEST_DIR, 'spam_link')
os.symlink(SPAM_FILE, spam_link)
self.session.reload()
digest = self.session.get_digest('spam_link')
self.session.add_entry('spam_link2', digest)
self.session.process()
spam_link2 = os.path.join(TEST_DIR, 'spam_link2')
self.assertEqual(os.readlink(spam_link2), SPAM_FILE)
def test_file_save_outside_roamer(self):
digest = self.session.get_digest('egg.txt')
path = os.path.join(TEST_DIR, 'egg.txt')
with open(path, 'a') as egg_file:
egg_file.write(' extra content')
os.utime(path, (1330712280, 1330712292))
self.session.process()
second_session = MockSession(DOC_DIR)
second_session.add_entry('egg.txt', digest)
second_session.process()
path = os.path.join(DOC_DIR, 'egg.txt')
self.assertTrue(os.path.exists(path))
with open(path, 'r') as egg_file:
self.assertEqual(egg_file.read(), 'egg file content extra content')