2 changed files with 218 additions and 2 deletions
@ -0,0 +1,122 @@ |
|||||
|
import datetime |
||||
|
from django.utils import timezone |
||||
|
from django.test import TestCase |
||||
|
from unittest.mock import patch |
||||
|
from django.core.files.uploadedfile import SimpleUploadedFile |
||||
|
|
||||
|
from dj_language.models import Language |
||||
|
from apps.account.models import StudentUser, Notification |
||||
|
from apps.course.models.course import Course, CourseCategory |
||||
|
from apps.course.models.participant import Participant |
||||
|
from apps.course.models.live_session import CourseLiveSession, LiveSessionRecording |
||||
|
|
||||
|
class LiveNotificationsFlowTests(TestCase): |
||||
|
def setUp(self): |
||||
|
# Create languages |
||||
|
self.lang_fa, _ = Language.objects.update_or_create( |
||||
|
code='fa', |
||||
|
defaults={'name': 'Persian', 'status': True, 'countries': []} |
||||
|
) |
||||
|
self.lang_en, _ = Language.objects.update_or_create( |
||||
|
code='en', |
||||
|
defaults={'name': 'English', 'status': True, 'countries': []} |
||||
|
) |
||||
|
|
||||
|
# Create category |
||||
|
self.category = CourseCategory.objects.create( |
||||
|
name=[{"title": "Category", "language_code": "en"}], |
||||
|
slug=[{"title": "category", "language_code": "en"}] |
||||
|
) |
||||
|
|
||||
|
# Create student user |
||||
|
self.student = StudentUser.objects.create( |
||||
|
email='student_live@example.com', |
||||
|
username='student_live', |
||||
|
fullname='Live Student', |
||||
|
language=self.lang_fa, |
||||
|
fcm='token_live_student' |
||||
|
) |
||||
|
|
||||
|
# Create Course |
||||
|
thumbnail = SimpleUploadedFile('thumb.jpg', b'filecontent', content_type='image/jpeg') |
||||
|
self.course = Course.objects.create( |
||||
|
title=[ |
||||
|
{"title": "Course 1", "language_code": "en"}, |
||||
|
{"title": "دوره ۱", "language_code": "fa"} |
||||
|
], |
||||
|
slug=[{"title": "course-1", "language_code": "en"}], |
||||
|
category=self.category, |
||||
|
thumbnail=thumbnail, |
||||
|
video_type=Course.VedioTypeChoices.YOUTUBE_LINK, |
||||
|
video_link='https://example.com/video', |
||||
|
is_online=False, |
||||
|
level=Course.LevelChoices.BEGINNER, |
||||
|
duration=10, |
||||
|
lessons_count=0, |
||||
|
description='Description', |
||||
|
short_description='Short description', |
||||
|
status=[{"title": "ongoing", "language_code": "en"}], |
||||
|
is_free=True, |
||||
|
) |
||||
|
|
||||
|
# Enroll student in the course |
||||
|
self.participant = Participant.objects.create( |
||||
|
student=self.student, |
||||
|
course=self.course, |
||||
|
is_active=True |
||||
|
) |
||||
|
|
||||
|
# Create Live Session |
||||
|
self.session = CourseLiveSession.objects.create( |
||||
|
course=self.course, |
||||
|
subject='Lesson One Live', |
||||
|
started_at=timezone.now() + datetime.timedelta(days=1), |
||||
|
is_cancelled=False, |
||||
|
reminder_sent=False |
||||
|
) |
||||
|
|
||||
|
@patch('apps.account.notification_service.send_notification') |
||||
|
def test_live_class_cancelled_notification(self, mock_send): |
||||
|
# Event 1: LIVE Class Cancelled |
||||
|
self.session.is_cancelled = True |
||||
|
self.session.save() |
||||
|
|
||||
|
# Check DB notification for Persian student |
||||
|
notif = Notification.objects.filter(user=self.student, title="کلاس زنده لغو شد").first() |
||||
|
self.assertIsNotNone(notif) |
||||
|
self.assertIn("Lesson One Live", notif.message) |
||||
|
self.assertIn("دوره ۱", notif.message) |
||||
|
self.assertTrue(mock_send.called) |
||||
|
|
||||
|
@patch('apps.account.notification_service.send_notification') |
||||
|
def test_live_class_rescheduled_notification(self, mock_send): |
||||
|
# Event 2: LIVE Class Rescheduled |
||||
|
old_time = self.session.started_at |
||||
|
new_time = old_time + datetime.timedelta(hours=2) |
||||
|
|
||||
|
self.session.started_at = new_time |
||||
|
self.session.save() |
||||
|
|
||||
|
# Check DB notification |
||||
|
notif = Notification.objects.filter(user=self.student, title="کلاس زنده تغییر زمان یافت").first() |
||||
|
self.assertIsNotNone(notif) |
||||
|
self.assertIn("Lesson One Live", notif.message) |
||||
|
self.assertIn("دوره ۱", notif.message) |
||||
|
self.assertTrue(mock_send.called) |
||||
|
|
||||
|
@patch('apps.account.notification_service.send_notification') |
||||
|
def test_live_recording_available_notification(self, mock_send): |
||||
|
# Event 3: LIVE Recording Available |
||||
|
# Create a LiveSessionRecording with a dummy file |
||||
|
recording_file = SimpleUploadedFile('rec.mp4', b'video_data', content_type='video/mp4') |
||||
|
recording = LiveSessionRecording.objects.create( |
||||
|
session=self.session, |
||||
|
file=recording_file |
||||
|
) |
||||
|
|
||||
|
# Check DB notification |
||||
|
notif = Notification.objects.filter(user=self.student, title="ضبط کلاس زنده در دسترس است").first() |
||||
|
self.assertIsNotNone(notif) |
||||
|
self.assertIn("Lesson One Live", notif.message) |
||||
|
self.assertIn("دوره ۱", notif.message) |
||||
|
self.assertTrue(mock_send.called) |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue