def list_user_devices(self, request): """Lists all devices assigned to the user.""" self.check_xsrf_token(self.request_state) user = user_lib.get_user_email() guest_permitted = config_model.Config.get('allow_guest_mode') device_message_list = [] for device in device_model.Device.list_by_user(user): device_message_list.append( api_utils.build_device_message_from_model(device, guest_permitted)) return device_messages.ListUserDeviceResponse(devices=device_message_list)
def testListUserDeviceResponse(self): device1 = device_messages.Device(serial_number='FAKE-DEVICE-SERIAL-1') device2 = device_messages.Device(serial_number='FAKE-DEVICE-SERIAL-2') list_device_resp = device_messages.ListUserDeviceResponse( devices=[device1, device2]) self.assertEqual(list_device_resp.devices[0].serial_number, 'FAKE-DEVICE-SERIAL-1') self.assertEqual(list_device_resp.devices[1].serial_number, 'FAKE-DEVICE-SERIAL-2')