def validate_to_run_key(task_key):
"""Validates a ndb.Key to a TaskToRun entity. Raises ValueError if invalid."""
# This also validates the key kind.
request_key = task_to_run_key_to_request_key(task_key)
key_id = task_key.integer_id()
if not key_id or key_id >= 2**32:
raise ValueError(
'TaskToRun key id should be between 1 and 2**32, found %s' %
task_key.id())
task_request.validate_request_key(request_key)
def validate_to_run_key(task_key):
"""Validates a ndb.Key to a TaskToRun entity. Raises ValueError if invalid."""
# This also validates the key kind.
request_key = task_to_run_key_to_request_key(task_key)
key_id = task_key.integer_id()
if not key_id or key_id >= 2**32:
raise ValueError(
'TaskToRun key id should be between 1 and 2**32, found %s' %
task_key.id())
task_request.validate_request_key(request_key)
def test_validate_request_key(self):
task_request.validate_request_key(task_pack.unpack_request_key("10"))
task_request.validate_request_key(
ndb.Key("TaskRequestShard", "a" * task_pack.DEPRECATED_SHARDING_LEVEL, "TaskRequest", 0x100)
)
with self.assertRaises(ValueError):
task_request.validate_request_key(ndb.Key("TaskRequest", 1))
with self.assertRaises(ValueError):
key = ndb.Key("TaskRequestShard", "a" * (task_pack.DEPRECATED_SHARDING_LEVEL + 1), "TaskRequest", 0x100)
task_request.validate_request_key(key)
def test_validate_request_key(self):
task_request.validate_request_key(task_pack.unpack_request_key('10'))
task_request.validate_request_key(
ndb.Key('TaskRequestShard',
'a' * task_pack.DEPRECATED_SHARDING_LEVEL, 'TaskRequest',
0x100))
with self.assertRaises(ValueError):
task_request.validate_request_key(ndb.Key('TaskRequest', 1))
with self.assertRaises(ValueError):
key = ndb.Key('TaskRequestShard',
'a' * (task_pack.DEPRECATED_SHARDING_LEVEL + 1),
'TaskRequest', 0x100)
task_request.validate_request_key(key)
def test_validate_request_key(self):
task_request.validate_request_key(task_pack.unpack_request_key('10'))
task_request.validate_request_key(
ndb.Key(
'TaskRequestShard', 'a' * task_pack.DEPRECATED_SHARDING_LEVEL,
'TaskRequest', 0x100))
with self.assertRaises(ValueError):
task_request.validate_request_key(ndb.Key('TaskRequest', 1))
with self.assertRaises(ValueError):
key = ndb.Key(
'TaskRequestShard', 'a' * (task_pack.DEPRECATED_SHARDING_LEVEL + 1),
'TaskRequest', 0x100)
task_request.validate_request_key(key)
def test_validate_request_key(self):
task_request.validate_request_key(task_pack.unpack_request_key('11'))
with self.assertRaises(ValueError):
task_request.validate_request_key(ndb.Key('TaskRequest', 1))