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)