def testSaveCountShouldPassThroughFieldPrefix(self, mock_save): from stockandflow.models import Facet f = Facet("test_slug", "test name", "test_field", [1,2]) f.to_count = MagicMock() s = Stock(*self.stock_args) s.facets = [(f, "test_prefix")] s.save_count() self.assertEqual(f.to_count.call_args, (("test_prefix",), {}))
def testCreateAGeneratorOfQObjectsForAllValuesWithAFieldPrefix(self): from stockandflow.models import Facet f = Facet("test_slug", "test name", "test_field", [1,2]) q_gen = f.to_count("yada") rv = q_gen.next() self.assertEqual(rv[0], 1) self.assertEqual(str(rv[1]), "(AND: ('yada__test_field', 1))") rv = q_gen.next() self.assertEqual(rv[0], 2) self.assertEqual(str(rv[1]), "(AND: ('yada__test_field', 2))")