Add crud functions to get ufora courses

tests/test_database/test_crud/conftest.py

@@ -0,0 +1,34 @@
+import datetime
+
+import pytest
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database.models import UforaAnnouncement, UforaCourse, UforaCourseAlias
+
+
+@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 course_with_alias(database_session: AsyncSession, course: UforaCourse) -> UforaCourse:
+    """Fixture to create a course with an alias"""
+    alias = UforaCourseAlias(course_id=course.course_id, alias="alias")
+    database_session.add(alias)
+    await database_session.commit()
+    await database_session.refresh(course)
+    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

tests/test_database/test_crud/test_ufora_announcements.py

@@ -7,24 +7,6 @@ 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)

tests/test_database/test_crud/test_ufora_courses.py

@@ -0,0 +1,22 @@
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database.crud import ufora_courses as crud
+from database.models import UforaCourse
+
+
+async def test_get_course_by_name_exact(database_session: AsyncSession, course: UforaCourse):
+    """Test getting a course by its name when the query is an exact match"""
+    match = await crud.get_course_by_name(database_session, "Test")
+    assert match == course
+
+
+async def test_get_course_by_name_substring(database_session: AsyncSession, course: UforaCourse):
+    """Test getting a course by its name when the query is a substring"""
+    match = await crud.get_course_by_name(database_session, "es")
+    assert match == course
+
+
+async def test_get_course_by_name_alias(database_session: AsyncSession, course_with_alias: UforaCourse):
+    """Test getting a course by its name when the name doesn't match, but the alias does"""
+    match = await crud.get_course_by_name(database_session, "ali")
+    assert match == course_with_alias