import io

import pytest

import falcon
from falcon import testing

unicode_message = 'Unicode: \x80'


@pytest.fixture
def client():
    app = falcon.App()

    tehlogger = LoggerResource()
    app.add_route('/logger', tehlogger)
    return testing.TestClient(app)


class LoggerResource:
    def on_get(self, req, resp):
        req.log_error(unicode_message)

    def on_head(self, req, resp):
        req.log_error(unicode_message.encode('utf-8'))


class TestWSGIError:
    def setup_method(self, method):
        self.wsgierrors_buffer = io.BytesIO()

        # Simulate Gunicorn's behavior under Python 3
        self.wsgierrors = io.TextIOWrapper(
            self.wsgierrors_buffer, line_buffering=True, encoding='utf-8'
        )

    def test_responder_logged_bytestring(self, client):
        client.simulate_request(
            path='/logger', wsgierrors=self.wsgierrors, query_string='amount=10'
        )

        log = self.wsgierrors_buffer.getvalue()

        assert unicode_message.encode('utf-8') in log
        assert b'?amount=10' in log