def test_main_ok_like_2(self):
params = {'pathParameters': {'article_id': 'testidlike02'}}
article_liked_user = ArticlesLikesShow(event=params,
context={},
dynamodb=self.dynamodb)
response = article_liked_user.main()
self.assertEqual(response['statusCode'], 200)
self.assertEqual(json.loads(response['body'])['count'], 2)
def test_call_validate_article_existence(self):
params = {'pathParameters': {'article_id': 'testidlike02'}}
mock_lib = MagicMock()
with patch('articles_likes_show.DBUtil', mock_lib):
ArticlesLikesShow(event=params, context={},
dynamodb=self.dynamodb).main()
args, kwargs = mock_lib.validate_article_existence.call_args
self.assertTrue(mock_lib.validate_article_existence.called)
self.assertTrue(args[0])
self.assertTrue(args[1])
self.assertEqual(kwargs['status'], 'public')
def lambda_handler(event, context):
articles_likes_get = ArticlesLikesShow(event=event,
context=context,
dynamodb=dynamodb)
return articles_likes_get.main()
def assert_bad_request(self, params):
test_function = ArticlesLikesShow(params, {}, self.dynamodb)
response = test_function.main()
self.assertEqual(response['statusCode'], 400)