def testGettingAttrValues(self): d1 = Driver('d1') d2 = Driver('d2') d1.add_attr('foo', 'bar') d1.add_attr('foo0', 'bar1') d2.add_attr('d1', d1) clusto.flush() self.assertEqual(sorted(['bar', 'bar1']), sorted(d1.attr_values('foo.*', regex=True))) self.assertEqual([d1], d2.attr_values())
def testParents(self): class OtherDriver(Driver): _clusto_type = 'otherdriver' _driver_name = 'otherdriver' p1 = Pool('toppool') d1 = Pool('grandparent') d1a = Pool('othergrandparent') d1b = OtherDriver('anothergrandparent') d2 = Driver('parent') d3 = Driver('child') d1b.add_attr('foo', 'someval') p1.insert(d1a) d1b.insert(d2) d1a.insert(d2) d1.insert(d2) d2.insert(d3) self.assertEqual( sorted([d1, d1a, p1]), sorted(d3.parents(clusto_types=[Pool], search_parents=True))) self.assertEqual([d2], d3.parents()) self.assertEqual(['someval'], d3.attr_values('foo', merge_container_attrs=True))
def testParents(self): class OtherDriver(Driver): _clusto_type = 'otherdriver' _driver_name = 'otherdriver' p1 = Pool('toppool') d1 = Pool('grandparent') d1a = Pool('othergrandparent') d1b = OtherDriver('anothergrandparent') d2 = Driver('parent') d3 = Driver('child') d1b.add_attr('foo', 'someval') p1.insert(d1a) d1b.insert(d2) d1a.insert(d2) d1.insert(d2) d2.insert(d3) self.assertEqual(sorted([d1, d1a, p1]), sorted(d3.parents(clusto_types=[Pool], search_parents=True))) self.assertEqual([d2], d3.parents()) self.assertEqual(['someval'], d3.attr_values('foo', merge_container_attrs=True))