def test_default_unauthorized_callback(self): with self.app.test_request_context(): m = JWTManager(self.app) result = m._unauthorized_callback("Missing Authorization Header") status_code, data = self._parse_callback_result(result) self.assertEqual(status_code, 401) self.assertEqual(data, {'msg': 'Missing Authorization Header'})
def test_custom_unauthorized_callback(self): with self.app.test_request_context(): m = JWTManager(self.app) @m.unauthorized_loader def custom_unauthorized(err_str): return jsonify({"err": err_str}), 200 result = m._unauthorized_callback("GOTTA LOGIN FOOL") status_code, data = self._parse_callback_result(result) self.assertEqual(status_code, 200) self.assertEqual(data, {'err': 'GOTTA LOGIN FOOL'})