"""Tests for handlers."""

from unittest.mock import MagicMock

from owsresponse import response

from blacklist_manager.logic import blacklist_words


def test_delete_blacklist_word_success(
        monkeypatch, client, valid_oa_headers):
    """Test DELETE /blacklist-word/{blacklist_id} handler success state."""
    monkeypatch.setattr(
        blacklist_words, 'delete_blacklist_word',
        value=MagicMock(return_value=response.Response(status=200)))

    result = client.delete('/blacklist-word/10', headers=valid_oa_headers)

    assert result.status_code == 200
    assert blacklist_words.delete_blacklist_word.called


def test_delete_blacklist_word_by_id_not_found_error(
        monkeypatch, client, valid_oa_headers):
    """Test DELETE /blacklist-word/{blacklist_id} handler failure state."""
    monkeypatch.setattr(
        blacklist_words, 'delete_blacklist_word',
        value=MagicMock(return_value=response.create_not_found_response(
            message='Blacklist Id not found.')))

    result = client.delete('/blacklist-word/10', headers=valid_oa_headers)

    assert result.status_code == 404
    assert blacklist_words.delete_blacklist_word.called