Simplify test by removing ThreadPool

This commit is contained in:
richard-dds 2018-06-14 13:10:19 -04:00
parent e89be59d3e
commit 606bd61d3a

View File

@ -1,7 +1,7 @@
import re import re
import pytest import pytest
import tornado.web import tornado.web
from concurrent.futures import ThreadPoolExecutor import tornado.gen
@pytest.mark.gen_test @pytest.mark.gen_test
@ -17,19 +17,22 @@ def test_redirects_when_not_logged_in(http_client, base_url):
@pytest.mark.gen_test @pytest.mark.gen_test
def test_login_with_valid_bearer_token(app, monkeypatch, http_client, base_url): def test_login_with_valid_bearer_token(app, monkeypatch, http_client, base_url):
with ThreadPoolExecutor(max_workers=1) as executor: @tornado.gen.coroutine
monkeypatch.setattr( def _validate_login_token(c, t):
"atst.handlers.login.Login._validate_login_token", return True
lambda c,t: executor.submit(lambda: True),
) monkeypatch.setattr(
response = yield http_client.fetch( "atst.handlers.login.Login._validate_login_token",
base_url + "/login?bearer-token=abc-123", _validate_login_token
follow_redirects=False, )
raise_error=False response = yield http_client.fetch(
) base_url + "/login?bearer-token=abc-123",
assert response.headers["Set-Cookie"].startswith("atst") follow_redirects=False,
assert response.headers['Location'] == '/home' raise_error=False
assert response.code == 302 )
assert response.headers["Set-Cookie"].startswith("atst")
assert response.headers['Location'] == '/home'
assert response.code == 302
@pytest.mark.gen_test @pytest.mark.gen_test