"""Test for update_vend_contact_user."""

import json
from unittest.mock import MagicMock

from owsresponse import response

from users.app import app
from users.logic import user_info


def test_update_vend_contact_user(monkeypatch):
    """Test update_vend_contact_user success without contact data."""
    expected = {'foo': 'bar'}
    app.testing = True
    monkeypatch.setattr(
        user_info,
        'update_vend_contact_user',
        MagicMock(return_value=response.Response(message=expected)),
    )

    data = {'login': 'new_name'}
    request = app.test_client().patch(
        '/users/alw/1234', data=json.dumps(data), content_type='application/json'
    )
    response_data = json.loads(request.data.decode('utf-8'))
    assert request.status_code == 200
    assert response_data == expected
    user_info.update_vend_contact_user.assert_called_with('1234', data)


def test_update_with_contact_data(monkeypatch):
    """Test update with contact data success."""
    expected = {'foo': 'bar'}
    app.testing = True
    monkeypatch.setattr(
        user_info,
        'update_vend_contact_user',
        MagicMock(return_value=response.Response(message=expected)),
    )

    data = {'login': 'new_name', 'contact': {'first_name': 'new_name'}}
    request = app.test_client().patch(
        '/users/alw/1234', data=json.dumps(data), content_type='application/json'
    )
    response_data = json.loads(request.data.decode('utf-8'))
    assert request.status_code == 200
    assert response_data == expected
    user_info.update_vend_contact_user.assert_called_with('1234', data)