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.app.config['directory_base']) test_utils.clear_flask_context()
def tearDown(self): shutil.rmtree(self.app.config["directory_base"]) test_utils.clear_flask_context()