def testDamagedRequest(self):
device = device_messages.DeviceRequest(asset_tag='FAKE-TAG')
damaged_request = device_messages.DamagedRequest(
device=device, damaged_reason='FAKE-DAMAGED-REASON')
self.assertEqual(damaged_request.device.asset_tag, 'FAKE-TAG')
self.assertEqual(damaged_request.damaged_reason, 'FAKE-DAMAGED-REASON')
def test_mark_damaged__unauthorized(self, mock_xsrf, mock_markdamaged):
del mock_xsrf # unusued.
self.login_endpoints_user()
mock_markdamaged.side_effect = device_model.UnauthorizedError()
with self.assertRaises(endpoints.UnauthorizedException):
self.service.mark_damaged(
device_messages.DamagedRequest(
device=device_messages.DeviceRequest(
urlkey=self.device.key.urlsafe()),
damaged_reason='Foo'))
def test_mark_damaged(self, mock_xsrf_token, mock_markdamaged):
self.login_endpoints_user()
self.service.mark_damaged(
device_messages.DamagedRequest(
device=device_messages.DeviceRequest(
urlkey=self.device.key.urlsafe()),
damaged_reason='Foo'))
mock_markdamaged.assert_called_once_with(
user_email=loanertest.USER_EMAIL, damaged_reason='Foo')
assert mock_xsrf_token.call_count == 1
mock_xsrf_token.reset_mock()
mock_markdamaged.reset_mock()
self.service.mark_damaged(
device_messages.DamagedRequest( # No reason given.
device=device_messages.DeviceRequest(
urlkey=self.device.key.urlsafe())))
mock_markdamaged.assert_called_once_with(
user_email=loanertest.USER_EMAIL, damaged_reason=None)
assert mock_xsrf_token.call_count == 1
