def test_key_comparator_with_single_path(self):
k1 = Key()
k2 = Key()
p1 = k1.path.add()
p2 = k2.path.add()
p1.kind = p2.kind = 'dummy'
self.assertEqual(helper.key_comparator(k1, k2), 0)
def test_key_comparator_with_multiple_paths_2(self):
k1 = Key()
k2 = Key()
p11 = k1.path.add()
p21 = k2.path.add()
p22 = k2.path.add()
p11.kind = p21.kind = p22.kind = 'dummy'
self.assertLess(helper.key_comparator(k1, k2), 0)
def test_compare_path_with_different_kind(self):
p1 = Key.PathElement()
p1.kind = 'dummy1'
p2 = Key.PathElement()
p2.kind = 'dummy2'
self.assertLess(helper.compare_path(p1, p2), 0)
def test_key_comparator_with_multiple_paths_4(self):
k1 = Key()
k2 = Key()
p11 = k1.path.add()
p12 = k2.path.add()
p21 = k2.path.add()
p11.kind = p12.kind = 'dummy'
# make path2 greater than path1
p21.kind = 'dummy1'
self.assertLess(helper.key_comparator(k1, k2), 0)
def test_compare_path_of_different_type(self):
p1 = Key.PathElement()
p1.kind = 'dummy'
p1.id = 10
p2 = Key.PathElement()
p2.kind = 'dummy'
p2.name = 'dummy'
self.assertLess(helper.compare_path(p1, p2), 0)
def test_key_comparator_with_different_partition(self):
k1 = Key()
k1.partition_id.namespace_id = 'dummy1'
k2 = Key()
k2.partition_id.namespace_id = 'dummy2'
self.assertRaises(ValueError, helper.key_comparator, k1, k2)
