class Test_Client(B3TestCase):
def setUp(self):
B3TestCase.setUp(self)
self.client = Client(console=self.console)
def test_construct(self):
c = Client(name="Courgette", guid="1234567890")
self.assertEqual(c.name, "Courgette")
self.assertEqual(c.exactName, "Courgette^7")
self.assertEqual(c.guid, "1234567890")
self.assertEqual(c.team, TEAM_UNKNOWN)
self.assertTrue(c.connected)
self.assertFalse(c.hide)
self.assertEqual(c.ip, '')
self.assertEqual(c.greeting, '')
self.assertEqual(c.pbid, '')
def test_team(self):
m = Mock()
self.client.team = m
self.assertEqual(self.client.team, m)
def test_team_change(self):
# GIVEN
self.client._team = TEAM_RED
# THEN
with self.assertRaiseEvent(event_type='EVT_CLIENT_TEAM_CHANGE', event_client=self.client, event_data=TEAM_BLUE):
# WHEN
self.client.team = TEAM_BLUE
# GIVEN
self.client._team = TEAM_RED
# THEN
with self.assertRaiseEvent(event_type='EVT_CLIENT_TEAM_CHANGE2', event_client=self.client,
event_data={'previous': TEAM_RED, 'new': TEAM_BLUE}):
# WHEN
self.client.team = TEAM_BLUE
def test_name_change(self):
# GIVEN
self.client.authed = True
# THEN
with self.assertRaiseEvent(event_type='EVT_CLIENT_NAME_CHANGE', event_client=self.client, event_data="cucurb"):
# WHEN
self.client.name = "cucurb"
def test_makeAlias_new(self):
self.client.id = 123
self.console.storage.getClientAlias = Mock(side_effect = KeyError())
self.client.makeAlias("bar")
self.assertEqual(self.console.storage.getClientAlias.call_count, 1)
alias = self.console.storage.getClientAlias.call_args[0][0]
self.assertIsInstance(alias, Alias)
self.assertEqual(alias.alias, "bar")
self.assertEqual(alias.numUsed, 1)
def test_makeAlias_existing(self):
self.client.id = 123
aliasFoo = Alias()
aliasFoo.alias = "foo"
aliasFoo.clientId = self.client.id
aliasFoo.numUsed = 48
self.console.storage.getClientAlias = Mock(side_effect = lambda x: aliasFoo)
self.client.makeAlias("whatever")
self.assertEqual(self.console.storage.getClientAlias.call_count, 1)
self.assertIsInstance(aliasFoo, Alias)
self.assertEqual(aliasFoo.alias, "foo")
self.assertEqual(aliasFoo.numUsed, 49)
def test_guid_readonly(self):
self.assertFalse(self.client.authed)
self.client.guid = "foo"
self.assertEqual(self.client.guid, "foo")
self.client.auth()
self.assertTrue(self.client.authed)
# upon guid change, prevent change and consider client not
# authed anymore
self.client.guid = "bar"
self.assertFalse(self.client.authed)
self.client.guid = "foo"
def test_set_ip(self):
self.client.ip = "1.2.3.4"
self.assertEqual(self.client._ip, "1.2.3.4")
self.client.ip = "5.6.7.8:27960"
self.assertEqual(self.client._ip, "5.6.7.8")
def test_makeIpAlias_new(self):
self.client.id = 123
self.console.storage.getClientIpAddress = Mock(side_effect = KeyError())
self.client.makeIpAlias("1.4.7.8")
self.assertEqual(self.console.storage.getClientIpAddress.call_count, 1)
alias = self.console.storage.getClientIpAddress.call_args[0][0]
self.assertIsInstance(alias, IpAlias)
self.assertEqual(alias.ip, "1.4.7.8")
self.assertEqual(alias.numUsed, 1)
def test_makeIpAlias_existing(self):
self.client.id = 123
aliasFoo = IpAlias()
aliasFoo.ip = "9.5.4.4"
aliasFoo.clientId = self.client.id
aliasFoo.numUsed = 8
self.console.storage.getClientIpAddress = Mock(side_effect = lambda x: aliasFoo)
self.client.makeIpAlias("whatever")
self.assertEqual(self.console.storage.getClientIpAddress.call_count, 1)
self.assertIsInstance(aliasFoo, IpAlias)
self.assertEqual(aliasFoo.ip, "9.5.4.4")
self.assertEqual(aliasFoo.numUsed, 9)
class Test_Client(B3TestCase):
def setUp(self):
B3TestCase.setUp(self)
self.client = Client(console=self.console)
def test_construct(self):
c = Client(name="Courgette", guid="1234567890")
self.assertEqual(c.name, "Courgette")
self.assertEqual(c.exactName, "Courgette^7")
self.assertEqual(c.guid, "1234567890")
self.assertEqual(c.team, TEAM_UNKNOWN)
self.assertTrue(c.connected)
self.assertFalse(c.hide)
self.assertEqual(c.ip, '')
self.assertEqual(c.greeting, '')
self.assertEqual(c.pbid, '')
def test_team(self):
m = Mock()
self.client.team = m
self.assertEqual(self.client.team, m)
def test_team_change(self):
self.console.queueEvent = Mock()
self.client.team = 24
self.assertEqual(self.client.team, 24)
self.console.queueEvent.assert_called()
args = self.console.queueEvent.call_args
eventraised = args[0][0]
self.assertEquals(eventraised.type, EVT_CLIENT_TEAM_CHANGE)
self.assertEquals(eventraised.data, 24)
def test_name_change(self):
self.console.queueEvent = Mock()
self.client.authed = True
self.client.name = "cucurb"
self.assertEqual(self.client.name, "cucurb")
self.console.queueEvent.assert_called()
args = self.console.queueEvent.call_args
eventraised = args[0][0]
self.assertEquals(eventraised.type, EVT_CLIENT_NAME_CHANGE)
self.assertEquals(eventraised.data, 'cucurb')
def test_makeAlias_new(self):
self.client.id = 123
self.console.storage.getClientAlias = Mock(side_effect = KeyError())
self.client.makeAlias("bar")
self.assertEquals(self.console.storage.getClientAlias.call_count, 1)
alias = self.console.storage.getClientAlias.call_args[0][0]
self.assertIsInstance(alias, Alias)
self.assertEqual(alias.alias, "bar")
self.assertEqual(alias.numUsed, 1)
def test_makeAlias_existing(self):
self.client.id = 123
aliasFoo = Alias()
aliasFoo.alias = "foo"
aliasFoo.clientId = self.client.id
aliasFoo.numUsed = 48
self.console.storage.getClientAlias = Mock(side_effect = lambda x: aliasFoo)
self.client.makeAlias("whatever")
self.assertEquals(self.console.storage.getClientAlias.call_count, 1)
self.assertIsInstance(aliasFoo, Alias)
self.assertEqual(aliasFoo.alias, "foo")
self.assertEqual(aliasFoo.numUsed, 49)
def test_guid_readonly(self):
self.assertFalse(self.client.authed)
self.client.guid = "foo"
self.assertEqual(self.client.guid, "foo")
self.client.auth()
self.assertTrue(self.client.authed)
# upon guid change, prevent change and consider client not
# authed anymore
self.client.guid = "bar"
self.assertFalse(self.client.authed)
self.client.guid = "foo"
def test_set_ip(self):
self.client.ip = "1.2.3.4"
self.assertEqual(self.client._ip, "1.2.3.4")
self.client.ip = "5.6.7.8:27960"
self.assertEqual(self.client._ip, "5.6.7.8")
def test_makeIpAlias_new(self):
self.client.id = 123
self.console.storage.getClientIpAddress = Mock(side_effect = KeyError())
self.client.makeIpAlias("1.4.7.8")
self.assertEquals(self.console.storage.getClientIpAddress.call_count, 1)
alias = self.console.storage.getClientIpAddress.call_args[0][0]
self.assertIsInstance(alias, IpAlias)
self.assertEqual(alias.ip, "1.4.7.8")
self.assertEqual(alias.numUsed, 1)
def test_makeIpAlias_existing(self):
self.client.id = 123
aliasFoo = IpAlias()
aliasFoo.ip = "9.5.4.4"
aliasFoo.clientId = self.client.id
aliasFoo.numUsed = 8
self.console.storage.getClientIpAddress = Mock(side_effect = lambda x: aliasFoo)
self.client.makeIpAlias("whatever")
self.assertEquals(self.console.storage.getClientIpAddress.call_count, 1)
self.assertIsInstance(aliasFoo, IpAlias)
self.assertEqual(aliasFoo.ip, "9.5.4.4")
self.assertEqual(aliasFoo.numUsed, 9)
class Test_Client(B3TestCase):
def setUp(self):
B3TestCase.setUp(self)
self.client = Client(console=b3.console)
def test_construct(self):
c = Client(name="Courgette", guid="1234567890")
self.assertEqual(c.name, "Courgette")
self.assertEqual(c.exactName, "Courgette^7")
self.assertEqual(c.guid, "1234567890")
self.assertEqual(c.team, b3.TEAM_UNKNOWN)
self.assertTrue(c.connected)
self.assertFalse(c.hide)
self.assertEqual(c.ip, '')
self.assertEqual(c.greeting, '')
self.assertEqual(c.pbid, '')
def test_team(self):
m = Mock()
self.client.team = m
self.assertEqual(self.client.team, m)
def test_team_change(self):
self.client.team = 24
self.assertEqual(self.client.team, 24)
b3.console.queueEvent.assert_called()
args = b3.console.queueEvent.call_args
eventraised = args[0][0]
self.assertEquals(eventraised.type, b3.events.EVT_CLIENT_TEAM_CHANGE)
self.assertEquals(eventraised.data, 24)
def test_name_change(self):
self.client.authed = True
self.client.name = "cucurb"
self.assertEqual(self.client.name, "cucurb")
b3.console.queueEvent.assert_called()
args = b3.console.queueEvent.call_args
eventraised = args[0][0]
self.assertEquals(eventraised.type, b3.events.EVT_CLIENT_NAME_CHANGE)
self.assertEquals(eventraised.data, 'cucurb')
def test_makeAlias_new(self):
self.client.id = 123
b3.console.storage.getClientAlias.side_effect = KeyError()
self.client.makeAlias("bar")
self.assertEquals(b3.console.storage.getClientAlias.call_count, 1)
alias = b3.console.storage.getClientAlias.call_args[0][0]
self.assertIsInstance(alias, b3.clients.Alias)
self.assertEqual(alias.alias, "bar")
self.assertEqual(alias.numUsed, 1)
def test_makeAlias_existing(self):
self.client.id = 123
aliasFoo = b3.clients.Alias()
aliasFoo.alias = "foo"
aliasFoo.clientId = self.client.id
aliasFoo.numUsed = 48
b3.console.storage.getClientAlias.side_effect = lambda x: aliasFoo
self.client.makeAlias("whatever")
self.assertEquals(b3.console.storage.getClientAlias.call_count, 1)
self.assertIsInstance(aliasFoo, b3.clients.Alias)
self.assertEqual(aliasFoo.alias, "foo")
self.assertEqual(aliasFoo.numUsed, 49)
def test_guid_readonly(self):
self.assertFalse(self.client.authed)
self.client.guid = "foo"
self.assertEqual(self.client.guid, "foo")
self.client.auth()
self.assertTrue(self.client.authed)
# upon guid change, prevent change and consider client not
# authed anymore
self.client.guid = "bar"
self.assertFalse(self.client.authed)
self.client.guid = "foo"
def test_set_ip(self):
self.client.ip = "1.2.3.4"
self.assertEqual(self.client._ip, "1.2.3.4")
self.client.ip = "5.6.7.8:27960"
self.assertEqual(self.client._ip, "5.6.7.8")
def test_makeIpAlias_new(self):
self.client.id = 123
b3.console.storage.getClientIpAddress.side_effect = KeyError()
self.client.makeIpAlias("1.4.7.8")
self.assertEquals(b3.console.storage.getClientIpAddress.call_count, 1)
alias = b3.console.storage.getClientIpAddress.call_args[0][0]
self.assertIsInstance(alias, b3.clients.IpAlias)
self.assertEqual(alias.ip, "1.4.7.8")
self.assertEqual(alias.numUsed, 1)
def test_makeIpAlias_existing(self):
self.client.id = 123
aliasFoo = b3.clients.IpAlias()
aliasFoo.ip = "9.5.4.4"
aliasFoo.clientId = self.client.id
aliasFoo.numUsed = 8
b3.console.storage.getClientIpAddress.side_effect = lambda x: aliasFoo
self.client.makeIpAlias("whatever")
self.assertEquals(b3.console.storage.getClientIpAddress.call_count, 1)
self.assertIsInstance(aliasFoo, b3.clients.IpAlias)
self.assertEqual(aliasFoo.ip, "9.5.4.4")
self.assertEqual(aliasFoo.numUsed, 9)
