Beispiel #1
0
class MTSExtensionTestCase(TestCase):

    def setUp(self):
        self.mts = MTSExtension(mock_bot())
        self.message = Mock(message='foo', channel='bar', from_nick='me')

    def test_record_last(self):
        self.mts.record_last(self.message)

        assert self.mts.last['bar']['me'] == 'foo'

    def test_clear_last(self):
        self.mts.last['bar']['me'] = 'foo'
        self.mts.clear_last(self.message)

        assert self.mts.last['bar']['me'] == ''

    def test_get_last(self):
        self.mts.last['bar']['me'] = 'foo'

        assert self.mts.get_last(self.message) == 'foo'

    def test_contextualize_replaces_text(self):
        self.mts.record_last(self.message)
        next = Mock(message='s/foo/baz/', channel='bar', from_nick='me')
        self.mts.contextualize(next)

        assert 'meant to say: baz' in next.response

    def test_contextualize_replaces_text_without_trailing_slash(self):
        self.mts.record_last(self.message)
        next = Mock(message='s/foo/baz', channel='bar', from_nick='me')
        self.mts.contextualize(next)

        assert 'meant to say: baz' in next.response

    def test_contextualize_replaces_text_in_msg(self):
        self.mts.record_last(self.message)
        next = Mock(message='i mean s/foo/baz is what i meant', channel='bar', from_nick='me')
        self.mts.contextualize(next)

        assert 'meant to say: baz' in next.response

    def test_contextualize_ignores_if_no_last(self):
        next = Mock(message='s/foo/baz/', channel='bar', from_nick='me', response=None)
        self.mts.contextualize(next)

        assert next.response is None
Beispiel #2
0
 def setUp(self):
     self.mts = MTSExtension(mock_bot())
     self.message = Mock(message='foo', channel='bar', from_nick='me')