import pytest
from bulkperformancerights.logic import user


def test_prefix_for_user():
    test_user = {'type': 'vendor', 'id': 1}
    known_good_prefix = '{}/{}/'.format(test_user.get('type'),
                                        test_user.get('id'))
    assert user.prefix_for_user(test_user) == known_good_prefix


def test_prefix_for_user_bad_id():
    with pytest.raises(NameError):
        test_user = {'type': 'vendor'}
        user.prefix_for_user(test_user)
        test_user['id'] = 'notadigit'
        user.prefix_for_user(test_user)


def test_prefix_for_user_bad_type():
    with pytest.raises(NameError):
        test_user = {'id': 1}
        user.prefix_for_user(test_user)
        test_user['type'] = 'abacadaba'
        user.prefix_for_user(test_user)


def test_valid_user_id():
    assert user.valid_user_id(1) == 1
    assert user.valid_user_id('1') == 1
    assert user.valid_user_id(None) == 0


def test_xlsx_to_json_prefix_for_user():
    test_user = {'type': 'vendor', 'id': 1}
    known_good_prefix = '{}_{}/'.format(test_user.get('type'),
                                        test_user.get('id'))
    assert user.xlsx_to_json_prefix_for_user(test_user) == known_good_prefix


def test_user_info(db_fixture):
    user_id = 1
    image_path = 'https://images.theorchard.com/vendor/1/1/1.png'
    expected_dict = {'vendor_logo_path': image_path,
                     'vendor_contact_email': 'foo@theorchard.com'}
    assert user.user_info(user_id) == expected_dict