def test_handle_registration_id_change(self):
     handler = RegistrationHandler(self.device_registry)
     handler.handle_registration_id_change_for_node(self.registration_request)
     verify(self.device_registry).change_registration_id_for_node(self.node_id, self.registration_id)
 def test_handle_registration(self):
     handler = RegistrationHandler(self.device_registry)
     handler.handle_registration(self.registration_request)
     verify(self.device_registry).save_device_registration(self.node_id, self.registration_id)