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))")
