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')