# -*- coding: utf-8 -*-

"""

 (c) 2016 - Copyright Red Hat Inc

 Authors:

   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>

   Farhaan Bukhsh <farhaan.bukhsh@gmail.com></farhaan.bukhsh@gmail.com>

"""

from __future__ import unicode_literals, absolute_import

import unittest

import shutil

import sys

import os

import hashlib

import six

from mock import patch

sys.path.insert(0, os.path.join(os.path.dirname(

    os.path.abspath(__file__)), '..'))

import pagure.lib.login

from pagure.exceptions import PagureException

import tests

class PagureLibLogintests(tests.Modeltests):

    """ Tests for pagure.lib.login """

    def test_id_generator(self):

        ''' Test pagure.lib.login.id_generator. '''

        self.assertEqual(

            pagure.lib.login.id_generator(size=3, chars=['a']),

            'aaa'

        )

    def test_get_session_by_visitkey(self):

        ''' Test pagure.lib.login.get_session_by_visitkey. '''

        session = pagure.lib.login.get_session_by_visitkey(self.session, 'foo')

        self.assertEqual(session, None)

    def test_generate_hashed_value(self):

        ''' Test pagure.lib.login.generate_hashed_value. '''

        password = pagure.lib.login.generate_hashed_value('foo')

        self.assertTrue(password.startswith('$2$'))

        self.assertEqual(len(password), 63)

    def test_check_password(self):

        ''' Test pagure.lib.login.check_password. '''

        # Version 2

        password = pagure.lib.login.generate_hashed_value('foo')

        self.assertTrue(

            pagure.lib.login.check_password('foo', password))

        self.assertFalse(

            pagure.lib.login.check_password('bar', password))

        # Version 1

        password = '%s%s' % ('foo', pagure.config.config.get('PASSWORD_SEED', None))

        if isinstance(password, six.text_type):

            password = password.encode('utf-8')

        password = '$1$' + hashlib.sha512(password).hexdigest()

        password = password.encode("utf-8")

        self.assertTrue(pagure.lib.login.check_password('foo', password))

        self.assertFalse(pagure.lib.login.check_password('bar', password))

        # Invalid password  -  No version

        password = '%s%s' % ('foo', pagure.config.config.get('PASSWORD_SEED', None))

        if isinstance(password, six.text_type):

            password = password.encode('utf-8')

        password = hashlib.sha512(password).hexdigest()

        password = password.encode("utf-8")

        self.assertRaises(

            PagureException,

            pagure.lib.login.check_password,

            'foo', password

        )

        # Invalid password  -  Invalid version

        password = b'$3$' + password

        self.assertRaises(

            PagureException,

            pagure.lib.login.check_password,

            'foo',

            password

        )

        password = '%s%s' % ('foo', pagure.config.config.get('PASSWORD_SEED', None))

        if isinstance(password, six.text_type):

            password = password.encode('utf-8')

        password = hashlib.sha512(password).hexdigest()

        password = password.encode("utf-8")

        self.assertRaises(

            PagureException,

            pagure.lib.login.check_password,

            'foo', password

        )

        # Invalid password  -  Invalid version

        password = b'$3$' + password

        self.assertRaises(

            PagureException,

            pagure.lib.login.check_password,

            'foo',

            password

        )

    def test_unicode_required(self):

        ''' Test to check for non-ascii password

        '''

        self.assertRaises(

            ValueError,

            pagure.lib.login.generate_hashed_value,

            'hunter2'.encode('utf-8')

        )

        password = pagure.lib.login.generate_hashed_value('foo')

        self.assertRaises(

            ValueError,

            pagure.lib.login.check_password,

            'foo'.encode('utf-8'),

            password

        )

if __name__ == '__main__':

    unittest.main(verbosity=2)