def test_empty_context(self): logger = Logger(path=TEST_LOG_PATH) context = Context() logger.restore_context(context) self.assertEqual(context.count_documents(), 0) self.assertEqual(context.count_tags(), 0) self.assertEqual(context.count_relations(), 0)
def test_relation_creation(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': 'create_tag', 'id': 123, 'name': 'python'} logger.save_operation(operation) operation = { 'method': 'create_relation', 'document_id': 456, 'tag_id': 123 } logger.save_operation(operation) context = Context() logger.restore_context(context) self.assertEqual(context.count_documents(), 1) self.assertEqual(context.count_tags(), 1) self.assertEqual(context.count_relations(), 1) self.assertEqual(context.find_document_ids([123]), [456]) self.assertEqual(context.find_tag_ids([456]), [123])
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_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_document_creation_and_remove(self): logger = Logger(path=TEST_LOG_PATH) operation = { 'method': 'create_document', 'id': 1, 'name': 'first.txt', 'type': 'txt', 'path': '/tmp/first.txt' } logger.save_operation(operation) operation = {'method': 'destroy_document', 'id': 1} logger.save_operation(operation) context = Context() logger.restore_context(context) self.assertEqual(context.count_documents(), 0)
def test_document_counting(self): context = Context() self.assertEqual(context.count_documents(), 0) context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt') self.assertEqual(context.count_documents(), 1) context.create_document(2, 'second.csv', 'csv', '/tmp/second.csv') self.assertEqual(context.count_documents(), 2) context.create_document(3, 'other.dat', 'data', '/tmp/other.dat') self.assertEqual(context.count_documents(), 3) context.destroy_document(3) self.assertEqual(context.count_documents(), 2) context.destroy_document(1) self.assertEqual(context.count_documents(), 1) context.destroy_document(2) self.assertEqual(context.count_documents(), 0)
def test_empty_database(self): context = Context() self.assertEqual(context.count_documents(), 0) self.assertEqual(context.count_tags(), 0) self.assertEqual(context.count_relations(), 0)