def test_one_file_commit(self):
testfile = os.path.join(self.tempdir, 'testfile.dat')
self.create_file(testfile, 'one')
ftrans = FileTransaction()
fp = ftrans.open(testfile, 'r')
self.assertEqual(fp.read(), 'one')
fp = ftrans.open(testfile, 'w')
fp.write('one,two')
fp.close()
fp = ftrans.open(testfile, 'r')
self.assertEqual(fp.read(), 'one,two')
fp = ftrans.open(testfile, 'a')
fp.write(',three')
fp.close()
fp = ftrans.open(testfile, 'r')
self.assertEqual(fp.read(), 'one,two,three')
ftrans.commit()
self.assertEqual(open(testfile, 'r').read(), 'one,two,three')
def test_abort_file_added(self):
" commit aborted because file added "
filename = os.path.join(self.tempdir, 'file.dat')
ftrans = FileTransaction()
fp = ftrans.open(filename, 'w')
fp.write('record1')
open(filename, 'w').close()
self.assertRaises(FileTransactionException, ftrans.commit)
def test_abort_file_added(self):
" commit aborted because file added "
filename = os.path.join(self.tempdir, 'file.dat')
ftrans = FileTransaction()
fp = ftrans.open(filename, 'w')
fp.write('record1')
open(filename, 'w').close()
self.assertRaises(FileTransactionException, ftrans.commit)
def test_abort_file_removed(self):
" commit aborted because file removed "
filename = os.path.join(self.tempdir, 'file.dat')
self.create_file(filename, 'record1')
ftrans = FileTransaction()
fp = ftrans.open(filename, 'w')
fp.write('record2')
os.unlink(filename)
self.assertRaises(FileTransactionException, ftrans.commit)
def test_abort_file_removed(self):
" commit aborted because file removed "
filename = os.path.join(self.tempdir, 'file.dat')
self.create_file(filename, 'record1')
ftrans = FileTransaction()
fp = ftrans.open(filename, 'w')
fp.write('record2')
os.unlink(filename)
self.assertRaises(FileTransactionException, ftrans.commit)
def test_one_file_commit(self):
testfile = os.path.join(self.tempdir, 'testfile.dat')
self.create_file(testfile, 'one')
ftrans = FileTransaction()
fp = ftrans.open(testfile, 'r')
self.assertEqual(fp.read(), 'one')
fp = ftrans.open(testfile, 'w')
fp.write('one,two')
fp.close()
fp = ftrans.open(testfile, 'r')
self.assertEqual(fp.read(), 'one,two')
fp = ftrans.open(testfile, 'a')
fp.write(',three')
fp.close()
fp = ftrans.open(testfile, 'r')
self.assertEqual(fp.read(), 'one,two,three')
ftrans.commit()
self.assertEqual(open(testfile, 'r').read(), 'one,two,three')
def test_mkdir_rollback(self):
ftrans = FileTransaction()
ftrans.mkdir(os.path.join(self.tempdir, '1'))
ftrans.mkdir(os.path.join(self.tempdir, '1', '2'))
fp = ftrans.open(os.path.join(self.tempdir, '1', '2', 'file'), 'w')
fp.write('data')
fp.close()
ftrans.rollback()
self.assertFalse(os.path.exists(os.path.join(self.tempdir, '1')))
def test_abort_mtime(self):
" commit aborted because mtime changed "
filename = os.path.join(self.tempdir, 'file.dat')
self.create_file(filename, 'record1')
ftrans = FileTransaction()
fp = ftrans.open(filename, 'w')
fp.write('record2')
time.sleep(1)
_fp = open(filename, 'w')
_fp.write('record3')
_fp.close()
self.assertRaises(FileTransactionException, ftrans.commit)
def test_abort_mtime(self):
" commit aborted because mtime changed "
filename = os.path.join(self.tempdir, 'file.dat')
self.create_file(filename, 'record1')
ftrans = FileTransaction()
fp = ftrans.open(filename, 'w')
fp.write('record2')
time.sleep(1)
_fp = open(filename, 'w')
_fp.write('record3')
_fp.close()
self.assertRaises(FileTransactionException, ftrans.commit)
def test_commit(self):
init_data = "init data,"
data1 = "test data one"
data2 = "test data two"
file1 = os.path.join(self.tempdir, "file1.dat")
file2 = os.path.join(self.tempdir, "file2.dat")
self.create_file(file1, init_data)
ftrans = FileTransaction()
fp1 = ftrans.open(file1, "a")
fp2 = ftrans.open(file2, "w")
fp1.write(data1)
fp2.write(data2)
ftrans.commit()
self.assertEqual(open(file1).read(), init_data + data1)
self.assertEqual(open(file2).read(), data2)
def test_commit(self):
init_data = "init data,"
data1 = "test data one"
data2 = "test data two"
file1 = os.path.join(self.tempdir, "file1.dat")
file2 = os.path.join(self.tempdir, "file2.dat")
self.create_file(file1, init_data)
ftrans = FileTransaction()
fp1 = ftrans.open(file1, "a")
fp2 = ftrans.open(file2, "w")
fp1.write(data1)
fp2.write(data2)
ftrans.commit()
self.assertEqual(open(file1).read(), init_data + data1)
self.assertEqual(open(file2).read(), data2)
def test_mkdir_commit(self):
ftrans = FileTransaction()
ftrans.mkdir(os.path.join(self.tempdir, '1'))
ftrans.mkdir(os.path.join(self.tempdir, '1', '2'))
fp = ftrans.open(os.path.join(self.tempdir, '1', '2', 'file'), 'w')
fp.write('data')
fp.close()
ftrans.commit()
del ftrans
self.assertTrue(
open(os.path.join(self.tempdir, '1', '2', 'file'), 'r').read(),
'data')
def test_mkdir_rollback(self):
ftrans = FileTransaction()
ftrans.mkdir(os.path.join(self.tempdir, '1'))
ftrans.mkdir(os.path.join(self.tempdir, '1', '2'))
fp = ftrans.open(os.path.join(self.tempdir, '1', '2', 'file'), 'w')
fp.write('data')
fp.close()
ftrans.rollback()
self.assertFalse(os.path.exists(os.path.join(self.tempdir, '1')))
def test_mkdir_commit(self):
ftrans = FileTransaction()
ftrans.mkdir(os.path.join(self.tempdir, '1'))
ftrans.mkdir(os.path.join(self.tempdir, '1', '2'))
fp = ftrans.open(os.path.join(self.tempdir, '1', '2', 'file'), 'w')
fp.write('data')
fp.close()
ftrans.commit()
del ftrans
self.assertTrue(open(os.path.join(self.tempdir, '1', '2', 'file'), 'r').read(), 'data')
def test_rollback(self):
init_data = "init data,"
data1 = "test data one"
data2 = "test data two"
file1 = os.path.join(self.tempdir, "file1.dat")
file2 = os.path.join(self.tempdir, "file2.dat")
self.create_file(file1, init_data)
ftrans = FileTransaction()
fp1 = ftrans.open(file1, "a")
fp2 = ftrans.open(file2, "w")
fp1.write(data1)
fp2.write(data2)
fp1.close()
fp2.close()
ftrans.rollback()
self.assertEqual(open(file1).read(), init_data)
self.assertFalse(os.path.exists(file2))
def test_rollback(self):
init_data = "init data,"
data1 = "test data one"
data2 = "test data two"
file1 = os.path.join(self.tempdir, "file1.dat")
file2 = os.path.join(self.tempdir, "file2.dat")
self.create_file(file1, init_data)
ftrans = FileTransaction()
fp1 = ftrans.open(file1, "a")
fp2 = ftrans.open(file2, "w")
fp1.write(data1)
fp2.write(data2)
fp1.close()
fp2.close()
ftrans.rollback()
self.assertEqual(open(file1).read(), init_data)
self.assertFalse(os.path.exists(file2))
def test_name_max(self):
name_max = os.pathconf(self.tempdir, os.pathconf_names['PC_NAME_MAX'])
file1 = os.path.join(
self.tempdir, ''.join(
random.choice(string.ascii_letters) for _ in range(name_max)))
ftrans = FileTransaction()
ftrans.open(file1, "w")
ftrans.commit()
self.assertTrue(os.path.exists(file1))
file2 = file1 + random.choice(string.ascii_letters)
ftrans = FileTransaction()
self.assertRaises(IOError, ftrans.open, file2, "w")
def test_name_max(self):
name_max = os.pathconf(self.tempdir, os.pathconf_names['PC_NAME_MAX'])
file1 = os.path.join(
self.tempdir,
''.join(random.choice(string.ascii_letters) for _ in range(name_max)))
ftrans = FileTransaction()
ftrans.open(file1, "w")
ftrans.commit()
self.assertTrue(os.path.exists(file1))
file2 = file1 + random.choice(string.ascii_letters)
ftrans = FileTransaction()
self.assertRaises(IOError, ftrans.open, file2, "w")
