# coding=utf-8

import unittest
from server_apis import database
from server_apis.api_core.models.regdb.user import User
from server_apis.api_core.modules.user_profile import registerProfile, makeLoginFromPhone
import server_apis.api_error as api_error


class RegistrationTestCase(unittest.TestCase):
    test_phone_belarus = "+375000000000"
    test_phone = "+70000000000"
    test_email = "sc0rch@me.com"

    def deleteProfile(self, session, phone):
        user = database.model(session, User).filter_one(
            User.login == makeLoginFromPhone(phone),
            User.date_deleted == None,
            must_exists=False)
        if user:
            session.delete(user)

    @database.transaction
    def test_Error_NullParameters(self):
        def unit(session):
            with self.assertRaises(api_error.APIValidationError):
                registerProfile(
                    session,
                    phone=None,
                    email=self.test_email, password="1234")
            session.rollback()
        return unit

    @database.transaction
    def test_Error_WrongParameter_Phone(self):
        def unit(session):
            with self.assertRaises(api_error.APIValidationError):
                registerProfile(
                    session,
                    phone="7000000000",
                    email=self.test_email, password="1234")
            session.rollback()
        return unit

    @database.transaction
    def test_Error_WrongParameter_Phone_Belarus(self):
        def unit(session):
            with self.assertRaises(api_error.APIValidationError):
                registerProfile(
                    session,
                    phone="+37500000000",
                    email=self.test_email, password="1234")
            session.rollback()
        return unit

    @database.transaction
    def test_Error_EmptyParameter_Phone(self):
        def unit(session):
            with self.assertRaises(api_error.APIValidationError):
                registerProfile(
                    session,
                    phone="",
                    email=self.test_email, password="1234")
            session.rollback()
        return unit

    @database.transaction
    def test_Error_EmptyParameter_Email(self):
        def unit(session):
            with self.assertRaises(api_error.APIValidationError):
                registerProfile(
                    session,
                    phone=self.test_phone,
                    email="", password="1234")
            session.rollback()
        return unit

    @database.transaction
    def test_Error_EmptyParameter_Password(self):
        def unit(session):
            with self.assertRaises(api_error.APIValidationError):
                registerProfile(
                    session,
                    phone=self.test_phone,
                    email=self.test_email, password='')
            session.rollback()
        return unit

    @database.transaction
    def test_Success(self):
        def unit(session):
            self.deleteProfile(session, self.test_phone)
            registerProfile(
                session,
                phone=self.test_phone,
                email=self.test_email, password='1234')

            with self.assertRaises(api_error.APIError):
                registerProfile(
                    session,
                    phone=self.test_phone,
                    email="other@mail.com", password='1234')
            self.deleteProfile(session, self.test_phone)
            session.rollback()
        return unit

    @database.transaction
    def test_Success_Belarus(self):
        def unit(session):
            self.deleteProfile(session, self.test_phone_belarus)
            registerProfile(
                session,
                phone=self.test_phone_belarus,
                email=self.test_email, password='1234')

            with self.assertRaises(api_error.APIError):
                registerProfile(
                    session,
                    phone=self.test_phone_belarus,
                    email="other@mail.com", password='1234')
            self.deleteProfile(session, self.test_phone_belarus)
            session.rollback()
        return unit


if __name__ == '__main__':
    unittest.main()