From eda770ee0fe96020959e88d70d45dfbc06441f0c Mon Sep 17 00:00:00 2001
From: Valentin Lorentz <progval@progval.net>
Date: Fri, 1 Feb 2013 19:50:21 +0100
Subject: [PATCH] AttackProtector: Fix tests.

---
 AttackProtector/test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/AttackProtector/test.py b/AttackProtector/test.py
index 0f35b5c..94d9648 100644
--- a/AttackProtector/test.py
+++ b/AttackProtector/test.py
@@ -49,7 +49,7 @@ class AttackProtectorTestCase(ChannelPluginTestCase):
         return False
     def _getIfAnswerIsThisBan(self, banmask=None):
         if banmask is None:
-            banmask = self.prefix
+            banmask = '*!' + (self.prefix.split('!')[1])
         return self._getIfAnswerIsEqual(ircmsgs.ban(self.channel, banmask))
     def _getIfAnswerIsThisKick(self, kind):
         reason = '%s flood detected' % kind