Some improvements to images test case coverage (this needs to be revisited and updated to webtest)

Jon Wayne Parrott · Jon Wayne Parrott · commit c48772b51527 · 2015-09-14T10:46:14.000-07:00
diff --git a/appengine/images/tests/test_guestbook.py b/appengine/images/tests/test_guestbook.py
@@ -28,6 +28,12 @@ def setUp(self):
         reload(main)
 
     def test_get(self):
+        main.Greeting(
+            parent=main.guestbook_key('default_guestbook'),
+            author='123',
+            content='abc'
+        ).put()
+
         # Build a request object passing the URI path to be tested.
         # You can also pass headers, query arguments etc.
         request = webapp2.Request.blank('/')
@@ -43,7 +49,7 @@ def test_post(self, mock_images):
         request = webapp2.Request.blank(
             '/sign',
             POST={'content': 'asdf'},
-            )
+        )
         response = request.get_response(main.app)
         mock_images.resize.assert_called_once_with(mock.ANY, 32, 32)
 
@@ -53,10 +59,11 @@ def test_post(self, mock_images):
     def test_img(self):
         greeting = main.Greeting(
             parent=main.guestbook_key('default_guestbook'),
-            id=123,
+            id=123
         )
         greeting.author = 'asdf'
         greeting.content = 'asdf'
+        greeting.avatar = b'123'
         greeting.put()
 
         request = webapp2.Request.blank(
@@ -79,7 +86,7 @@ def test_post_and_get(self, mock_images):
         request = webapp2.Request.blank(
             '/sign',
             POST={'content': 'asdf'},
-            )
+        )
         response = request.get_response(main.app)
 
         request = webapp2.Request.blank('/')
