Limnoria/plugins/QuoteGrabs/test.py

187 lines
8.4 KiB
Python

###
# Copyright (c) 2004, Daniel DiPaolo
# Copyright (c) 2008, James McCoy
# Copyright (c) 2010-2021, Valentin Lorentz
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
from supybot.test import *
class QuoteGrabsTestCase(ChannelPluginTestCase):
plugins = ('QuoteGrabs',)
def testQuoteGrab(self):
testPrefix = 'foo!bar@baz'
self.assertError('grab foo')
# Test join/part/notice (shouldn't grab)
self.irc.feedMsg(ircmsgs.join(self.channel, prefix=testPrefix))
self.assertError('grab foo')
self.irc.feedMsg(ircmsgs.part(self.channel, prefix=testPrefix))
self.assertError('grab foo')
# Test privmsgs
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'something',
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertResponse('quote foo', '<foo> something')
# Test actions
self.irc.feedMsg(ircmsgs.action(self.channel, 'moos',
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertResponse('quote foo', '* foo moos')
def testQuoteGrabReplyDisabled(self):
testPrefix = 'foo!bar@baz'
prefixChar = conf.supybot.reply.whenAddressedBy.chars()[0]
self.irc.feedMsg(ircmsgs.IrcMsg(
server_tags={'msgid': 'aaaa'}, prefix=testPrefix,
command='PRIVMSG', args=(self.channel, 'something')))
self.irc.feedMsg(ircmsgs.IrcMsg(
server_tags={'msgid': 'bbbb'}, prefix=testPrefix,
command='PRIVMSG', args=(self.channel, 'something else')))
# supybot.protocols.irc.experimentalExtensions is not enabled, so
# +draft/reply is ignored.
self.irc.feedMsg(ircmsgs.IrcMsg(
server_tags={'+draft/reply': 'aaaa'}, prefix=self.prefix,
command='PRIVMSG', args=(self.channel, prefixChar+'grab foo')))
self.assertResponse(' ', 'The operation succeeded.')
self.assertResponse('quote foo', '<foo> something else')
def testQuoteGrabReply(self):
testPrefix = 'foo!bar@baz'
prefixChar = conf.supybot.reply.whenAddressedBy.chars()[0]
self.irc.feedMsg(ircmsgs.IrcMsg(
server_tags={'msgid': 'aaaa'}, prefix=testPrefix,
command='PRIVMSG', args=(self.channel, 'something')))
self.irc.feedMsg(ircmsgs.IrcMsg(
server_tags={'msgid': 'bbbb'}, prefix=testPrefix,
command='PRIVMSG', args=(self.channel, 'something else')))
with conf.supybot.protocols.irc.experimentalExtensions.context(True):
self.irc.feedMsg(ircmsgs.IrcMsg(
server_tags={'+draft/reply': 'aaaa'}, prefix=self.prefix,
command='PRIVMSG', args=(self.channel, prefixChar+'grab foo')))
self.assertResponse(' ', 'The operation succeeded.')
self.assertResponse('quote foo', '<foo> something')
def testUngrab(self):
testPrefix = 'foo!bar@baz'
# nothing yet
self.assertError('ungrab')
self.assertError('ungrab 2')
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'something',
prefix=testPrefix))
# still not grabbed
self.assertError('ungrab')
self.assertError('ungrab 3')
# grab and ungrab a quote
self.assertNotError('grab foo')
self.assertNotError('ungrab')
self.assertNotError('grab foo')
# this is not there...
self.assertError('ungrab 8883')
# ...unlike this...
self.assertNotError('ungrab 1')
# ...but not now anymore :-D
self.assertError('ungrab')
# grab two quotes and ungrab them by id
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'something',
prefix=testPrefix))
self.assertNotError('grab foo')
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'another',
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertNotError('ungrab 1')
self.assertNotError('ungrab 2')
self.assertError('ungrab')
def testList(self):
testPrefix = 'foo!bar@baz'
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'testList',
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertResponse('quotegrabs list foo', '#1: testList')
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'a' * 80,
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertResponse('quotegrabs list foo',
'#2: %s... and #1: testList' %\
('a'*43)) # 50 - length of "#2: ..."
def testDuplicateGrabs(self):
testPrefix = 'foo!bar@baz'
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'testDupe',
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertNotError('grab foo') # note:NOTanerror,stillwon'tdupe
self.assertResponse('quotegrabs list foo', '#1: testDupe')
def testCaseInsensitivity(self):
testPrefix = 'foo!bar@baz'
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'testCI',
prefix=testPrefix))
self.assertNotError('grab FOO')
self.assertNotError('quote foo')
self.assertNotError('quote FoO')
self.assertNotError('quote Foo')
self.assertNotError('quotegrabs list FOO')
self.assertNotError('quotegrabs list fOo')
def testRandom(self):
testPrefix = 'foo!bar@baz'
self.assertError('random')
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'testRandom',
prefix=testPrefix))
self.assertError('random') # still none in the db
self.assertNotError('grab foo')
self.assertResponse('random', '<foo> testRandom')
self.assertResponse('random foo', '<foo> testRandom')
self.assertResponse('random FOO', '<foo> testRandom')
def testGet(self):
testPrefix= 'foo!bar@baz'
self.assertError('quotegrabs get asdf')
self.assertError('quotegrabs get 1')
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'testGet',
prefix=testPrefix))
self.assertNotError('grab foo')
self.assertNotError('quotegrabs get 1')
def testSearch(self):
testPrefix= 'foo!bar@baz'
self.irc.feedMsg(ircmsgs.privmsg(self.channel, 'testSearch',
prefix=testPrefix))
self.assertError('quotegrabs search test') # still none in db
self.assertNotError('grab foo')
self.assertNotError('quotegrabs search test')
class QuoteGrabsNonChannelTestCase(QuoteGrabsTestCase):
config = { 'databases.plugins.channelSpecific' : False }
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: