def get_object(self):
obj_id = self.kwargs.get('id', None)
return Article.get_by_id_or_404(obj_id)
def get_object(self):
obj_id = self.kwargs.get('id', None)
return Article.get_by_id_or_404(obj_id)
def test_get_by_id_or_404_no_article(self):
key = Article(title='title', body='body').put()
free_key = key.id() + 1
with self.assertRaises(Http404):
Article.get_by_id_or_404(free_key)
def test_get_by_id_or_404_article_found(self):
key = Article(title='title', body='body').put()
result = Article.get_by_id_or_404(key.id())
self.assertEqual(key, result.key())
def test_get_by_id_or_404_id_is_none(self):
with self.assertRaises(Http404):
Article.get_by_id_or_404(None)
