def lease(self, build_id, lease_expiration_date=None, expect_success=True):
success, build = service.lease(
build_id,
lease_expiration_date=lease_expiration_date,
)
self.assertEqual(success, expect_success)
return build
def lease(self, lease_expiration_date=None):
if self.test_build.key is None:
self.test_build.put()
success, self.test_build = service.lease(
self.test_build.key.id(),
lease_expiration_date=lease_expiration_date,
)
return success
def lease(self, request):
"""Leases a build.
Response may contain an error.
"""
success, build = service.lease(
request.id,
lease_expiration_date=parse_datetime(request.lease_expiration_ts),
)
if not success:
return BuildResponseMessage(error=ErrorMessage(
message='Could not lease build',
reason=ErrorReason.CANNOT_LEASE_BUILD,
))
assert build.lease_key is not None
return build_to_response_message(build, include_lease_key=True)
def lease(self, request):
"""Leases a build.
Response may contain an error.
"""
success, build = service.lease(
request.id,
lease_expiration_date=parse_datetime(request.lease_expiration_ts),
)
if not success:
return BuildResponseMessage(error=ErrorMessage(
message='Could not lease build',
reason=ErrorReason.CANNOT_LEASE_BUILD,
))
assert build.lease_key is not None
return build_to_response_message(build, include_lease_key=True)
def test_cannot_lease_a_nonexistent_build(self):
with self.assertRaises(errors.BuildNotFoundError):
service.lease(build_id=42)
def test_cannot_lease_a_nonexistent_build(self):
with self.assertRaises(errors.BuildNotFoundError):
service.lease(build_id=42)