def test_source_model_calling_a_filter(self):
sm_filter = SourceModelCatalogFilter()
sm_filter.filter_eqs = Mock()
sm_filter.filter_eqs.return_value = []
catalog_filter = CatalogFilter(sm_filter)
gen = catalog_filter.filter_eqs([dict(a=1), dict(b=2)], [])
self.assertEqual((dict(a=1), []), gen.next())
self.assertEqual((dict(b=2), []), gen.next())
self.assertEqual(dict(a=1), sm_filter.filter_eqs.call_args_list[0][0][0])
self.assertEqual(dict(b=2), sm_filter.filter_eqs.call_args_list[1][0][0])
def test_source_model_calling_a_filter(self):
sm_filter = SourceModelCatalogFilter()
sm_filter.filter_eqs = Mock()
sm_filter.filter_eqs.return_value = []
catalog_filter = CatalogFilter(sm_filter)
gen = catalog_filter.filter_eqs([dict(a=1), dict(b=2)], [])
self.assertEqual((dict(a=1), []), gen.next())
self.assertEqual((dict(b=2), []), gen.next())
self.assertEqual(dict(a=1),
sm_filter.filter_eqs.call_args_list[0][0][0])
self.assertEqual(dict(b=2),
sm_filter.filter_eqs.call_args_list[1][0][0])
def test_empty_source_model(self):
catalog_filter = CatalogFilter(None)
self.assertRaises(StopIteration, catalog_filter.filter_eqs([], []).next)
def test_empty_source_model(self):
catalog_filter = CatalogFilter(None)
self.assertRaises(StopIteration,
catalog_filter.filter_eqs([], []).next)
