### # Copyright (c) 2015,2021 James Lu # 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. ### import time import uuid from supybot.test import * class RelayNextDBTestCase(PluginTestCase): plugins = ('RelayNext',) def testAdd(self): self.assertNotError("relaynext add test #channel1@somenet #channel2@othernet") self.assertRegexp("relaynext list", "#channel1@somenet") self.assertRegexp("relaynext list", "#channel2@othernet") # Incorrect amount of arguments self.assertError("relaynext add test") self.assertError("relaynext add test afdasfader agkajeoig #validchannel@othernet") self.assertError("relaynext add nonexistant-relay #channel1@somenet") def testSet(self): self.assertNotError("relaynext set test #channel1@somenet #channel2@othernet") self.assertRegexp("relaynext list", "#channel1@somenet") self.assertRegexp("relaynext list", "#channel2@othernet") self.assertError("relaynext set test") self.assertError("relaynext set test #channel1@somenet") def testList(self): # This should error if no relays are defined self.assertError("relaynext list") self.assertNotError("relaynext set abcd #moo@moo #test@test") self.assertRegexp("relaynext list", "#moo@moo") self.assertRegexp("relaynext list", "#test@test") def testSetCaseInsensitive(self): self.assertError("relaynext set foobar #deV@overdrive-irc #DEV@overdrive-irc") self.assertError("relaynext set foobar #dev@overdrive-irc #DEV@OVERdrive-IRC") self.assertError("relaynext set foobar #dev@overdrive-irc #dev@OVERdrive-IRC") self.assertNotError("relaynext set abcd #moo@moo #MOO@mOO #test@test") self.assertRegexp("relaynext list", "#moo@moo") self.assertRegexp("relaynext list", "#test@test") def testAddCaseInsensitive(self): self.assertError("relaynext add foobar #deV@overdrive-irc #DEV@overdrive-irc") self.assertError("relaynext add foobar #dev@overdrive-irc #DEV@OVERdrive-IRC") self.assertError("relaynext add foobar #dev@overdrive-irc #dev@OVERdrive-IRC") self.assertNotError("relaynext add abcd #moo@moo #MOO@mOO #test@test") self.assertRegexp("relaynext list", "#moo@moo") self.assertRegexp("relaynext list", "#test@test") def testRemoveEntireRelay(self): self.assertError("relaynext remove some-relay") self.assertNotError("relaynext add some-relay #channel1@somenet #channel2@othernet #abcd@test123") # 'remove' without arguments removes an entire relay self.assertNotError("relaynext remove some-relay") # No relays should be defined now. self.assertError("relaynext list") def testRemoveChannelsFromExistingRelay(self): self.assertNotError("relaynext add some-relay #channel1@somenet #channel2@othernet #abcd@test123") # This should give a warning (removing a channel that's not in the specified relay) self.assertRegexp("relaynext remove some-relay #asfdafaergea@random", 'not found in the original relay: ' '#asfdafaergea@random') self.assertNotError("relaynext remove some-relay #abcd@test123") self.assertRegexp("relaynext list", "#channel1@somenet") self.assertRegexp("relaynext list", "#channel2@othernet") self.assertNotRegexp("relaynext list", "#abcd@test123") def testAutoremoveWhenLessThanTwoChannels(self): self.assertNotError("relaynext add some-relay #channel1@somenet #channel2@othernet #abcd@test123") self.assertNotError("relaynext remove some-relay #abcd@test123 #channel2@othernet") self.assertError("relaynext list") def testAddChannelsToExistingRelay(self): self.assertNotError("relaynext add test #channel1@somenet #channel2@othernet") self.assertNotError("relaynext add test #somewhereElse@mynet") self.assertRegexp("relaynext list", "#channel1@somenet") self.assertRegexp("relaynext list", "#channel2@otherne") self.assertRegexp("relaynext list", "#somewhereelse@mynet") def testClear(self): self.assertNotError("relaynext set test #channel1@somenet #channel2@othernet") self.assertNotError("relaynext clear") self.assertError("relaynext list") class RelayNextTestCase(PluginTestCase): plugins = ('RelayNext',) # Disable colours to make output checking more predictable config = {'plugins.RelayNext.color': False} def setUp(self): super().setUp() # These test network names are defined in scripts/supybot-test. Anything unknown will fail with: # supybot.registry.NonExistentRegistryEntry: 'net1' is not a valid entry in 'supybot.networks' self.irc1 = getTestIrc("testnet1") self.chan1 = irclib.ChannelState() self.chan1name = '#' + uuid.uuid4().hex self.chan1.addUser(self.irc1.nick) # Add the bot to the channel self.irc1.state.channels = {self.chan1name: self.chan1} self.irc2 = getTestIrc("testnet2") self.chan2 = irclib.ChannelState() self.chan2.addUser(self.irc2.nick) self.chan2name = '#' + uuid.uuid4().hex self.irc2.state.channels = {self.chan2name: self.chan2} self.assertNotError("relaynext set testRelay %s@testnet1 %s@testnet2" % (self.chan1name, self.chan2name)) def getCommandResponse(self, irc): # Adapted from PluginTestCase._feedMsg() to account for other irc objects response = irc.takeMsg() fed = time.time() while response is None and time.time() - fed < self.timeout: time.sleep(0.01) drivers.run() response = irc.takeMsg() return response def testJoin(self): msg = ircmsgs.join(self.chan1name, prefix='testUser1!testuser@example.com') self.irc1.feedMsg(msg) output = self.getCommandResponse(self.irc2) self.assertEqual('\x02[testnet1]\x02 testUser1 (testuser@example.com) has joined %s' % self.chan1name, output.args[1]) def testPartBare(self): msg = ircmsgs.part(self.chan1name, prefix='foo!bar@baz') self.irc1.feedMsg(msg) output = self.getCommandResponse(self.irc2) self.assertEqual('\x02[testnet1]\x02 foo (bar@baz) has left %s' % self.chan1name, output.args[1]) def testPartWithReason(self): msg = ircmsgs.part(self.chan1name, 'foobar', prefix='foo!bar@baz') self.irc1.feedMsg(msg) output = self.getCommandResponse(self.irc2) self.assertEqual('\x02[testnet1]\x02 foo (bar@baz) has left %s (foobar)' % self.chan1name, output.args[1]) # TODO: all other messages # TODO: toggles to show/hide specific events # TODO: colours # TODO: blockHighlights # TODO: relaySelfMessages and ignores # TODO: !nicks command # TODO: antiflood # TODO: announcement routing (trigger relay() manually) # vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79: