class ProximityClientTestCase(TestCase):
def setUp(self):
self.old_setting = settings.PROXIMITY_SERVER['default']
settings.PROXIMITY_SERVER['default'] = settings.PROXIMITY_SERVER['test']
self.proximity_client = ProximityClient()
self.proximity_client.flush()
def tearDown(self):
settings.PROXIMITY_SERVER['default'] = self.old_setting
def test_good_add_set_get_flush(self):
self.proximity_client.add_device('aa:aa:aa:aa:aa:aa')
group = self.proximity_client.get_group('aa:aa:aa:aa:aa:aa')
self.assertEqual(group, [])
self.proximity_client.add_device('bb:bb:bb:bb:bb:bb')
group = self.proximity_client.get_group('aa:aa:aa:aa:aa:aa')
self.assertEqual(group, [])
self.proximity_client.add_device('cc:cc:cc:cc:cc:cc')
group = self.proximity_client.get_group('cc:cc:cc:cc:cc:cc')
self.assertEqual(group, [])
self.proximity_client.set_group('aa:aa:aa:aa:aa:aa', ['bb:bb:bb:bb:bb:bb'])
group = self.proximity_client.get_group('aa:aa:aa:aa:aa:aa')
self.assertEqual(group, ['bb:bb:bb:bb:bb:bb'])
group = self.proximity_client.get_group('bb:bb:bb:bb:bb:bb')
self.assertEqual(group, ['aa:aa:aa:aa:aa:aa'])
group = self.proximity_client.get_group('cc:cc:cc:cc:cc:cc')
self.assertEqual(group, [])
self.proximity_client.set_group('cc:cc:cc:cc:cc:cc', ['bb:bb:bb:bb:bb:bb'])
group = self.proximity_client.get_group('aa:aa:aa:aa:aa:aa')
self.assertEqual(group, ['bb:bb:bb:bb:bb:bb', 'cc:cc:cc:cc:cc:cc'])
group = self.proximity_client.get_group('bb:bb:bb:bb:bb:bb')
self.assertEqual(group, ['aa:aa:aa:aa:aa:aa', 'cc:cc:cc:cc:cc:cc'])
group = self.proximity_client.get_group('cc:cc:cc:cc:cc:cc')
self.assertEqual(group, ['aa:aa:aa:aa:aa:aa', 'bb:bb:bb:bb:bb:bb'])
self.proximity_client.flush()
group = self.proximity_client.get_group('aa:aa:aa:aa:aa:aa')
self.assertEqual(group, [])
group = self.proximity_client.get_group('bb:bb:bb:bb:bb:bb')
self.assertEqual(group, [])
group = self.proximity_client.get_group('cc:cc:cc:cc:cc:cc')
self.assertEqual(group, [])
def set_proximity_devices(self, neighbours):
"""Set the proximity devices of the device.
This method sets the proximity devices, i.e., the neighbouring
devices as it's proximity devices.
Args:
* neighbours: A list of mac_addresses as strings, e.g.,
['aa:aa:aa:aa:aa:aa', 'bb:bb:bb:bb:bb:bb']
"""
client = ProximityClient()
try:
client.set_group(self.mac_address, neighbours)
except ProximityClientConnectionError, e:
logger.error(e)
event_handler.add_event(e)