Atheneum/server/tests/service/test_patch_service.py

from datetime import datetime, timedelta

import pytest
from mock import MagicMock, patch

from atheneum import errors
from atheneum.model import UserToken, User
from atheneum.service import patch_service, role_service

service_module = 'atheneum.service.patch_service'


@patch(service_module + '.db.session.commit')
def test_patch_models(
        mock_db_session_commit: MagicMock):
    request_user = User()
    request_user.role = role_service.Role.ADMIN

    user = User()
    user.name = 'TestUser'
    user.version = 1
    user.last_login_time = datetime.now() - timedelta(days=1)

    user_patch = User()
    user_patch.name = 'TestUser'
    user_patch.version = 1
    user_patch.last_login_time = datetime.now()

    patched_user = patch_service.patch(request_user, user, user_patch)
    assert patched_user.version > 1
    assert patched_user.last_login_time == user_patch.last_login_time
    mock_db_session_commit.assert_called_once()


def test_patch_of_different_types():
    request_user = User()
    request_user.role = role_service.Role.ADMIN

    user = User()
    user_token = UserToken()

    with pytest.raises(errors.ValidationError) as error_info:
        patch_service.patch(request_user, user, user_token)


def test_patch_different_ids():
    request_user = User()
    request_user.role = role_service.Role.ADMIN

    user1 = User()
    user1.id = 1

    user2 = User()
    user2.id = 2

    with pytest.raises(errors.ValidationError) as error_info:
        patch_service.patch(request_user, user1, user2)


def test_patch_different_versions():
    request_user = User()
    request_user.role = role_service.Role.ADMIN

    user1 = User()
    user1.version = 1

    user2 = User()
    user2.version = 2

    with pytest.raises(errors.ValidationError) as error_info:
        patch_service.patch(request_user, user1, user2)


def test_patch_restricted_attributes():
    request_user = User()
    request_user.role = role_service.Role.USER

    user1 = User()
    user1.version = 1
    user1.name = 'Bob'

    user2 = User()
    user2.version = 1
    user2.name = 'Chris'

    with pytest.raises(errors.ValidationError) as error_info:
        patch_service.patch(request_user, user1, user2)