def get(self, endpoint, **kwargs):
status_code = kwargs.pop('status_code', 200)
follow_redirects = kwargs.pop('follow_redirects', False)
if endpoint in ('index', 'browse'):
page_class = self.list_page_class
elif endpoint == 'remove':
page_class = self.confirm_page_class
elif endpoint == 'sort' and follow_redirects:
page_class = self.list_page_class
else:
page_class = self.generic_page_class
with kwargs.pop('client', None) or self.app.test_client() as client:
response = client.get(
self.url_for(endpoint, **kwargs),
follow_redirects=follow_redirects
)
if response.status_code != status_code:
raise self.page_exceptions.get(
response.status_code,
self.page_exceptions[None]
)(response.status_code)
result = page_class.from_source(response.data, response)
response.close()
test_utils.clear_flask_context()
return result
def post(self, endpoint, **kwargs):
status_code = kwargs.pop('status_code', 200)
data = kwargs.pop('data') if 'data' in kwargs else {}
with kwargs.pop('client', None) or self.app.test_client() as client:
response = client.post(self.url_for(endpoint, **kwargs),
data=data,
follow_redirects=True)
if response.status_code != status_code:
raise self.page_exceptions.get(response.status_code,
self.page_exceptions[None])(
response.status_code)
result = self.list_page_class.from_source(response.data, response)
test_utils.clear_flask_context()
return result
def post(self, endpoint, **kwargs):
status_code = kwargs.pop('status_code', 200)
data = kwargs.pop('data') if 'data' in kwargs else {}
with kwargs.pop('client', None) or self.app.test_client() as client:
response = client.post(
self.url_for(endpoint, **kwargs),
data=data,
follow_redirects=True
)
if response.status_code != status_code:
raise self.page_exceptions.get(
response.status_code,
self.page_exceptions[None]
)(response.status_code)
result = self.list_page_class.from_source(response.data, response)
test_utils.clear_flask_context()
return result
def tearDown(self):
self.app.config['plugin_namespaces'] = self.original_namespaces
self.manager.clear()
test_utils.clear_flask_context()
def tearDown(self):
shutil.rmtree(self.workbench)
test_utils.clear_flask_context()
def tearDown(self):
shutil.rmtree(self.base)
test_utils.clear_flask_context()
def tearDown(self):
shutil.rmtree(self.workbench)
test_utils.clear_flask_context()
def tearDown(self):
shutil.rmtree(self.app.config['directory_base'])
test_utils.clear_flask_context()
def tearDown(self):
shutil.rmtree(self.app.config["directory_base"])
test_utils.clear_flask_context()
def tearDown(self):
shutil.rmtree(self.base)
test_utils.clear_flask_context()
def tearDown(self):
self.app.config['plugin_namespaces'] = self.original_namespaces
self.manager.clear()
test_utils.clear_flask_context()
