def setUp(self):
self.test_file_base = os.path.join(TEST_DIRECTORY, TEST_DATA_DIRECTORY, 'data')
self.test_file_path = self.test_file_base + '.test'
backup_file_path = self.test_file_base
# reset contents
test_file = open(self.test_file_path, 'w')
backup_file = open(backup_file_path, 'r')
test_file.write(backup_file.read())
test_file.close()
self.bf = BaseFile(self.test_file_base, 'test')
self.bf2 = BaseFile(self.test_file_base, 'test')
class TestCRF(unittest.TestCase):
def setUp(self):
self.test_file_base = os.path.join(TEST_DIRECTORY, TEST_DATA_DIRECTORY, 'data')
self.test_file_path = self.test_file_base + '.test'
backup_file_path = self.test_file_base
# reset contents
test_file = open(self.test_file_path, 'w')
backup_file = open(backup_file_path, 'r')
test_file.write(backup_file.read())
test_file.close()
self.bf = BaseFile(self.test_file_base, 'test')
self.bf2 = BaseFile(self.test_file_base, 'test')
def tearDown(self):
self.bf.close()
self.bf2.close()
def read_file(self, length=-1):
return open(self.test_file_path, 'r').read(length)
def test_read_base(self):
self.assertEqual(self.bf.read(), self.read_file())
self.assertEqual(self.bf.read(10), self.read_file(10))
def test_write_base(self):
content = 'blah'
self.bf.save(content)
self.assertEqual(self.bf.read(), self.read_file())
self.assertEqual(self.bf.read(2), self.read_file(2))
def test_write_lock(self):
self.bf.open('w')
self.assertTrue(os.path.exists(self.bf.lock_path))
try:
open(self.bf.lock_path, 'w')
self.assertTrue(False)
except:
pass
def test_write_lock_open(self):
self.bf.open('w')
self.assertFalse(self.bf2.open('w'))
self.assertFalse(self.bf2.open('r'))
self.bf.close()
self.assertTrue(self.bf2.open('w'))
self.assertFalse(self.bf2.open('r'))
self.bf2.close()
self.assertTrue(self.bf2.open('r'))
self.assertTrue(self.bf2.open('w'))
