Return comment_id when comment created

Author: matsumatsu20

src/handlers/me/articles/comments/create/me_articles_comments_create.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+import json
 import logging
 import os
 import traceback
@@ -66,7 +67,10 @@ def exec_main_proc(self):
             logging.fatal(err)
             traceback.print_exc()
         finally:
-            return {'statusCode': 200}
+            return {
+                'statusCode': 200,
+                'body': json.dumps({'comment_id': comment_id})
+            }
 
     def __create_comment_notification(self, article_info, comment_id, user_id):
         notification_table = self.dynamodb.Table(os.environ['NOTIFICATION_TABLE_NAME'])

tests/handlers/me/articles/comments/create/test_me_articles_comments_create.py

@@ -101,6 +101,7 @@ def test_main_ok(self):
         unread_manager = self.unread_notification_manager_table.get_item(Key={'user_id': 'article_user01'}).get('Item')
 
         self.assertEqual(response['statusCode'], 200)
+        self.assertEqual(json.loads(response['body'])['comment_id'], 'HOGEHOGEHOGE')
         self.assertEqual(len(comment_after) - len(comment_before), 1)
         self.assertEqual(len(notification_after) - len(notification_before), 1)
         self.assertEqual(len(unread_notification_manager_after) - len(unread_notification_manager_before), 1)