def get_proximity_mac_addresses(self):
"""Get the mac addresses of the proximity devices.
Returns:
* A list of mac_addresses as strings, e.g.,
['aa:aa:aa:aa:aa:aa', 'bb:bb:bb:bb:bb:bb']
* None, if the connection to the proximity server failed.
"""
client = ProximityClient()
try:
mac_addresses = client.get_group(self.mac_address)
except ProximityClientConnectionError, e:
logger.error(e)
event_handler.add_event(e)
return None
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, [])
