def test_search(self):
"""Test the search utility."""
def is_venture2(ci):
return ci.uid == 'br-2'
self.assertEqual(breadth_first_search_ci(self.server_b, is_venture2),
(self.venture2, True))
self.assertEqual(breadth_first_search_ci(self.server_a, is_venture2),
(None, None))
def test_search(self):
"""Test the search utility."""
def is_venture2(ci):
return ci.uid == 'br-2'
self.assertEqual(
breadth_first_search_ci(self.server_b, is_venture2),
(self.venture2, True)
)
self.assertEqual(
breadth_first_search_ci(self.server_a, is_venture2),
(None, None)
)
def dehydrate(self, bundle, **kwargs):
def get_owners(ci):
owners = CIOwner.objects.filter(ciownership__type=self.owner_type, ciownership__ci=ci)
result = []
for owner in owners:
result.append(self.dehydrate_related(bundle, self.get_related_resource(owner)))
return result
if self.effective:
found, result = breadth_first_search_ci(bundle.obj, get_owners)
return result
else:
return get_owners(bundle.obj)
def dehydrate(self, bundle, **kwargs):
def get_owners(ci):
owners = CIOwner.objects.filter(
ciownership__type=self.owner_type,
ciownership__ci=ci,
)
result = []
for owner in owners:
result.append(
self.dehydrate_related(bundle,
self.get_related_resource(owner)))
return result
if self.effective:
found, result = breadth_first_search_ci(bundle.obj, get_owners)
return result
else:
return get_owners(bundle.obj)
