class Test_plugin(SpamcontrolTestCase):
def setUp(self):
SpamcontrolTestCase.setUp(self)
self.adminPlugin = AdminPlugin(self.console, '@b3/conf/plugin_admin.ini')
when(self.console).getPlugin("admin").thenReturn(self.adminPlugin)
self.adminPlugin.onLoadConfig()
self.adminPlugin.onStartup()
with open(b3.functions.getAbsolutePath('@b3/conf/plugin_spamcontrol.ini')) as default_conf:
self.init_plugin(default_conf.read())
self.joe = FakeClient(self.console, name="Joe", guid="zaerezarezar", groupBits=1)
self.joe.connects("1")
self.superadmin = FakeClient(self.console, name="Superadmin", guid="superadmin_guid", groupBits=128)
self.superadmin.connects("2")
def assertSpaminsPoints(self, client, points):
actual = client.var(self.p, 'spamins', 0).value
self.assertEqual(points, actual, "expecting %s to have %s spamins points" % (client.name, points))
def test_say(self):
when(self.p).getTime().thenReturn(0).thenReturn(1).thenReturn(20).thenReturn(120)
self.assertSpaminsPoints(self.joe, 0)
self.joe.says("doh") # 0s
self.assertSpaminsPoints(self.joe, 2)
self.joe.says("foo") # 1s
self.assertSpaminsPoints(self.joe, 4)
self.joe.says("bar") # 20s
self.assertSpaminsPoints(self.joe, 3)
self.joe.says("hi") # 120s
self.assertSpaminsPoints(self.joe, 0)
class Test_game_specific_spam(SpamcontrolTestCase):
def setUp(self):
SpamcontrolTestCase.setUp(self)
with open(getAbsolutePath('@b3/conf/plugin_spamcontrol.ini')) as default_conf:
self.init_plugin(default_conf.read())
self.joe = FakeClient(self.console, name="Joe", exactName="Joe", guid="zaerezarezar", groupBits=1)
self.joe.connects("1")
# let's say our game has a new event : EVT_CLIENT_RADIO
EVT_CLIENT_RADIO = self.console.Events.createEvent('EVT_CLIENT_RADIO', 'Event client radio')
# teach the Spamcontrol plugin how to react on such events
def onRadio(this, event):
new_event = Event(type=event.type, client=event.client, target=event.target, data=event.data['text'])
this.onChat(new_event)
self.p.onRadio = MethodType(onRadio, self.p)
self.p.registerEvent('EVT_CLIENT_RADIO', self.p.onRadio)
# patch joe to make him able to send radio messages
def radios(me, text):
me.console.queueEvent(Event(type=EVT_CLIENT_RADIO, client=me, data={'text': text}))
self.joe.radios = MethodType(radios, self.joe)
def test_radio_spam(self):
when(self.p).getTime().thenReturn(0)
self.joe.warn = Mock()
self.joe.says("doh 1")
self.joe.radios("doh 2")
self.joe.says("doh 3")
self.joe.radios("doh 4")
self.joe.says("doh 5")
self.assertEqual(1, self.joe.warn.call_count)
class Test_cmd_balance(Iourt43TestCase):
def setUp(self):
super(Test_cmd_balance, self).setUp()
self.conf = CfgConfigParser()
self.conf.loadFromString(
dedent("""
[commands]
pabalance-balance: 1
[skillbalancer]
min_bal_interval: 1
interval: 0
difference: 0.5
mode: 2
"""))
self.p = PoweradminurtPlugin(self.console, self.conf)
self.init_default_cvar()
self.p.onLoadConfig()
self.p.onStartup()
self.console.say = Mock()
self.console.write = Mock()
with logging_disabled():
from tests.fake import FakeClient
self.blue1 = FakeClient(self.console,
name="Blue1",
guid="zaerezarezar",
groupBits=1,
team=TEAM_BLUE)
self.blue2 = FakeClient(self.console,
name="Blue2",
guid="qsdfdsqfdsqf",
groupBits=1,
team=TEAM_BLUE)
self.blue3 = FakeClient(self.console,
name="Blue3",
guid="qsdfdsqfdsqf33",
groupBits=1,
team=TEAM_BLUE)
self.blue4 = FakeClient(self.console,
name="Blue4",
guid="sdf455ezr",
groupBits=1,
team=TEAM_BLUE)
self.red1 = FakeClient(self.console,
name="Red1",
guid="875sasda",
groupBits=1,
team=TEAM_RED)
self.red2 = FakeClient(self.console,
name="Red2",
guid="f4qfer654r",
groupBits=1,
team=TEAM_RED)
# connect clients
self.blue1.connects('1')
self.blue2.connects('2')
self.blue3.connects('3')
self.blue4.connects('4')
self.red1.connects('5')
self.red2.connects('6')
self.p.countteams = Mock(return_value=True)
self.p._teamred = 2
self.p._teamblue = 4
def test_balancing_already_done(self):
# GIVEN
self.blue1.clearMessageHistory()
self.p._lastbal = 0
self.p.ignoreCheck = Mock(return_value=True)
self.console.time = Mock(return_value=10)
# WHEN
self.blue1.says('!balance')
# THEN
self.assertEqual(self.blue1.message_history,
['Teams changed recently, please wait a while'])
def test_non_round_based_gametype(self):
# GIVEN
self.blue1.clearMessageHistory()
self.p._lastbal = 0
self.p.ignoreCheck = Mock(return_value=False)
self.console.time = Mock(return_value=120)
self.console.game.gameType = 'tdm'
# WHEN
self.blue1.says('!balance')
# THEN
self.console.write.assert_has_calls(
[call('bigtext "Balancing teams!"')])
def test_round_based_gametype_delayed_announce_only(self):
# GIVEN
self.blue1.clearMessageHistory()
self.p._lastbal = 0
self.p.ignoreCheck = Mock(return_value=False)
self.console.time = Mock(return_value=120)
self.console.game.gameType = 'bm'
# WHEN
self.blue1.says('!balance')
# THEN
self.assertEqual(self.blue1.message_history,
['Teams will be balanced at the end of this round'])
self.assertFalse(self.p._pending_teambalance)
self.assertTrue(self.p._pending_skillbalance)
self.assertEqual(self.p.cmd_pabalance, self.p._skillbalance_func)
def test_round_based_gametype_delayed_execution(self):
# GIVEN
self.blue1.clearMessageHistory()
self.p._lastbal = 0
self.p.ignoreCheck = Mock(return_value=False)
self.console.time = Mock(return_value=120)
self.console.game.gameType = 'bm'
self.blue1.says('!balance')
self.assertEqual(self.blue1.message_history,
['Teams will be balanced at the end of this round'])
self.assertFalse(self.p._pending_teambalance)
self.assertTrue(self.p._pending_skillbalance)
self.assertEqual(self.p.cmd_pabalance, self.p._skillbalance_func)
# WHEN
self.console.queueEvent(self.console.getEvent('EVT_GAME_ROUND_END'))
# THEN
self.console.write.assert_has_calls(
[call('bigtext "Balancing teams!"')])
self.assertFalse(self.p._pending_teambalance)
self.assertFalse(self.p._pending_skillbalance)
self.assertIsNone(self.p._skillbalance_func)
class Test_commands(SpamcontrolTestCase):
def setUp(self):
SpamcontrolTestCase.setUp(self)
self.adminPlugin = AdminPlugin(self.console, '@b3/conf/plugin_admin.ini')
when(self.console).getPlugin("admin").thenReturn(self.adminPlugin)
self.adminPlugin.onLoadConfig()
self.adminPlugin.onStartup()
with open(b3.functions.getAbsolutePath('@b3/conf/plugin_spamcontrol.ini')) as default_conf:
self.init_plugin(default_conf.read())
self.joe = FakeClient(self.console, name="Joe", guid="zaerezarezar", groupBits=1)
self.joe.connects("1")
self.superadmin = FakeClient(self.console, name="Superadmin", guid="superadmin_guid", groupBits=128)
self.superadmin.connects("2")
def test_cmd_spamins(self):
# GIVEN
when(self.p).getTime().thenReturn(0).thenReturn(3).thenReturn(4).thenReturn(4).thenReturn(500)
self.joe.says("doh") # 0s
self.joe.says("doh") # 3s
self.joe.says("doh") # 4s
# WHEN
self.superadmin.clearMessageHistory()
self.superadmin.says("!spamins joe")
# THEN
self.assertListEqual(['Joe currently has 9 spamins, peak was 9'], self.superadmin.message_history) # 4s
# WHEN
self.superadmin.clearMessageHistory()
self.superadmin.says("!spamins joe")
self.assertListEqual(['Joe currently has 0 spamins, peak was 9'], self.superadmin.message_history) # 500s
def test_cmd_spamins_lowercase(self):
# GIVEN
mike = FakeClient(self.console, name="Mike")
mike.connects("3")
# WHEN
self.superadmin.clearMessageHistory()
self.superadmin.says("!spamins mike")
# THEN
self.assertListEqual(['Mike currently has 0 spamins, peak was 0'], self.superadmin.message_history)
def test_cmd_spamins_uppercase(self):
# GIVEN
mike = FakeClient(self.console, name="Mike")
mike.connects("3")
# WHEN
self.superadmin.clearMessageHistory()
self.superadmin.says("!spamins MIKE")
# THEN
self.assertListEqual(['Mike currently has 0 spamins, peak was 0'], self.superadmin.message_history)
def test_cmd_spamins_unknown_player(self):
# WHEN
self.superadmin.clearMessageHistory()
self.superadmin.says("!spamins nobody")
# THEN
self.assertListEqual(['No players found matching nobody'], self.superadmin.message_history)
def test_cmd_spamins_no_argument(self):
# GIVEN
self.console.getPlugin('admin')._warn_command_abusers = True
# WHEN
self.joe.clearMessageHistory()
self.joe.says("!spamins")
# THEN
self.assertListEqual(['You need to be in group Moderator to use !spamins'], self.joe.message_history)
# WHEN
self.superadmin.says("!putgroup joe mod")
self.joe.clearMessageHistory()
self.joe.says("!spamins")
# THEN
self.assertListEqual(['Joe is too cool to spam'], self.joe.message_history)
def test_joe_gets_warned(self):
# GIVEN
when(self.p).getTime().thenReturn(0)
self.joe.warn = Mock()
# WHEN
self.joe.says("doh 1")
self.joe.says("doh 2")
self.joe.says("doh 3")
self.joe.says("doh 4")
self.joe.says("doh 5")
# THEN
self.assertEqual(1, self.joe.warn.call_count)
class Test_welcome(Welcome_functional_test):
def setUp(self):
Welcome_functional_test.setUp(self)
self.load_config()
# disabled event handling (spawns threads and is of no use for that test)
self.p.onEvent = lambda *args, **kwargs: None
self.client = FakeClient(console=self.console, name="Jack", guid="JackGUID")
self.client._connections = 0
self.client.greeting = 'hi everyone :)'
self.client.connects("0")
self.superadmin.connects("1")
self.say_patcher = patch.object(self.console, "say")
self.say_mock = self.say_patcher.start()
def tearDown(self):
Welcome_functional_test.tearDown(self)
self.say_patcher.stop()
def Test_get_client_info(self):
self.parser_conf.add_section('b3')
self.parser_conf.set('b3', 'time_zone', 'UTC')
self.parser_conf.set('b3', 'time_format', '%I:%M%p %Z %m/%d/%y')
self.assertDictEqual({'connections': '1',
'group': 'Super Admin',
'id': '2',
'lastVisit': 'Unknown',
'level': '100',
'name': u'SuperAdmin^7'}, self.p.get_client_info(self.superadmin))
# WHEN
self.superadmin.lastVisit = 1364821993
self.superadmin._connections = 2
# THEN
self.assertDictEqual({'connections': '2',
'group': u'Super Admin',
'id': '2',
'lastVisit': '02:13PM CET 04/01/13',
'level': '100',
'name': 'SuperAdmin^7'}, self.p.get_client_info(self.superadmin))
# WHEN
self.superadmin.says("!mask mod")
# THEN
self.assertDictEqual({'connections': '2',
'group': u'Moderator',
'id': '2',
'lastVisit': '02:13PM CET 04/01/13',
'level': '20',
'name': 'SuperAdmin^7'}, self.p.get_client_info(self.superadmin))
def test_0(self):
# GIVEN
self.p._welcomeFlags = 0
# WHEN
self.p.welcome(self.superadmin)
# THEN
self.assertListEqual([], self.say_mock.mock_calls)
self.assertListEqual([], self.superadmin.message_history)
def test_first(self):
# GIVEN
self.client._connections = 0
self.p._welcomeFlags = F_FIRST
# WHEN
self.p.welcome(self.client)
# THEN
self.assertListEqual([], self.say_mock.mock_calls)
self.assertListEqual(['Welcome Jack, this must be your first visit, you are player #1. Type !help for '
'help'], self.client.message_history)
def test_newb(self):
# GIVEN
self.client._connections = 2
self.p._welcomeFlags = F_NEWB
# WHEN
self.p.welcome(self.client)
# THEN
self.assertListEqual([], self.say_mock.mock_calls)
self.assertListEqual(['[Authed] Welcome back Jack [@1], last visit Unknown. Type !register in chat to register.'
' Type !help for help'], self.client.message_history)
def test_user(self):
# GIVEN
self.client._connections = 2
self.p._welcomeFlags = F_USER
self.client.says("!register")
self.client.clearMessageHistory()
# WHEN
self.p.welcome(self.client)
# THEN
self.assertListEqual([call(u'^7Jack^7 ^7put in group User')], self.say_mock.mock_calls)
self.assertListEqual(["[Authed] Welcome back Jack [@1], last visit Unknown, you're a User, played 2 times"],
self.client.message_history)
def test_announce_first(self):
# GIVEN
self.client._connections = 0
self.p._welcomeFlags = F_ANNOUNCE_FIRST
# WHEN
self.p.welcome(self.client)
# THEN
self.assertListEqual([call('^7Everyone welcome Jack^7^7, player number ^3#1^7, to the server')],
self.say_mock.mock_calls)
self.assertListEqual([], self.client.message_history)
def test_announce_user(self):
# GIVEN
self.client._connections = 2
self.p._welcomeFlags = F_ANNOUNCE_USER
self.client.says("!register")
self.client.clearMessageHistory()
# WHEN
self.p.welcome(self.client)
# THEN
self.assertListEqual([call(u'^7Jack^7 ^7put in group User'),
call('^7Everyone welcome back Jack^7^7, player number ^3#1^7, to the server, played 2 '
'times')], self.say_mock.mock_calls)
self.assertListEqual([], self.client.message_history)
def test_custom_greeting(self):
# GIVEN
self.client._connections = 2
self.p._welcomeFlags = F_CUSTOM_GREETING
self.client.says("!register")
self.client.clearMessageHistory()
# WHEN
self.p.welcome(self.client)
# THEN
self.assertListEqual([call(u'^7Jack^7 ^7put in group User'), call('^7Jack^7^7 joined: hi everyone :)')],
self.say_mock.mock_calls)
self.assertListEqual([], self.client.message_history)
