def test_simple_run(self):
agg_instance = ModelAggregator(TestModel)
agg_instance.run()
self.assertEqual(kaggregate.get_aggregate_value("count"), 10)
self.assertEqual(kaggregate.get_aggregate_value("sum"), 45)
self.assertEqual(kaggregate.get_aggregate_value("average"), 4.5)
def test_simple_run_with_prefix(self):
agg_instance = TestAggregator()
agg_instance.run("foo_")
self.assertEqual(kaggregate.get_aggregate_value("foo_count"), 10)
self.assertEqual(kaggregate.get_aggregate_value("foo_sum"), 45)
self.assertEqual(kaggregate.get_aggregate_value("average"), None)
self.assertEqual(kaggregate.get_aggregate_value("count"), None)
def test_management_command(self):
management.call_command('gen_aggregates')
self.assertEqual(kaggregate.get_aggregate_value("count"), 10)
self.assertEqual(kaggregate.get_aggregate_value("sum"), 45)
self.assertEqual(kaggregate.get_aggregate_value("bar_count"), 10)
self.assertEqual(kaggregate.get_aggregate_value("bar_sum"), 45)
self.assertEqual(kaggregate.get_aggregate_value("foo-count"), 10)
self.assertEqual(kaggregate.get_aggregate_value("foo-sum"), 45)
self.assertEqual(kaggregate.get_aggregate_value("foo-average"), 4.5)
def test_simple_run(self):
agg_instance = TestAggregator()
agg_instance.run()
self.assertEqual(kaggregate.get_aggregate_value("count"), 10)
self.assertEqual(kaggregate.get_aggregate_value("sum"), 45)
