"""Mocks for owsrequest."""

from unittest.mock import MagicMock

from owsrequest import request


def head_response(mocker, status_code=200):
    """Mock the head response."""
    head_response = MagicMock(status_code=status_code)
    mocker.patch.object(request, 'head', return_value=head_response)


def get_response(mocker, json={}, status_code=200):
    """Mock the head response."""
    r = MagicMock(status_code=status_code)
    r.json = MagicMock(return_value=json)
    return mocker.patch.object(request, 'get', return_value=r)


def post_response(mocker, json=None, status_code=200):
    """Mock the head response."""
    if json is None:
        json = {}
    response = MagicMock(status_code=status_code, json=lambda: json)
    return mocker.patch.object(request, 'post', return_value=response)


def put_response(mocker, json={}, status_code=200):
    """Mock the head response."""
    put_response = MagicMock(status_code=status_code, json=lambda: json)
    return mocker.patch.object(request, 'put', return_value=put_response)


def delete_response(mocker, status_code=200):
    """Mock the head response."""
    result = MagicMock(status_code=status_code)
    result.json = MagicMock(return_value={})
    return mocker.patch.object(request, 'delete', return_value=result)