import sqlalchemy
from pytest import fail

from app.main.dao import verify_codes_dao
from app.main.encryption import check_hash
from tests.app.main import create_test_user, create_another_test_user


def test_insert_new_code_and_get_it_back(notifications_admin, notifications_admin_db, notify_db_session):
    user = create_test_user('pending')

    verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='email')
    saved_codes = verify_codes_dao.get_codes(user_id=user.id, code_type='email')
    assert len(saved_codes) == 1
    saved_code = saved_codes[0]
    assert saved_code.user_id == user.id
    assert check_hash('12345', saved_code.code)
    assert saved_code.code_type == 'email'
    assert saved_code.code_used is False


def test_insert_new_code_should_thrw_exception_when_type_does_not_exist(notifications_admin,
                                                                        notifications_admin_db,
                                                                        notify_db_session):
    user = create_test_user('pending')
    try:
        verify_codes_dao.add_code(user_id=user.id, code='23545', code_type='not_real')
        fail('Should have thrown an exception')
    except sqlalchemy.exc.DataError as e:
        assert 'invalid input value for enum verify_code_types: "not_real"' in e.orig.pgerror


def test_should_throw_exception_when_user_does_not_exist(notifications_admin,
                                                         notifications_admin_db,
                                                         notify_db_session):
    try:
        verify_codes_dao.add_code(user_id=1, code='12345', code_type='email')
        fail('Should throw exception')
    except sqlalchemy.exc.IntegrityError as e:
        assert 'ERROR:  insert or update on table "verify_codes" violates ' \
               'foreign key constraint "verify_codes_user_id_fkey"' in e.orig.pgerror


def test_should_return_none_if_code_is_used(notifications_admin,
                                            notifications_admin_db,
                                            notify_db_session):
    user = create_test_user('pending')

    verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='email')
    verify_codes_dao.use_code(user_id=user.id, code='12345', code_type='email')
    saved_code = verify_codes_dao.get_code_by_code(user_id=user.id, code_type='email', code='12345')
    assert saved_code.code_used is True


def test_should_return_none_if_code_is_used(notifications_admin,
                                            notifications_admin_db,
                                            notify_db_session):
    user = create_test_user('pending')

    verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
    code = verify_codes_dao.get_codes(user_id=user.id, code_type='sms')
    verify_codes_dao.use_code(code[0].id)
    used_code = verify_codes_dao.get_codes(user_id=user.id, code_type='sms')
    assert used_code == []


def test_should_return_all_unused_code_when_there_are_many(notifications_admin,
                                                           notifications_admin_db,
                                                           notify_db_session):
    user = create_test_user('pending')
    another_user = create_another_test_user('active')
    verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
    id = verify_codes_dao.add_code(user_id=user.id, code='09876', code_type='email')
    verify_codes_dao.use_code(id)
    verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='email')
    verify_codes_dao.add_code(user_id=user.id, code='23456', code_type='email')
    verify_codes_dao.add_code(user_id=another_user.id, code='12345', code_type='sms')
    verify_codes_dao.add_code(user_id=another_user.id, code='12345', code_type='email')

    user_codes = verify_codes_dao.get_codes(user_id=user.id, code_type='email')
    assert len(user_codes) == 2
    s = sorted(user_codes, key=lambda r: r.expiry_datetime)
    assert check_hash('12345', s[0].code)
    assert check_hash('23456', s[1].code)
    assert [(code.code_used, code.code_type, code.user_id) for code in user_codes] == \
           [(False, 'email', user.id), (False, 'email', user.id)]