#!/usr/bin/env python ### # Copyright (c) 2003, Daniel DiPaolo # 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 testsupport import * try: import sqlite except ImportError: sqlite = None if sqlite is not None: MoobotFactoids = Owner.loadPluginModule('MoobotFactoids') MF = MoobotFactoids class OptionListTestCase(unittest.TestCase): def testPickOptions(self): for i in xrange(10): self.failUnless(MF.pickOptions('(a|b)') in ['a', 'b']) self.failUnless(MF.pickOptions('a') == 'a') self.failUnless(MF.pickOptions('(a|b (c|d))') in ['a', 'b c', 'b d']) self.failUnless(MF.pickOptions('(a|(b|)c)') in ['a', 'bc', 'c']) self.failUnless(MF.pickOptions('(a(b|)|(c|)d)') in ['a', 'ab', 'cd', 'd']) self.failUnless(MF.pickOptions('(a|)') in ['a', '']) self.failUnless(MF.pickOptions('(|a)') in ['a', '']) self.failUnless(MF.pickOptions('((a)|(b))') in ['(a)', '(b)']) class FactoidsTestCase(PluginTestCase, PluginDocumentation): plugins = ('MoobotFactoids', 'User', 'Utilities') def setUp(self): PluginTestCase.setUp(self) # Create a valid user to use self.prefix = 'foo!bar@baz' self.assertNotError('register tester moo') def testAddFactoid(self): self.assertNotError('moo is foo') # Check stripping punctuation self.assertError('moo!? is foo') # 'moo' already exists self.assertNotError('foo!? is foo') self.assertResponse('foo', 'foo is foo') self.assertNotError('bar is moo is moo') self.assertResponse('bar', 'moo is moo') # Check substitution self.assertNotError('who is $who') self.assertResponse('who', 'foo') def testLiteral(self): self.assertError('literal moo') # no factoids yet self.assertNotError('moo is foo') self.assertResponse('literal moo', 'foo') self.assertNotError('moo2 is moo!') self.assertResponse('literal moo2', 'moo!') self.assertNotError('moo3 is foo') self.assertResponse('literal moo3', 'foo') def testGetFactoid(self): self.assertNotError('moo is foo') self.assertResponse('moo', 'foo') self.assertNotError('moo2 is moo!') self.assertResponse('moo2', 'moo2 is moo!') self.assertNotError('moo3 is foo') self.assertAction('moo3', 'foo') # Test and make sure it's parsing self.assertNotError('moo4 is (1|2|3)') self.assertRegexp('moo4', '^(1|2|3)$') # Check case-insensitivity self.assertResponse('MOO', 'foo') self.assertResponse('mOo', 'foo') self.assertResponse('MoO', 'foo') # Check the "_is_" ability self.assertNotError('delete moo') self.assertNotError('moo _is_ foo') self.assertResponse('moo', 'foo') self.assertNotError('foo is bar _is_ baz') self.assertResponse('foo is bar', 'foo is bar is baz') def testFactinfo(self): self.assertNotError('moo is foo') self.assertRegexp('factinfo moo', '^moo: Created by tester on.*$') self.assertNotError('moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Last requested by foo!bar@baz on .*?, ' 'requested 1 time.$') self.assertNotError('moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Last requested by foo!bar@baz on .*?, ' 'requested 2 times.$') self.assertNotError('moo =~ s/foo/bar/') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Last modified by tester on .*?\. ' 'Last requested by foo!bar@baz on .*?, ' 'requested 2 times.$') self.assertNotError('lock moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Last modified by tester on .*?\. ' 'Last requested by foo!bar@baz on .*?, ' 'requested 2 times. ' 'Locked by tester on .*\.$') self.assertNotError('unlock moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Last modified by tester on .*?\. ' 'Last requested by foo!bar@baz on .*?, ' 'requested 2 times.$') # Make sure I solved this bug # Check and make sure all the other stuff is reset self.assertNotError('foo is bar') self.assertNotError('foo =~ s/bar/blah/') self.assertNotError('foo') self.assertNotError('no foo is baz') self.assertRegexp('factinfo foo', '^foo: Created by tester on' '(?!(request|modif)).*?\.$') def testLockUnlock(self): self.assertNotError('moo is moo') self.assertNotError('lock moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Locked by tester on .*?\.') # switch user self.prefix = 'moo!moo@moo' self.assertNotError('register nottester moo') self.assertError('unlock moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on' '.*?\. Locked by tester on .*?\.') # switch back self.prefix = 'foo!bar@baz' self.assertNotError('identify tester moo') self.assertNotError('unlock moo') self.assertRegexp('factinfo moo', '^moo: Created by tester on.*?\.') def testChangeFactoid(self): self.assertNotError('moo is moo') self.assertNotError('moo =~ s/moo/moos/') self.assertResponse('moo', 'moos') self.assertNotError('moo =~ s/reply/action/') self.assertAction('moo', 'moos') self.assertNotError('moo =~ s/moos/(moos|woofs)/') self.assertActionRegexp('moo', '^(moos|woofs)$') self.assertError('moo =~ s/moo/') def testMost(self): userPrefix1 = 'moo!bar@baz'; userNick1 = 'moo' userPrefix2 = 'boo!bar@baz'; userNick2 = 'boo' self.irc.feedMsg(ircmsgs.privmsg(self.irc.nick, 'register %s bar' % userNick1, prefix=userPrefix1)) self.irc.feedMsg(ircmsgs.privmsg(self.irc.nick, 'register %s bar' % userNick2, prefix=userPrefix2)) _ = self.irc.takeMsg() _ = self.irc.takeMsg() # Check an empty database self.assertError('most popular') self.assertError('most authored') self.assertError('most recent') # Check singularity response self.prefix = userPrefix1 self.assertNotError('moogle is moo') self.assertError('most popular') self.assertResponse('most authored', 'Most prolific author: moo (1)') self.assertResponse('most recent', "1 latest factoid: 'moogle'") self.assertResponse('moogle', 'moo') self.assertResponse('most popular', "Top 1 requested factoid: 'moogle' (1)") # Check plural response self.prefix = userPrefix2 self.assertNotError('mogle is mo') self.assertResponse('most authored', 'Most prolific authors: moo (1) and boo (1)') self.assertResponse('most recent', "2 latest factoids: 'mogle' and 'moogle'") self.assertResponse('moogle', 'moo') self.assertResponse('most popular', "Top 1 requested factoid: 'moogle' (2)") self.assertResponse('mogle', 'mo') self.assertResponse('most popular', "Top 2 requested factoids: " "'moogle' (2) and 'mogle' (1)") # Check most author ordering self.assertNotError('moo is oom') self.assertResponse('most authored', 'Most prolific authors: boo (2) and moo (1)') def testListkeys(self): self.assertResponse('listkeys %', 'No keys matching \'%\' found.') self.assertNotError('moo is moo') # If only one key, it should respond with the factoid self.assertResponse('listkeys moo', 'moo') self.assertResponse('listkeys foo', 'No keys matching \'foo\' ' 'found.') # Throw in a bunch more for i in range(10): self.assertNotError('moo%s is moo' % i) self.assertRegexp('listkeys moo', '^Key search for \'moo\' ' '\(11 found\): (\'moo\d*\', )+and \'moo9\'$') self.assertNotError('foo is bar') self.assertRegexp('listkeys %', '^Key search for \'\%\' ' '\(12 found\): \'foo\', (\'moo\d*\', )+and ' '\'moo9\'$') # Check quoting self.assertNotError('foo\' is bar') self.assertResponse('listkeys foo', 'Key search for \'foo\' ' '(2 found): \'foo\' and "foo\'"') def testListvalues(self): self.assertNotError('moo is moo') self.assertResponse('listvalues moo', 'Value search for \'moo\' (1 found): \'moo\'') def testListauth(self): self.assertNotError('moo is moo') self.assertRegexp('listauth tester', 'tester.*\(1 found\):.*moo') self.assertError('listauth moo') def testDelete(self): self.assertNotError('moo is moo') self.assertNotError('lock moo') self.assertError('delete moo') self.assertNotError('unlock moo') self.assertNotError('delete moo') def testAugmentFactoid(self): self.assertNotError('moo is foo') self.assertNotError('moo is also bar') self.assertResponse('moo', 'moo is foo, or bar') self.assertNotError('moo is bar _is_ foo') self.assertNotError('moo is bar is also foo') self.assertResponse('moo is bar', 'moo is bar is foo, or foo') def testReplaceFactoid(self): self.assertNotError('moo is foo') self.assertNotError('no moo is bar') self.assertResponse('moo', 'moo is bar') self.assertNotError('no, moo is baz') self.assertResponse('moo', 'moo is baz') self.assertNotError('lock moo') self.assertError('no moo is qux') self.assertNotError('foo is bar _is_ foo') self.assertNotError('no foo is bar _is_ baz') self.assertResponse('foo is bar', 'foo is bar is baz') def testRegexpNotCalledIfAlreadyHandled(self): self.assertResponse('echo foo is bar', 'foo is bar') self.assertNoResponse(' ', 3) def testNoResponseToCtcp(self): self.assertNotError('foo is bar') self.assertResponse('foo', 'foo is bar') self.irc.feedMsg(ircmsgs.privmsg(self.irc.nick, '\x01VERSION\x01')) m = self.irc.takeMsg() self.failIf(m) # vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: