"""Tests for blacklist words handlers."""
import json
from unittest.mock import MagicMock

from owsresponse import response

from blacklist_manager.logic import blacklist_words


def test_export_blacklist_words_success(monkeypatch, client):
    """Test export_blacklist_words when logic returns success."""
    url = '/blacklist-words/export'
    monkeypatch.setattr(
        blacklist_words, 'export_blacklist_words',
        value=MagicMock(return_value=response.Response({'foo': 'bar'})))

    result = client.get(url)
    blacklist_words.export_blacklist_words.assert_called_once()
    assert result.status_code == 200
    response_data = json.loads(result.data.decode('utf-8'))
    assert response_data == {'foo': 'bar'}


def test_export_blacklist_words_failure(monkeypatch, client):
    """Test export_blacklist_words for failures."""
    url = '/blacklist-words/export'
    monkeypatch.setattr(
        blacklist_words, 'export_blacklist_words',
        value=MagicMock(return_value=response.create_not_found_response()))

    result = client.get(url)
    blacklist_words.export_blacklist_words.assert_called_once()
    assert result.status_code == 404