from django.test import TestCase
from django.contrib.gis.geos import Point

# Local imports
from private_app.models import (
    Address,
    Favorite,
    Link,
    School,
    StudyProgram,
    User,
    UserProfile,
)


class ModelTests(TestCase):
    @classmethod
    def setUpTestData(cls):
        # Create an address
        coordinates = Point(47.09624640173994, -1.6599277027341754)
        address = Address.objects.create(
            street_address="1 rue de Test",
            postcode=44000,
            locality="Testville",
            geolocation=coordinates,
        )

        # Create a school website link longer than 30char
        link = Link.objects.create(
            link_type="SchoolWebsite", link_url="http://www.testing.com/123aaaaaaaaaa"
        )

        # Create a school with the above address and link
        school = School.objects.create(
            name="Test University",
            school_url=link,
            description="Test University was founded in Testville in 3023 to provide a well-rounded testing education to aspiring testers",
            address=address,
            school_type="private",
        )

        # Create a study program at the above school
        program = StudyProgram.objects.create(
            cod_aff_form=123456,
            name="Django and Angular Testing Studies",
            school=school,
            discipline="Sciences_Sante",
            url_parcoursup=Link.objects.create(
                link_type="ParcoursSupliink",
                link_url="https://dossier.parcoursup.fr/Candidats/public/fiches/afficherFicheFormation?g_ta_cod=123456",
            ),
            acceptance_rate="73.2",
            L1_success_rate="37.7",
            insertion_rate="64.8",
            insertion_time_period=12,
            description="Learn to test Djangular applications like the best!",
        )

        # Create a user
        user = User.objects.create(
            first_name="Ted",
            last_name="Testerton",
            username="TedTest",
            email="ted@test.com",
            password="Testing123!",
        )

        # Create a user profile set to private by default and no student_at attribute
        profile = UserProfile.objects.create(
            user=user,
            url_tiktok=Link.objects.create(
                link_type="Tiktok",
                link_url="http://www.tiktok.com/user/TedTestington123",
            ),
            url_instagram=Link.objects.create(
                link_type="Instagram",
                link_url="http://www.instagram.com/user/TedTestington123",
            ),
            about_me="I'm Ted Testerton and I am looking for a great study program.",
        )

        # Create a favorite with the user and program created above
        favorite = Favorite.objects.create(
            user=user,
            study_program=program,
            note="This program looks like one to test!",
            status="interested",
        )


class YourTestClass(TestCase):
    @classmethod
    def setUpTestData(cls):
        print(
            "setUpTestData: Run once to set up non-modified data for all class methods."
        )
        pass