def test_get_document(self): scope = Scope(database=self._database) for document_id in range(1, 9): document = scope.get_document(document_id) self.assertEqual(document.id, document_id) self.assertEqual(document.name, sample_documents[document_id]['name']) self.assertEqual(document.type, sample_documents[document_id]['type']) self.assertEqual(document.path, sample_documents[document_id]['path'])
def test_get_invalid_document(self): scope = Scope(database=self._database) with self.assertRaises(ValueError): _ = scope.get_document(0) with self.assertRaises(ValueError): _ = scope.get_document(9)