def on_new_device(token):
if not utils.validate_device(token):
return 422, "Invalid device"
try:
storage.add_device(token)
notifier.notify_device_added(token)
return 200, "Device added"
except Exception as e:
return 500, "Cannot process device: " + str(e)
def test_token_update(self):
device_1 = dict(device_name='d1', device_token='t1')
device_2 = dict(device_name='d2', device_token='t2')
storage.add_device(device_1)
devices = storage.get_all_devices()
self.assertEqual(1, len(devices))
storage.add_device(device_2)
devices = storage.get_all_devices()
self.assertEqual(2, len(devices))
device_1['device_token'] = 't_updated'
storage.add_device(device_1)
devices = storage.get_all_devices()
self.assertEqual(2, len(devices))
self.assertTrue(('d1', 't_updated') in devices)
