Avoid listing all permutations

This commit is contained in:
Valentin Lorentz 2023-07-08 16:41:21 +02:00
parent 1cbf992016
commit d5af301db1

View File

@ -29,8 +29,6 @@
# POSSIBILITY OF SUCH DAMAGE.
###
import itertools
from supybot.test import *
import supybot.conf as conf
@ -165,10 +163,10 @@ class ChannelTestCase(ChannelPluginTestCase):
def assertKban(self, query, *hostmasks, **kwargs):
m = self.getMsg(query, **kwargs)
self.assertIn(m, [
ircmsgs.bans(self.channel, permutation)
for permutation in itertools.permutations(hostmasks)
])
self.assertEqual(m.command, "MODE", m)
self.assertEqual(m.args[0], self.channel, m)
self.assertEqual(m.args[1], "+" + "b" * len(hostmasks), m)
self.assertCountEqual(m.args[2:], hostmasks, m)
m = self.getMsg(' ')
self.assertEqual(m.command, 'KICK')