def setUp(self): UserTestCase.setUp(self) self.app = app.test_client() # Patch the Flask/Werkzeug open to support required features orig_open = self.app.open def myopen(*args, **kwargs): headers = kwargs.get("headers", []) # Support basic authentication if kwargs.get("auth", True): userpass = self.username + ":" + self.password uphash = base64.b64encode(userpass) headers.append(("Authorization", "Basic %s" % uphash)) kwargs.update(headers=headers) if "auth" in kwargs: del kwargs["auth"] r = orig_open(*args, **kwargs) # If the response is json, parse it r.json = None if r.content_type == "application/json": r.json = json.loads(r.data) return r self.app.open = myopen self.get = self.app.get self.post = self.app.post self.delete = self.app.delete self.put = self.app.put
def setUp(self): logging.basicConfig(level=logging.ERROR) app.config["TESTING"] = True self.client = app.test_client() self.get = self.client.get self.post = self.client.post self.put = self.client.put self.delete = self.client.delete self.get_context = self.get_with_context self.post_context = self.post_with_context self.put_context = self.put_with_context self.delete_context = self.delete_with_context