You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
4.8 KiB
122 lines
4.8 KiB
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_ru, _ = Language.objects.update_or_create(
|
|
code='ru',
|
|
defaults={'name': 'Russian', '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_ru,
|
|
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": "Курс 1", "language_code": "ru"}
|
|
],
|
|
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 Russian student
|
|
notif = Notification.objects.filter(user=self.student, title="Живой урок отменен").first()
|
|
self.assertIsNotNone(notif)
|
|
self.assertIn("Lesson One Live", notif.message)
|
|
self.assertIn("Курс 1", 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("Курс 1", 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("Курс 1", notif.message)
|
|
self.assertTrue(mock_send.called)
|