def test_unpack_request_key(self):
# Old style keys.
self.assertEqual(
ndb.Key('TaskRequestShard', 'f71849', 'TaskRequest', 256),
task_pack.unpack_request_key('10'))
# New style key.
self.assertEqual(ndb.Key('TaskRequest', 0x7fffffffffffffee),
task_pack.unpack_request_key('11'))
with self.assertRaises(ValueError):
task_pack.unpack_request_key('2')
def test_unpack_request_key(self):
# Old style keys.
self.assertEqual(
ndb.Key('TaskRequestShard', 'f71849', 'TaskRequest', 256),
task_pack.unpack_request_key('10'))
# New style key.
self.assertEqual(
ndb.Key('TaskRequest', 0x7fffffffffffffee),
task_pack.unpack_request_key('11'))
with self.assertRaises(ValueError):
task_pack.unpack_request_key('2')
def test_request_key_to_result_summary_key(self):
# New style key.
request_key = task_pack.unpack_request_key('11')
result_key = task_pack.request_key_to_result_summary_key(request_key)
expected = ndb.Key('TaskRequest', 0x7fffffffffffffee,
'TaskResultSummary', 1)
self.assertEqual(expected, result_key)
# Old style key.
request_key = task_pack.unpack_request_key('10')
result_key = task_pack.request_key_to_result_summary_key(request_key)
expected = ndb.Key('TaskRequestShard', 'f71849', 'TaskRequest', 256,
'TaskResultSummary', 1)
self.assertEqual(expected, result_key)
def test_result_summary_key_to_request_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
actual = task_pack.result_summary_key_to_request_key(
result_summary_key)
self.assertEqual(request_key, actual)
def test_request_key_to_result_summary_key(self):
# New style key.
request_key = task_pack.unpack_request_key('11')
result_key = task_pack.request_key_to_result_summary_key(
request_key)
expected = ndb.Key(
'TaskRequest', 0x7fffffffffffffee, 'TaskResultSummary', 1)
self.assertEqual(expected, result_key)
# Old style key.
request_key = task_pack.unpack_request_key('10')
result_key = task_pack.request_key_to_result_summary_key(
request_key)
expected = ndb.Key(
'TaskRequestShard', 'f71849', 'TaskRequest', 256,
'TaskResultSummary', 1)
self.assertEqual(expected, result_key)
def test_run_result_key_to_result_summary_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
self.assertEqual(
result_summary_key,
task_pack.run_result_key_to_result_summary_key(run_result_key))
def test_run_result_key_to_performance_stats_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
perf_stats_key = task_pack.run_result_key_to_performance_stats_key(
run_result_key)
self.assertEqual('PerformanceStats',perf_stats_key.kind())
def test_run_result_key_to_result_summary_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
self.assertEqual(
result_summary_key,
task_pack.run_result_key_to_result_summary_key(run_result_key))
def test_run_result_key_to_performance_stats_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
perf_stats_key = task_pack.run_result_key_to_performance_stats_key(
run_result_key)
self.assertEqual('PerformanceStats', perf_stats_key.kind())
def test_pack_run_result_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
self.assertEqual('111', task_pack.pack_run_result_key(run_result_key))
with self.assertRaises(AssertionError):
task_pack.pack_run_result_key(result_summary_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_pack_run_result_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
self.assertEqual('111', task_pack.pack_run_result_key(run_result_key))
with self.assertRaises(AssertionError):
task_pack.pack_run_result_key(result_summary_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_result_summary_key_to_run_result_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
expected = ndb.Key('TaskRequest', 0x7fffffffffffffee, 'TaskResultSummary',
1, 'TaskRunResult', 1)
self.assertEqual(expected, run_result_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 2)
expected = ndb.Key(
'TaskRequest', 0x7fffffffffffffee, 'TaskResultSummary', 1,
'TaskRunResult', 2)
self.assertEqual(expected, run_result_key)
with self.assertRaises(ValueError):
task_pack.result_summary_key_to_run_result_key(result_summary_key, 0)
with self.assertRaises(ValueError):
task_pack.result_summary_key_to_run_result_key(result_summary_key, 3)
def test_result_summary_key_to_run_result_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 1)
expected = ndb.Key(
'TaskRequest', 0x7fffffffffffffee, 'TaskResultSummary', 1,
'TaskRunResult', 1)
self.assertEqual(expected, run_result_key)
run_result_key = task_pack.result_summary_key_to_run_result_key(
result_summary_key, 2)
expected = ndb.Key(
'TaskRequest', 0x7fffffffffffffee, 'TaskResultSummary', 1,
'TaskRunResult', 2)
self.assertEqual(expected, run_result_key)
with self.assertRaises(ValueError):
task_pack.result_summary_key_to_run_result_key(result_summary_key, 0)
with self.assertRaises(NotImplementedError):
task_pack.result_summary_key_to_run_result_key(result_summary_key, 3)
def test_result_summary_key_to_request_key(self):
request_key = task_pack.unpack_request_key('11')
result_summary_key = task_pack.request_key_to_result_summary_key(
request_key)
actual = task_pack.result_summary_key_to_request_key(result_summary_key)
self.assertEqual(request_key, actual)
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))
def test_unpack_request_key(self):
self.assertEqual(ndb.Key('TaskRequest', 0x7fffffffffffffee),
task_pack.unpack_request_key('11'))
with self.assertRaises(ValueError):
task_pack.unpack_request_key('2')
def test_request_key_to_result_summary_key(self):
request_key = task_pack.unpack_request_key('11')
result_key = task_pack.request_key_to_result_summary_key(request_key)
expected = ndb.Key('TaskRequest', 0x7fffffffffffffee,
'TaskResultSummary', 1)
self.assertEqual(expected, result_key)
def test_request_key_to_secret_bytes_key(self):
request_key = task_pack.unpack_request_key('11')
result_key = task_pack.request_key_to_secret_bytes_key(request_key)
expected = ndb.Key('TaskRequest', 0x7fffffffffffffee, 'SecretBytes', 1)
self.assertEqual(expected, result_key)
