from django.test import TestCase
from common.models import *
from django.contrib.auth.models import User
from django.urls import reverse

from django.core import mail

class UserManagerTests(TestCase):

    def setUp(self):
        institute = Institute.objects.create(name="CNRS",quota=999.0)

        period = Period.objects.create()

        science_theme = ScienceTheme.objects.create(name="Solar System")

        UserLevel.objects.create(name="Visitor",desc="Visitor description",priority=0,quota=0.0)

        UserLevel.objects.create(name="Observer",desc="Observer description",priority=2,quota=10.0)
        UserLevel.objects.create(name="Unit-PI",desc="Unit-PI description",priority=6,quota=100.0)
        UserLevel.objects.create(name="TAC",desc="TAC description",priority=3,quota=0)

        UserLevel.objects.create(name="Admin",desc="Admin description",priority=7,quota=1000.0)

        Country.objects.create(name="France")

        self.assertEqual(PyrosUser.objects.count(), 0, "There should be no User")
        path = reverse("user_signup_validation")
        response = self.client.post(path, {"email": "toto@titi.fr", "password": "aze", "password_confirm": "aze",
                                           "first_name": "toto", "last_name": "titi", "tel": "0123456789",
                                           "laboratory": "IRAP", "address": "ici","institute":institute.id,"reason":"this is a test",

                                           "roles":"Admin","question":"six","timer":"11","iambot":""

                                           })
        

        #sp1 = ScientificProgram.objects.create(name="sp1",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)
        #sp2 = ScientificProgram.objects.create(name="sp2",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)
        #sp3 = ScientificProgram.objects.create(name="sp3",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)
        #sp4 = ScientificProgram.objects.create(name="sp4",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)

        sp1 = ScientificProgram.objects.create(name="sp1",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)
        sp2 = ScientificProgram.objects.create(name="sp2",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)
        sp3 = ScientificProgram.objects.create(name="sp3",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)
        sp4 = ScientificProgram.objects.create(name="sp4",institute=institute,sp_pi=PyrosUser.objects.all().first(),science_theme=science_theme)

        SP_Period.objects.create(scientific_program=sp1,period=period)
        SP_Period.objects.create(scientific_program=sp2,period=period)
        SP_Period.objects.create(scientific_program=sp3,period=period)
        SP_Period.objects.create(scientific_program=sp4,period=period)

        self.assertTrue("success" in response.context.keys(), "There should be a success")
        self.assertEqual(PyrosUser.objects.count(), 1, "Theroue shld be one User")

    def test_creation(self):
        self.assertEqual(Country.objects.count(), 1, "There should be 1 Country")

        self.assertEqual(UserLevel.objects.count(), 5, "There should be 5 UserLevel")

        self.assertEqual(PyrosUser.objects.all()[0].first_name, 'toto')
        self.assertEqual(PyrosUser.objects.all()[0].email, 'toto@titi.fr')
        self.assertEqual(PyrosUser.objects.all()[0].user_level.filter(name="Visitor").count(),1,"There should be one UserLevel (=Visitor)")

    def test_login(self):
        self.assertEqual(Country.objects.count(), 1, "There should be 1 Country")

        self.assertEqual(UserLevel.objects.count(), 5, "There should be 5 UserLevel")

        # Activate user
        # La variable qui régit l'activation d'un compte est contenue dans pyrosUsers 
        # et s'appelle is_active, il suffit de passer cette variable à True
        current_user = PyrosUser.objects.all()[0]
        current_user.is_active=True

        current_user.set_password("aze")

        current_user.save()
        self.assertEqual(PyrosUser.objects.all()[0].is_active, True, "user should be active")

        # Log user

        path = reverse("login_validation")

        response = self.client.post(path, {"email": "toto@titi.fr", "password": "aze"})

        #self.assertTrue(response.context.get("success"))

        self.assertIn('_auth_user_id', self.client.session, "The user should be logged in")

    def test_login_not_active(self):
        self.assertEqual(Country.objects.count(), 1, "There should be 1 Country")

        self.assertEqual(UserLevel.objects.count(), 5, "There should be 5 UserLevel")

        # Activate user
        # La variable qui régit l'activation d'un compte est contenue dans pyrosUsers
        # et s'appelle is_active, il suffit de passer cette variable à True

        self.assertEqual(PyrosUser.objects.all()[0].is_active, False, "user should not be active")

        # Log user
        path = "/user_manager/login"
        response = self.client.post(path, {"email": "toto@titi.fr", "password": "aze"})

        #self.assertFalse(response.context.get("success"))

        self.assertNotIn('_auth_user_id', self.client.session, "The user should be logged in")

    def test_wrong_email(self):
        path = "/user_manager/login"
        response = self.client.post(path, {"email": "toto@tti.fr", "password": "aze"})
        self.assertIn("error", response.context.keys(), "There should be an error")
        self.assertNotIn('_auth_user_id', self.client.session, "There shouldn't be an authentified user")

    def test_wrong_password(self):
        path = "/user_manager/login"
        response = self.client.post(path, {"email": "toto@titi.fr", "password": "azee"})
        self.assertIn("error", response.context.keys(), "There should be an error")
        self.assertNotIn('_auth_user_id', self.client.session, "There shouldn't be an authentified user")

    def test_logout(self):
        self.client.login(username="toto@titi.fr", password="aze")
        path = "/user_manager/logout"
        self.client.get(path)
        self.assertNotIn('_auth_user_id', self.client.session, "There shouldn't be an authentified user")

    def test_get_user_priority(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        self.assertEqual(user.get_priority(),UserLevel.objects.get(name="Unit-PI").priority,"The priority should be equal to Unit-PI's priority")

    def test_wrong_get_user_priority(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        self.assertNotEqual(user.get_priority(),UserLevel.objects.get(name="Visitor").priority,"The priority shouldn't be equal to Visitor's priority")

    def test_get_roles_str(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        roles_str = user.get_roles_str()
        for role in user.user_level.all():
            self.assertIn(role.name,roles_str,f"The role {role} should be in the str representation")
    
    def test_wrong_get_roles_str(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        roles_str = user.get_roles_str()
        self.assertNotIn(UserLevel.objects.get(name="Admin").name,roles_str,"The role Admin shouldn't be in the str representation")

    def test_max_priority_desc(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        self.assertEqual(user.get_max_priority_desc(),UserLevel.objects.get(name="Unit-PI").desc,"The desc of user_level should be 'Unit-PI description' ")

    def test_wrong_max_priority_desc(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        # add Admin role to user (has a greatier priority than Unit-PI and Visitor roles)
        UserLevel.objects.get(name="Admin").pyros_users.add(user)
        self.assertNotEqual(user.get_max_priority_desc(),UserLevel.objects.get(name="Unit-PI").desc,"The desc of user_level shouldn't be 'Unit-PI description' ")

        
    def test_max_priority_quota(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        self.assertEqual(user.get_max_priority_quota(),UserLevel.objects.get(name="Unit-PI").quota,"The quota of user_level should be 100.0 ")

    def test_wrong_max_priority_quota(self):
        user = PyrosUser.objects.all()[0]
        # add Unit-PI role to user
        UserLevel.objects.get(name="Unit-PI").pyros_users.add(user)
        # add Admin role to user (has a greatier priority than Unit-PI and Visitor roles)
        UserLevel.objects.get(name="Admin").pyros_users.add(user)

        self.assertNotEqual(user.get_max_priority_quota(),UserLevel.objects.get(name="Unit-PI").desc,"The quota of user_level shouldn't be 1000.0 ")


    # new test of 2021

    def test_USR_can_register(self):
        institute = Institute.objects.get(name="CNRS")
        path = reverse("user_signup_validation")
        response = self.client.post(path, {"email": "unit_pi@toto.fr", "password": "aze", "password_confirm": "aze",
                                           "first_name": "unit", "last_name": "pi", "tel": "0123456789",
                                           "laboratory": "IRAP", "address": "ici","institute":institute.id,"reason":"this is a test",

                                           "roles":"Admin","question":"six","timer":"11","iambot":""

                                           })
        # We need to create an Unit-PI user
        unit_pi = PyrosUser.objects.get(username="unit_pi@toto.fr")
        unit_pi.user_level.set([UserLevel.objects.get(name="Unit-PI")])
        unit_pi.is_active = True
        unit_pi.set_password("password")
        unit_pi.save()
        path = reverse("user_signup_validation")
        response = self.client.post(path, {"email": "toto@titi.fr", "password": "aze", "password_confirm": "aze",
                                           "first_name": "toto", "last_name": "titi", "tel": "0123456789",
                                           "laboratory": "IRAP", "address": "ici","institute":institute.id,"reason":"this is a test",

                                           "roles":"Admin","question":"six","timer":"11","iambot":""

                                           })
        new_user = PyrosUser.objects.filter(username="toto@titi.fr").order_by('-id')[0]
        #new_user = PyrosUser.objects.get(username="toto@titi.fr")
        self.assertEqual(new_user.get_roles_str(),"Visitor")
        self.assertEqual(len(mail.outbox),2)
        self.assertIn("toto@titi.fr",mail.outbox[0].recipients())
        self.assertIn("unit_pi@toto.fr",mail.outbox[1].recipients())
        # log in as unit pi
        self.client.post(reverse("login_validation"),{"email":"unit_pi@toto.fr","password":"password"})
        self.assertEqual(int(self.client.session['_auth_user_id']), unit_pi.pk)
        path = reverse("user-edit",kwargs={"pk":new_user.id})
        response = self.client.post(path, {"username":new_user.username,"first_name":new_user.first_name,"last_name":new_user.last_name,"email":new_user.email,"country":Country.objects.all()[0].id,"institute":institute.id,"roles":[UserLevel.objects.get(name="Observer").id],"desc":"","tel":"","adress":"","laboratory":"test"})
        # new user should be Observer now
        self.assertIn("Observer",new_user.get_roles_str())
        # set new user to active
        path = reverse("change_activate",kwargs={"pk":new_user.id,"current_user_id":unit_pi.id})
        response = self.client.get(path)
        new_user = PyrosUser.objects.filter(username="toto@titi.fr").order_by('-id')[0]
        self.assertEqual(response.status_code,302)
        self.assertTrue(new_user.is_active)
        # log out as unit_pi
        path = reverse("user_logout")
        self.client.get(path)
        # log in as new user (toto)
        response = self.client.post(reverse("login_validation"),{"email":"toto@titi.fr","password":"aze"})
        self.assertEqual(int(self.client.session['_auth_user_id']), new_user.pk)

        response = self.client.get(reverse("index_scientific_program"))

        # new user has access to the page
        self.assertEqual(response.status_code,200)

    def test_USR_user_cannot_connect_with_bad_login(self):
        user = PyrosUser.objects.get(username="toto@titi.fr")
        user.is_active = True
        user.save()
        # user password is hashed in database, we need to use the raw password
        user_password = "aze"
        response = self.client.post(reverse("login_validation"),{"email":user.email+"bad","password":user_password})
        self.assertEqual(response.status_code,200)
        self.assertContains(response,"Your email and/or password were incorrect.")

    def test_USR_user_cannot_connect_with_bad_pass(self):
        user = PyrosUser.objects.get(username="toto@titi.fr")
        user.is_active = True
        user.save()
        # user password is hashed in database, we need to use the raw password
        user_password = "aze"
        response = self.client.post(reverse("login_validation"),{"email":user.email,"password":user_password+"bad"})
        self.assertEqual(response.status_code,200)
        self.assertContains(response,"Your email and/or password were incorrect.")
        
    def test_USR_can_connect_and_disconnect(self):
        user = PyrosUser.objects.get(username="toto@titi.fr")
        user.is_active = True
        user.save()
        response = self.client.post(reverse("login_validation"),{"email":"toto@titi.fr","password":"aze"})
        self.assertEqual(int(self.client.session['_auth_user_id']), user.pk)

        self.assertEqual(response.status_code,302)

        # test if user can access his profile page