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_update_missing_document(self):
context = Context()
context.create_document(1, 'first.txt', 'txt', '/tmp/first.txt')
with self.assertRaises(ValueError):
context.update_document(2, path='/tmp/nowhere.txt', type='missing')
