def test_get(self): pc = PostController() post_id = pc.create(1, 'hello world') post = pc.get(1, post_id) self.assertIsInstance(post, ttypes.Post) self.assertEqual(post_id, post.id) post_id = pc.create(1, 'hello world') post = pc.get(1, post_id) self.assertIsInstance(post, ttypes.Post) self.assertEqual(post_id, post.id) # self.assertTrue(pc.get()) # def test_foo(self): # self.fail()
def post(self): data = Post.parser.parse_args() auth_header = request.headers.get('Authorization') if auth_header: access_token = auth_header.split(" ")[1] else: return { "message": "This method requires an authorization header." }, 400 error, client_id = auth_by_token(access_token) if error: return {"message": error}, 401 error_message = PostController.create_post(data['theme'], data['anonymity'], client_id, data['content']) if error_message: return {"message": error_message}, 400 return {"message": "Success!"}, 201
def get(self): return PostController.get(self)
def post(self): return PostController.post(self)
def test_create(self): pc = PostController() self.assertRaises(TypeError, pc.create, 'one', 'foo') self.assertEqual(1, pc.create(1, 'hello world')) self.assertEqual(2, pc.create(1, 'sample post'))