mirror of https://github.com/stijndcl/didier
49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
import datetime
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from database.crud import ufora_announcements as crud
|
|
from database.schemas import UforaAnnouncement, UforaCourse
|
|
|
|
|
|
async def test_get_courses_with_announcements_none(postgres: AsyncSession):
|
|
"""Test getting all courses with announcements when there are none"""
|
|
results = await crud.get_courses_with_announcements(postgres)
|
|
assert len(results) == 0
|
|
|
|
|
|
async def test_get_courses_with_announcements(postgres: 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)
|
|
postgres.add_all([course_1, course_2])
|
|
await postgres.commit()
|
|
|
|
results = await crud.get_courses_with_announcements(postgres)
|
|
assert len(results) == 1
|
|
assert results[0] == course_1
|
|
|
|
|
|
async def test_create_new_announcement(postgres: AsyncSession, ufora_course: UforaCourse):
|
|
"""Test creating a new announcement"""
|
|
await crud.create_new_announcement(postgres, 1, course=ufora_course, publication_date=datetime.datetime.now())
|
|
await postgres.refresh(ufora_course)
|
|
assert len(ufora_course.announcements) == 1
|
|
|
|
|
|
async def test_remove_old_announcements(postgres: AsyncSession, ufora_announcement: UforaAnnouncement):
|
|
"""Test removing all stale announcements"""
|
|
course = ufora_announcement.course
|
|
ufora_announcement.publication_date -= datetime.timedelta(weeks=2)
|
|
announcement_2 = UforaAnnouncement(course_id=ufora_announcement.course_id, publication_date=datetime.datetime.now())
|
|
postgres.add_all([ufora_announcement, announcement_2])
|
|
await postgres.commit()
|
|
await postgres.refresh(course)
|
|
assert len(course.announcements) == 2
|
|
|
|
await crud.remove_old_announcements(postgres)
|
|
|
|
await postgres.refresh(course)
|
|
assert len(course.announcements) == 1
|
|
assert announcement_2.course.announcements[0] == announcement_2
|