coverage improvement on Reply.save_image()

thraxil · Feb 21, 2016 · 1416560 · 1416560
1 parent 61875e9
commit 1416560
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/spokehub/main/tests/test_models.py b/spokehub/main/tests/test_models.py
@@ -99,6 +99,10 @@ def test_add_reply_vimeo(self):
         self.assertEqual(r.vimeo_id, "foo")
 
 
+class DummyFile(object):
+    pass
+
+
 class ReplyTest(TestCase):
     def test_unicode(self):
         r = ReplyFactory()
@@ -131,6 +135,12 @@ def test_link_usernames(self):
                 u.username, u.username, u2.username, u2.username)
         )
 
+    def test_save_image_invalid_ext(self):
+        r = ReplyFactory()
+        d = DummyFile()
+        d.name = "invalid.extension"
+        self.assertIsNone(r.save_image(d))
+
 
 class NowPostTest(TestCase):
     def test_unicode(self):