import pytest

import falcon
from falcon import testing


def sink(req, resp, **kw):
    resp.text = 'sink'


async def sink_async(req, resp, **kw):
    resp.text = 'sink'


@pytest.fixture
def client(asgi, util, tmp_path):
    file = tmp_path / 'file.txt'
    file.write_text('foo bar')

    def make(sink_before_static_route):
        app = util.create_app(asgi, sink_before_static_route=sink_before_static_route)
        app.add_sink(sink_async if asgi else sink, '/sink')
        app.add_static_route('/sink/static', str(tmp_path))

        return testing.TestClient(app)

    return make


def test_sink_before_static_route(client):
    cl = client(True)
    res = cl.simulate_get('/sink/foo')
    assert res.text == 'sink'
    res = cl.simulate_get('/sink/static/file.txt')
    assert res.text == 'sink'
    res = cl.simulate_get('/sink/static/')
    assert res.text == 'sink'


def test_sink_after_static_route(client):
    cl = client(False)
    res = cl.simulate_get('/sink/foo')
    assert res.text == 'sink'
    res = cl.simulate_get('/sink/static/file.txt')
    assert res.text == 'foo bar'
    res = cl.simulate_get('/sink/static/')
    assert res.status == falcon.HTTP_404