add rank test to factoids, update factoids test code to sqlite3

This commit is contained in:
nanotube 2010-03-23 13:54:31 -04:00 committed by Valentin Lorentz
parent 5c0596f484
commit b922890b6f

View File

@ -31,136 +31,142 @@
from supybot.test import * from supybot.test import *
try: try:
import sqlite import sqlite3
except ImportError: except ImportError:
sqlite = None from pysqlite2 import dbapi2 as sqlite3 # for python2.4
if sqlite: class FactoidsTestCase(ChannelPluginTestCase):
class FactoidsTestCase(ChannelPluginTestCase): plugins = ('Factoids',)
plugins = ('Factoids',) def testRandomfactoid(self):
def testRandomfactoid(self): self.assertError('random')
self.assertError('random') self.assertNotError('learn jemfinch as my primary author')
self.assertNotError('learn jemfinch as my primary author') self.assertRegexp('random', 'primary author')
self.assertRegexp('random', 'primary author')
def testLearn(self): def testLearn(self):
self.assertError('learn as my primary author') self.assertError('learn as my primary author')
self.assertError('learn jemfinch as') self.assertError('learn jemfinch as')
self.assertNotError('learn jemfinch as my primary author') self.assertNotError('learn jemfinch as my primary author')
self.assertNotError('info jemfinch') self.assertNotError('info jemfinch')
self.assertRegexp('whatis jemfinch', 'my primary author') self.assertRegexp('whatis jemfinch', 'my primary author')
self.assertRegexp('whatis JEMFINCH', 'my primary author') self.assertRegexp('whatis JEMFINCH', 'my primary author')
self.assertRegexp('whatis JEMFINCH 1', 'my primary author') self.assertRegexp('whatis JEMFINCH 1', 'my primary author')
self.assertNotError('learn jemfinch as a bad assembly programmer') self.assertNotError('learn jemfinch as a bad assembly programmer')
self.assertRegexp('whatis jemfinch 2', 'bad assembly') self.assertRegexp('whatis jemfinch 2', 'bad assembly')
self.assertNotRegexp('whatis jemfinch 2', 'primary author') self.assertNotRegexp('whatis jemfinch 2', 'primary author')
self.assertRegexp('whatis jemfinch', r'.*primary author.*assembly') self.assertRegexp('whatis jemfinch', r'.*primary author.*assembly')
self.assertError('forget jemfinch') self.assertError('forget jemfinch')
self.assertError('forget jemfinch 3') self.assertError('forget jemfinch 3')
self.assertError('forget jemfinch 0') self.assertError('forget jemfinch 0')
self.assertNotError('forget jemfinch 2') self.assertNotError('forget jemfinch 2')
self.assertNotError('forget jemfinch 1') self.assertNotError('forget jemfinch 1')
self.assertError('whatis jemfinch') self.assertError('whatis jemfinch')
self.assertError('info jemfinch') self.assertError('info jemfinch')
self.assertNotError('learn foo bar as baz') self.assertNotError('learn foo bar as baz')
self.assertNotError('info foo bar') self.assertNotError('info foo bar')
self.assertRegexp('whatis foo bar', 'baz') self.assertRegexp('whatis foo bar', 'baz')
self.assertNotError('learn foo bar as quux') self.assertNotError('learn foo bar as quux')
self.assertRegexp('whatis foo bar', '.*baz.*quux') self.assertRegexp('whatis foo bar', '.*baz.*quux')
self.assertError('forget foo bar') self.assertError('forget foo bar')
self.assertNotError('forget foo bar 2') self.assertNotError('forget foo bar 2')
self.assertNotError('forget foo bar 1') self.assertNotError('forget foo bar 1')
self.assertError('whatis foo bar') self.assertError('whatis foo bar')
self.assertError('info foo bar') self.assertError('info foo bar')
self.assertError('learn foo bar baz') # No 'as' self.assertError('learn foo bar baz') # No 'as'
self.assertError('learn foo bar') # No 'as' self.assertError('learn foo bar') # No 'as'
def testChangeFactoid(self): def testChangeFactoid(self):
self.assertNotError('learn foo as bar')
self.assertNotError('change foo 1 s/bar/baz/')
self.assertRegexp('whatis foo', 'baz')
self.assertError('change foo 2 s/bar/baz/')
self.assertError('change foo 0 s/bar/baz/')
def testSearchFactoids(self):
self.assertNotError('learn jemfinch as my primary author')
self.assertNotError('learn strike as a cool guy working on me')
self.assertNotError('learn inkedmn as another of my developers')
self.assertNotError('learn jamessan as a developer of much python')
self.assertNotError('learn bwp as author of my weather command')
self.assertRegexp('factoids search --regexp /.w./', 'bwp')
self.assertRegexp('factoids search --regexp /^.+i/',
'jemfinch.*strike')
self.assertNotRegexp('factoids search --regexp /^.+i/', 'inkedmn')
self.assertRegexp('factoids search --regexp m/j/ --regexp m/ss/',
'jamessan')
self.assertRegexp('factoids search --regexp m/^j/ *ss*',
'jamessan')
self.assertRegexp('factoids search --regexp /^j/',
'jamessan.*jemfinch')
self.assertRegexp('factoids search j*', 'jamessan.*jemfinch')
self.assertRegexp('factoids search *ke*',
'inkedmn.*strike|strike.*inkedmn')
self.assertRegexp('factoids search ke',
'inkedmn.*strike|strike.*inkedmn')
self.assertRegexp('factoids search jemfinch',
'my primary author')
self.assertRegexp('factoids search --values primary author',
'my primary author')
def testWhatisOnNumbers(self):
self.assertNotError('learn 911 as emergency number')
self.assertRegexp('whatis 911', 'emergency number')
def testNotZeroIndexed(self):
self.assertNotError('learn foo as bar')
self.assertNotRegexp('info foo', '#0')
self.assertNotRegexp('whatis foo', '#0')
self.assertNotError('learn foo as baz')
self.assertNotRegexp('info foo', '#0')
self.assertNotRegexp('whatis foo', '#0')
def testInfoReturnsRightNumber(self):
self.assertNotError('learn foo as bar')
self.assertNotRegexp('info foo', '2 factoids')
def testLearnSeparator(self):
self.assertError('learn foo is bar')
self.assertNotError('learn foo as bar')
self.assertRegexp('whatis foo', 'bar')
orig = conf.supybot.plugins.Factoids.learnSeparator()
try:
conf.supybot.plugins.Factoids.learnSeparator.setValue('is')
self.assertError('learn bar as baz')
self.assertNotError('learn bar is baz')
self.assertRegexp('whatis bar', 'baz')
finally:
conf.supybot.plugins.Factoids.learnSeparator.setValue(orig)
def testShowFactoidIfOnlyOneMatch(self):
m1 = self.assertNotError('factoids search m/foo|bar/')
orig = conf.supybot.plugins.Factoids.showFactoidIfOnlyOneMatch()
try:
conf.supybot.plugins.Factoids. \
showFactoidIfOnlyOneMatch.setValue(False)
m2 = self.assertNotError('factoids search m/foo/')
self.failUnless(m1.args[1].startswith(m2.args[1]))
finally:
conf.supybot.plugins.Factoids. \
showFactoidIfOnlyOneMatch.setValue(orig)
def testInvalidCommand(self):
orig = conf.supybot.plugins.Factoids.replyWhenInvalidCommand()
try:
conf.supybot.plugins.Factoids.\
replyWhenInvalidCommand.setValue(True)
self.assertNotError('learn foo as bar') self.assertNotError('learn foo as bar')
self.assertNotError('change foo 1 s/bar/baz/') self.assertRegexp('foo', 'bar')
self.assertRegexp('whatis foo', 'baz') finally:
self.assertError('change foo 2 s/bar/baz/') conf.supybot.plugins.Factoids.\
self.assertError('change foo 0 s/bar/baz/') replyWhenInvalidCommand.setValue(orig)
def testSearchFactoids(self): def testRank(self):
self.assertNotError('learn jemfinch as my primary author') self.assertNotError('learn foo as bar')
self.assertNotError('learn strike as a cool guy working on me') self.assertNotError('learn moo as cow')
self.assertNotError('learn inkedmn as another of my developers') self.assertRegexp('factoids rank', '#1 foo \(0\), #2 moo \(0\)')
self.assertNotError('learn jamessan as a developer of much python') self.assertRegexp('whatis moo', '.*cow.*')
self.assertNotError('learn bwp as author of my weather command') self.assertRegexp('factoids rank', '#1 moo \(1\), #2 foo \(0\)')
self.assertRegexp('factoids search --regexp /.w./', 'bwp')
self.assertRegexp('factoids search --regexp /^.+i/',
'jemfinch.*strike')
self.assertNotRegexp('factoids search --regexp /^.+i/', 'inkedmn')
self.assertRegexp('factoids search --regexp m/j/ --regexp m/ss/',
'jamessan')
self.assertRegexp('factoids search --regexp m/^j/ *ss*',
'jamessan')
self.assertRegexp('factoids search --regexp /^j/',
'jamessan.*jemfinch')
self.assertRegexp('factoids search j*', 'jamessan.*jemfinch')
self.assertRegexp('factoids search *ke*',
'inkedmn.*strike|strike.*inkedmn')
self.assertRegexp('factoids search ke',
'inkedmn.*strike|strike.*inkedmn')
self.assertRegexp('factoids search jemfinch',
'my primary author')
self.assertRegexp('factoids search --values primary author',
'my primary author')
def testWhatisOnNumbers(self):
self.assertNotError('learn 911 as emergency number')
self.assertRegexp('whatis 911', 'emergency number')
def testNotZeroIndexed(self):
self.assertNotError('learn foo as bar')
self.assertNotRegexp('info foo', '#0')
self.assertNotRegexp('whatis foo', '#0')
self.assertNotError('learn foo as baz')
self.assertNotRegexp('info foo', '#0')
self.assertNotRegexp('whatis foo', '#0')
def testInfoReturnsRightNumber(self):
self.assertNotError('learn foo as bar')
self.assertNotRegexp('info foo', '2 factoids')
def testLearnSeparator(self):
self.assertError('learn foo is bar')
self.assertNotError('learn foo as bar')
self.assertRegexp('whatis foo', 'bar')
orig = conf.supybot.plugins.Factoids.learnSeparator()
try:
conf.supybot.plugins.Factoids.learnSeparator.setValue('is')
self.assertError('learn bar as baz')
self.assertNotError('learn bar is baz')
self.assertRegexp('whatis bar', 'baz')
finally:
conf.supybot.plugins.Factoids.learnSeparator.setValue(orig)
def testShowFactoidIfOnlyOneMatch(self):
m1 = self.assertNotError('factoids search m/foo|bar/')
orig = conf.supybot.plugins.Factoids.showFactoidIfOnlyOneMatch()
try:
conf.supybot.plugins.Factoids. \
showFactoidIfOnlyOneMatch.setValue(False)
m2 = self.assertNotError('factoids search m/foo/')
self.failUnless(m1.args[1].startswith(m2.args[1]))
finally:
conf.supybot.plugins.Factoids. \
showFactoidIfOnlyOneMatch.setValue(orig)
def testInvalidCommand(self):
orig = conf.supybot.plugins.Factoids.replyWhenInvalidCommand()
try:
conf.supybot.plugins.Factoids.\
replyWhenInvalidCommand.setValue(True)
self.assertNotError('learn foo as bar')
self.assertRegexp('foo', 'bar')
finally:
conf.supybot.plugins.Factoids.\
replyWhenInvalidCommand.setValue(orig)
def testQuoteHandling(self): def testQuoteHandling(self):
self.assertNotError('learn foo as "\\"bar\\""') self.assertNotError('learn foo as "\\"bar\\""')