chore: timeline 테스트 코드 작성 (#98)

* 타임라인 테스트 코드를 작성했습니다

Author: mjy926

wacruit/src/tests/timeline/conftest.py

@@ -0,0 +1,121 @@
+from datetime import datetime
+from typing import List
+
+import pytest
+from sqlalchemy.orm import Session
+
+from wacruit.src.apps.timeline.models import Timeline
+from wacruit.src.apps.timeline.models import TimelineCategory
+from wacruit.src.apps.timeline.models import TimelineGroupType
+from wacruit.src.apps.timeline.repositories import TimelineRepository
+from wacruit.src.apps.timeline.schemas import TimelineCategoryCreateUpdateRequest
+from wacruit.src.apps.timeline.schemas import TimelineCreateRequest
+from wacruit.src.apps.timeline.schemas import TimelineUpdateRequest
+from wacruit.src.apps.timeline.services import TimelineService
+from wacruit.src.apps.user.models import User
+from wacruit.src.database.connection import Transaction
+
+
+@pytest.fixture
+def timeline_repository(db_session: Session):
+    return TimelineRepository(session=db_session, transaction=Transaction(db_session))
+
+
+@pytest.fixture
+def timeline_service(timeline_repository: TimelineRepository) -> TimelineService:
+    return TimelineService(timeline_repository=timeline_repository)
+
+
+@pytest.fixture
+def timeline_category1(db_session: Session) -> TimelineCategory:
+    timeline_category = TimelineCategory(title="친목")
+    db_session.add(timeline_category)
+    db_session.commit()
+    return timeline_category
+
+
+@pytest.fixture
+def timeline_category2(db_session: Session) -> TimelineCategory:
+    timeline_category = TimelineCategory(title="세미나")
+    db_session.add(timeline_category)
+    db_session.commit()
+    return timeline_category
+
+
+@pytest.fixture
+def created_category_list(db_session: Session) -> List[TimelineCategory]:
+    category_list = []
+    for i in range(0, 5):
+        timeline_category = TimelineCategory(title=f"카테고리 {i}")
+        db_session.add(timeline_category)
+        category_list.append(timeline_category)
+    db_session.commit()
+    return category_list
+
+
+@pytest.fixture
+def created_timeline(
+    db_session: Session, timeline_category1: TimelineCategory
+) -> Timeline:
+    timeline = Timeline(
+        title="Created Timeline",
+        group=TimelineGroupType.ROOKIE,
+        category_id=timeline_category1.id,
+        start_date=datetime(2025, 1, 1),
+        end_date=datetime(2025, 1, 5),
+    )
+    db_session.add(timeline)
+    db_session.commit()
+    return timeline
+
+
+@pytest.fixture
+def created_timeline_list(
+    db_session: Session, timeline_category1: TimelineCategory
+) -> List[Timeline]:
+    timeline_list = []
+
+    for i in range(0, 5):
+        timeline = Timeline(
+            title=f"Created Timeline {i}",
+            group=TimelineGroupType.ROOKIE,
+            category_id=timeline_category1.id,
+            start_date=datetime(2025, 1, 1),
+            end_date=datetime(2025, 1, 5),
+        )
+        db_session.add(timeline)
+        timeline_list.append(timeline)
+    db_session.commit()
+    return timeline_list
+
+
+@pytest.fixture
+def timeline_create_dto(timeline_category1: TimelineCategory) -> TimelineCreateRequest:
+    return TimelineCreateRequest(
+        title="New Timeline",
+        group=TimelineGroupType.ROOKIE,
+        category_id=timeline_category1.id,
+        start_date=datetime(2025, 1, 1),
+        end_date=datetime(2025, 1, 5),
+    )
+
+
+@pytest.fixture
+def timeline_update_dto(timeline_category2: TimelineCategory) -> TimelineUpdateRequest:
+    return TimelineUpdateRequest(
+        title="Updated Timeline",
+        group=TimelineGroupType.PROGRAMMER,
+        category_id=timeline_category2.id,
+        start_date=datetime(2025, 3, 1),
+        end_date=datetime(2025, 3, 5),
+    )
+
+
+@pytest.fixture
+def category_create_dto() -> TimelineCategoryCreateUpdateRequest:
+    return TimelineCategoryCreateUpdateRequest(title="친목")
+
+
+@pytest.fixture
+def category_update_dto() -> TimelineCategoryCreateUpdateRequest:
+    return TimelineCategoryCreateUpdateRequest(title="세미나")

wacruit/src/tests/timeline/test_timeline_service.py

@@ -0,0 +1,100 @@
+from datetime import datetime
+from typing import List
+
+import pytest
+
+from wacruit.src.apps.timeline.exceptions import TimelineCategoryNotFoundException
+from wacruit.src.apps.timeline.exceptions import TimelineNotFoundException
+from wacruit.src.apps.timeline.models import Timeline
+from wacruit.src.apps.timeline.models import TimelineCategory
+from wacruit.src.apps.timeline.schemas import TimelineGroupType
+from wacruit.src.apps.timeline.services import TimelineService
+
+
+def test_create_category(timeline_service: TimelineService, category_create_dto):
+    response = timeline_service.create_category(category_create_dto)
+
+    assert response.title == "친목"
+
+
+def test_update_category(
+    timeline_service: TimelineService,
+    timeline_category1: TimelineCategory,
+    category_update_dto,
+):
+    response = timeline_service.update_category(
+        timeline_category1.id, category_update_dto
+    )
+
+    assert response.title == "세미나"
+
+
+def test_get_all_categories(
+    timeline_service: TimelineService, created_category_list: List[TimelineCategory]
+):
+    response = timeline_service.get_all_categories()
+    category_list = response.items
+
+    assert len(created_category_list) == len(category_list)
+    for i in range(0, 5):
+        assert category_list[i].title == f"카테고리 {i}"
+
+
+def test_delete_category(timeline_service: TimelineService, timeline_category1):
+    created_id = timeline_category1.id
+    timeline_service.delete_category(created_id)
+    with pytest.raises(TimelineCategoryNotFoundException):
+        timeline_service.delete_category(created_id)
+
+
+def test_create_timeline(timeline_service: TimelineService, timeline_create_dto):
+    response = timeline_service.create_timeline(timeline_create_dto)
+
+    assert response.title == "New Timeline"
+    assert response.group == TimelineGroupType.ROOKIE
+    assert response.category.title == "친목"
+    assert response.start_date == datetime(2025, 1, 1)
+    assert response.end_date == datetime(2025, 1, 5)
+
+
+def test_update_timeline(
+    timeline_service: TimelineService, created_timeline: Timeline, timeline_update_dto
+):
+    response = timeline_service.update_timeline(
+        created_timeline.id, timeline_update_dto
+    )
+
+    assert response.title == "Updated Timeline"
+    assert response.group == TimelineGroupType.PROGRAMMER
+    assert response.category.title == "세미나"
+    assert response.start_date == datetime(2025, 3, 1)
+    assert response.end_date == datetime(2025, 3, 5)
+
+
+def test_get_all_timelines(
+    timeline_service: TimelineService, created_timeline_list: List[Timeline]
+):
+    response = timeline_service.get_all_timelines(TimelineGroupType.ROOKIE)
+    timeline_list = response.items
+
+    assert len(timeline_list) == len(created_timeline_list)
+    for i in range(0, 5):
+        assert timeline_list[i].title == f"Created Timeline {i}"
+        assert timeline_list[i].group == TimelineGroupType.ROOKIE
+        assert timeline_list[i].category.title == "친목"
+        assert timeline_list[i].start_date == datetime(2025, 1, 1)
+        assert timeline_list[i].end_date == datetime(2025, 1, 5)
+
+
+def test_delete_timeline_success(
+    timeline_service: TimelineService, created_timeline: Timeline
+):
+    created_id = created_timeline.id
+    timeline_service.delete_timeline(created_id)
+    with pytest.raises(TimelineNotFoundException):
+        timeline_service.get_timeline_by_id(created_id)
+
+
+def test_delete_timeline_not_found(timeline_service: TimelineService):
+    with pytest.raises(TimelineNotFoundException):
+        timeline_service.delete_timeline(999)