phenny-1/modules/test/test_nsfw.py

27 lines
730 B
Python

"""
test_nsfw.py - some things just aren't safe for work, the test cases
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
"""
import re
import unittest
from mock import MagicMock, Mock
from modules.nsfw import nsfw
class TestNsfw(unittest.TestCase):
def setUp(self):
self.phenny = MagicMock()
def test_nsfw(self):
input = Mock(group=lambda x: "test")
nsfw(self.phenny, input)
self.phenny.say.assert_called_once_with(
"!!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")