added infrastructure code

tests/integration/annotation_repository_test.py

@@ -0,0 +1,86 @@
+"""Integration tests for AnnotationRepository."""
+
+from uuid import UUID
+import pytest
+from data_store.repositories.annotation_repository import AnnotationRepository
+from data_store.dto import (
+    Annotation,
+)
+
+
+def test_get_annotation(
+    annotation_in_minio: tuple[UUID, Annotation],
+    annotation_repository: AnnotationRepository,
+) -> None:
+    """Test retrieving an annotation from AnnotationRepository."""
+    # Arrange
+    job_id, expected_annotation = annotation_in_minio
+    # Act
+    retrieved_annotation = annotation_repository.get(job_id)
+    # Assert
+    assert retrieved_annotation is not None
+    assert isinstance(retrieved_annotation, Annotation)
+    assert retrieved_annotation == expected_annotation
+
+
+def test_put_annotation(
+    annotation_repository: AnnotationRepository,
+    annotation: Annotation,
+    job_id_list: list[UUID],
+) -> None:
+    """Test storing an annotation in AnnotationRepository."""
+    # Arrange
+    job_id = job_id_list[0]
+    assert annotation_repository.get(job_id) is None
+    # Act
+    annotation_repository.put(annotation, job_id)
+    # Assert
+    received_annotation = annotation_repository.get(job_id)
+    assert received_annotation is not None
+    assert isinstance(received_annotation, Annotation)
+    assert received_annotation == annotation
+
+
+def test_delete_annotation(
+    annotation_in_minio: tuple[UUID, Annotation],
+    annotation_repository: AnnotationRepository,
+) -> None:
+    """Test deleting an annotation from AnnotationRepository."""
+    # Arrange
+    job_id, _ = annotation_in_minio
+    assert annotation_repository.get(job_id) is not None
+    # Act
+    annotation_repository.delete(job_id)
+    # Assert
+    assert annotation_repository.get(job_id) is None
+
+
+def test_list_all_annotations(
+    annotation_in_minio: tuple[UUID, Annotation],
+    annotation_repository: AnnotationRepository,
+) -> None:
+    """Test listing all annotations in AnnotationRepository."""
+    # Arrange
+    job_id, _ = annotation_in_minio
+    # Act
+    annotation_ids = annotation_repository.list_all()
+    # Assert
+    assert isinstance(annotation_ids, list)
+    assert all(isinstance(i, UUID) for i in annotation_ids)
+    assert job_id in annotation_ids
+
+
+def test_list_all_annotations_empty(
+    annotation_repository: AnnotationRepository,
+) -> None:
+    """Test listing all annotations in an empty AnnotationRepository."""
+    # Act
+    annotation_ids = annotation_repository.list_all()
+    # Assert
+    assert isinstance(annotation_ids, list)
+    assert len(annotation_ids) == 0
+
+
+# Allows local debugging by running file as script
+if __name__ == "__main__":
+    pytest.main(["-s", "-v", __file__])