improve tests and test coverage

2013-08-25 14:32:42 -07:00
parent f5fb2fc37d
commit 23d99dd326
6 changed files with 55 additions and 20 deletions
--- a/modules/test/test_nsfw.py
+++ b/modules/test/test_nsfw.py
@@ -16,6 +16,11 @@ class TestNsfw(unittest.TestCase):
    def test_nsfw(self):
        input = Mock(group=lambda x: "test")
        nsfw(self.phenny, input)
-
        self.phenny.say.assert_called_once_with(
-                "!!NSFW!! -> test <- !!NSFW!!")
+            "!!NSFW!! -> test <- !!NSFW!!")
+
+    def test_nsfw_none(self):
+        input = Mock(group=lambda x: None)
+        nsfw(self.phenny, input)
+        self.phenny.say.assert_called_once_with(
+            ".nsfw <link> - for when a link isn't safe for work")