"""Unit tests for usier logic layer.""" import random import pytest from analytics.consts.user import AccountTypes from analytics.logic import user @pytest.fixture(params=['vendor', 'subaccount']) def test_user(request): """Testing fixture for accaount types.""" def random_id(): return int(1000 + 4000 * random.random()) return user.create(user_id=random_id(), user_type=request.param) def test_create_user(test_user): """Test the creation of a user.""" assert test_user assert test_user.message def test_user_account_type(test_user): """Test the account type of a user.""" response = test_user.message assert test_user.status == 200 assert user.is_user_vendor(response) != user.is_user_subaccount(response) if response.user_type == AccountTypes.SUBACCOUNT: assert user.is_user_subaccount(response) else: assert user.is_user_vendor(response)