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)