def test_setClientIpAddress(self):
ipalias_id1 = self.storage.setClientIpAddress(IpAlias(ip='1.2.3.4', clientId=1))
self.assertIsNotNone(ipalias_id1)
ipalias_id2 = self.storage.setClientIpAddress(IpAlias(ip='127.0.0.1', clientId=1))
self.assertEqual(ipalias_id1 + 1, ipalias_id2)
ipalias_id3 = self.storage.setClientIpAddress(IpAlias(id=ipalias_id1, ip='0.0.0.0', clientId=1))
self.assertEqual(ipalias_id1, ipalias_id3)
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)
def test_getClientIpAddress(self):
ipalias = IpAlias(ip='88.44.55.22', timeAdd=12, timeEdit=654, numUsed=7, clientId=54)
ipalias_id = self.storage.setClientIpAddress(ipalias)
ipalias = self.storage.getClientIpAddress(IpAlias(id=ipalias_id))
self.assertIsInstance(ipalias, IpAlias)
self.assertEqual(ipalias.ip, '88.44.55.22')
self.assertEqual(ipalias.timeAdd, 12)
self.assertEqual(ipalias.timeEdit, 654)
self.assertEqual(ipalias.numUsed, 7)
self.assertEqual(ipalias.clientId, 54)
ipalias2 = self.storage.getClientIpAddress(IpAlias(clientId=ipalias.clientId, ip=ipalias.ip))
self.assertEqual(ipalias.id, ipalias2.id)
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)
def test_getClientIpAddresses(self):
client = Mock()
client.id = 15
self.storage.setClientIpAddress(IpAlias(ip='44.44.44.44', clientId=client.id))
self.storage.setClientIpAddress(IpAlias(ip='55.55.55.55', clientId=client.id))
self.storage.setClientIpAddress(IpAlias(ip='66.66.66.66', clientId=0))
ipaliases = self.storage.getClientIpAddresses(client)
self.assertIsInstance(ipaliases, list)
self.assertEqual(len(ipaliases), 2)
ips = []
for i in ipaliases:
self.assertIsInstance(i, IpAlias)
self.assertEqual(i.clientId, client.id)
self.assertNotEqual(i.id, None)
self.assertNotEqual(i.ip, None)
self.assertNotEqual(i.timeAdd, None)
self.assertNotEqual(i.timeEdit, None)
self.assertNotEqual(i.numUsed, None)
ips.append(i.ip)
self.assertIn('44.44.44.44', ips)
self.assertIn('55.55.55.55', ips)
self.assertNotIn('66.66.66.66', ips)
