from django.test import TestCase
from course_service.courses.models import Course
from lms.tools import EXAMPLE_BASE64

from course_service.courses.api import InApiTeacher


class CourseInApiTestCase(TestCase):
    def setUp(self):
        self.first_course = Course.objects.create(
            title='Первый курс',
            slug='perviy-kuourse',
        )
        self.CDTeacher = InApiTeacher()

    def test_teacher(self):
        token1 = "token1"
        token2 = "token2"
        token3 = "token3"
        self.CDTeacher.add_teacher(slug=self.first_course.slug, token=token1)
        self.CDTeacher.add_teacher(slug=self.first_course.slug, token=token2)
        self.CDTeacher.add_teacher(slug=self.first_course.slug, token=token3)
        self.CDTeacher.delete_teacher(slug=self.first_course.slug, token=token2)
        self.assertEqual(self.CDTeacher.get_token_list(self.first_course.slug), ['token1', 'token3'])


class CourseCRUDTest(TestCase):
    def setUp(self):
        self.first_course = Course.objects.update_or_create_course(
            title='Новый курс',
            level="Базовый",
            direction="Бизнес",
        )

    def test_slug(self):
        self.assertIsNotNone(self.first_course.slug)

    def test_upload_img(self):
        Course.objects.update_or_create_course(
            id=self.first_course.id,
            title='Новый курс',
            image=EXAMPLE_BASE64,
            big_image=EXAMPLE_BASE64,
            level="Базовый",
            direction="Бизнес",
        )

        self.assertIsNone(self.first_course.big_mobile_image)
        self.assertIsNotNone(self.first_course.image)