Increase coverage

tests/test_database/test_crud/test_ufora_announcements.py

@@ -0,0 +1,67 @@
+import datetime
+
+import pytest
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database.crud import ufora_announcements as crud
+from database.models import UforaAnnouncement, UforaCourse
+
+
+@pytest.fixture
+async def course(database_session: AsyncSession) -> UforaCourse:
+    """Fixture to create a course"""
+    course = UforaCourse(name="test", code="code", year=1, log_announcements=True)
+    database_session.add(course)
+    await database_session.commit()
+    return course
+
+
+@pytest.fixture
+async def announcement(course: UforaCourse, database_session: AsyncSession) -> UforaAnnouncement:
+    """Fixture to create an announcement"""
+    announcement = UforaAnnouncement(course_id=course.course_id, publication_date=datetime.datetime.now())
+    database_session.add(announcement)
+    await database_session.commit()
+    return announcement
+
+
+async def test_get_courses_with_announcements_none(database_session: AsyncSession):
+    """Test getting all courses with announcements when there are none"""
+    results = await crud.get_courses_with_announcements(database_session)
+    assert len(results) == 0
+
+
+async def test_get_courses_with_announcements(database_session: AsyncSession):
+    """Test getting all courses with announcements"""
+    course_1 = UforaCourse(name="test", code="code", year=1, log_announcements=True)
+    course_2 = UforaCourse(name="test2", code="code2", year=1, log_announcements=False)
+    database_session.add_all([course_1, course_2])
+    await database_session.commit()
+
+    results = await crud.get_courses_with_announcements(database_session)
+    assert len(results) == 1
+    assert results[0] == course_1
+
+
+async def test_create_new_announcement(course: UforaCourse, database_session: AsyncSession):
+    """Test creating a new announcement"""
+    await crud.create_new_announcement(database_session, 1, course=course, publication_date=datetime.datetime.now())
+    await database_session.refresh(course)
+    assert len(course.announcements) == 1
+
+
+async def test_remove_old_announcements(announcement: UforaAnnouncement, database_session: AsyncSession):
+    """Test removing all stale announcements"""
+    course = announcement.course
+    announcement.publication_date -= datetime.timedelta(weeks=2)
+    announcement_2 = UforaAnnouncement(course_id=announcement.course_id, publication_date=datetime.datetime.now())
+    database_session.add_all([announcement, announcement_2])
+    await database_session.commit()
+    await database_session.refresh(course)
+    assert len(course.announcements) == 2
+
+    await crud.remove_old_announcements(database_session)
+
+    await database_session.refresh(course)
+    assert len(course.announcements) == 1
+    assert announcement_2.course.announcements[0] == announcement_2