def test_document_path_and_type_update(self):
logger = Logger(path=TEST_LOG_PATH)
operation = {
'method': 'create_document',
'id': 456,
'name': 'first.txt',
'type': 'txt',
'path': '/tmp/first.txt'
}
logger.save_operation(operation)
operation = {
'method': 'update_document',
'id': 456,
'path': '/tmp/new.rst',
'type': 'rst'
}
logger.save_operation(operation)
context = Context()
logger.restore_context(context)
self.assertEqual(context.count_documents(), 1)
document = context.get_document(456)
self.assertEqual(document.id, 456)
self.assertEqual(document.name, 'first.txt')
self.assertEqual(document.type, 'rst')
self.assertEqual(document.path, '/tmp/new.rst')
def test_update_document(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
context.update_document(1, 'second.csv', 'csv', '/tmp/second.csv')
document = context.get_document(1)
self.assertEqual(document.id, 1)
self.assertEqual(document.name, 'second.csv')
self.assertEqual(document.type, 'csv')
self.assertEqual(document.path, '/tmp/second.csv')
def test_update_document_path(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
context.update_document(1, path='/tmp/here/first.txt')
document = context.get_document(1)
self.assertEqual(document.id, 1)
self.assertEqual(document.name, 'first.txt')
self.assertEqual(document.type, 'txt')
self.assertEqual(document.path, '/tmp/here/first.txt')
def test_document_creation(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
self.assertEqual(context.count_documents(), 1)
document = context.get_document(1)
self.assertEqual(document.id, 1)
self.assertEqual(document.name, 'first.txt')
self.assertEqual(document.type, 'txt')
self.assertEqual(document.path, '/tmp/first.txt')
def test_reuse_document_identifier(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
context.destroy_document(1)
context.create_document(1, 'other.dat', 'data', '/tmp/other.dat')
document = context.get_document(1)
self.assertEqual(document.id, 1)
self.assertEqual(document.name, 'other.dat')
self.assertEqual(document.type, 'data')
self.assertEqual(document.path, '/tmp/other.dat')
def test_missing_document(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
with self.assertRaises(ValueError):
_ = context.get_document(2)
def test_remove_document(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
context.destroy_document(1)
with self.assertRaises(ValueError):
_ = context.get_document(1)
