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)