Ejemplo n.º 1
0
 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')
Ejemplo n.º 2
0
 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')
Ejemplo n.º 3
0
 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)
Ejemplo n.º 4
0
 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)
Ejemplo n.º 5
0
 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)
Ejemplo n.º 6
0
 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))
Ejemplo n.º 7
0
 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())
Ejemplo n.º 8
0
 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))
Ejemplo n.º 9
0
 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())
Ejemplo n.º 10
0
    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)
Ejemplo n.º 11
0
 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)
Ejemplo n.º 12
0
  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)
Ejemplo n.º 13
0
 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)
Ejemplo n.º 14
0
 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)
Ejemplo n.º 15
0
  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)
Ejemplo n.º 16
0
  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)
Ejemplo n.º 17
0
 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)
Ejemplo n.º 18
0
 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))
Ejemplo n.º 19
0
 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')
Ejemplo n.º 20
0
 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)
Ejemplo n.º 21
0
 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)