def testGenerateAListOfFacetsSortedBySlugBasedOnTheStocks(self):
from stockandflow.models import Stock, Facet
from stockandflow.views import Process
f1 = Facet("test_slug", "test name", "test_field", [1,2])
f2 = Facet("test_slug", "test name", "test_field", [1,2])
self.stock_args = []
s1 = Stock('test name 1', 'test_slug_1', Mock())
s2 = Stock('test name 2', 'test_slug_2', Mock())
s1.facets = [f1]
s2.facets = [f1, f2]
process = Process("process_test", "process test", [s1, s2])
self.assertEqual(process.facets, [f1, f2])
def testSaveCountShouldCreatesStockFacetRecord(self, mock_save):
from stockandflow.models import Facet
f = Facet("test_slug", "test name", "test_field", [1,2])
s = Stock(*self.stock_args)
s.facets = [f]
s.save_count()
self.assertEqual(mock_save.call_count, 2)
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",), {}))
