From 9210f5fb0cb30206d364ac05bd47fe280fdd6836 Mon Sep 17 00:00:00 2001 From: Arik Fraimovich Date: Tue, 29 Nov 2016 12:06:58 +0200 Subject: [PATCH] Remove unused code --- tests/handlers/__init__.py | 80 -------------------------------------- 1 file changed, 80 deletions(-) diff --git a/tests/handlers/__init__.py b/tests/handlers/__init__.py index 9375c8b4..e69de29b 100644 --- a/tests/handlers/__init__.py +++ b/tests/handlers/__init__.py @@ -1,80 +0,0 @@ -import json -from contextlib import contextmanager - -from tests.factories import user_factory -from redash.models import db -from redash.utils import json_dumps -from redash.wsgi import app - -app.config['TESTING'] = True - - -def authenticate_request(c, user): - with c.session_transaction() as sess: - if user.id is None: - db.session.flush() - sess['user_id'] = user.id - - -@contextmanager -def authenticated_user(c, user=None): - if not user: - user = user_factory.create() - db.session.commit() - authenticate_request(c, user) - - yield user - - -def json_request(method, path, data=None): - if data: - response = method(path, data=json_dumps(data)) - else: - response = method(path) - - if response.data: - response.json = json.loads(response.data) - else: - response.json = None - - return response - - -def make_request(method, path, user, data=None, is_json=True): - with app.test_client() as c: - if user: - authenticate_request(c, user) - - method_fn = getattr(c, method.lower()) - headers = {} - - if data and is_json: - data = json_dumps(data) - - if is_json: - content_type = 'application/json' - else: - content_type = None - - response = method_fn(path, data=data, headers=headers, content_type=content_type) - - if response.data and is_json: - response.json = json.loads(response.data) - - return response - - -def get_request(path, org=None): - if org: - path = "/{}{}".format(org.slug, path) - - with app.test_client() as c: - return c.get(path) - - -def post_request(path, data=None, org=None): - if org: - path = "/{}{}".format(org.slug, path) - - with app.test_client() as c: - return c.post(path, data=data)