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)
def notify(cmp_result, system_state):
push_sender.send_to_all(storage.get_all_devices(), system_state)
