def _get_weekly_aggregate_table_definition(cls):
spec = cls.get_monthly_config_spec()
spec.table_id = 'weekly_aggregate_pregnancies'
spec.primary_table.data_source_id = cls.case_data_source._id
spec.secondary_tables[0].data_source_id = cls.form_data_source._id
spec.time_aggregation.unit = AGGREGATION_UNIT_CHOICE_WEEK
return import_aggregation_models_from_spec(spec)
def _get_basic_aggregate_table_definition(cls):
spec = cls.get_basic_config_spec()
spec.primary_table.data_source_id = cls.case_data_source._id
spec.secondary_tables[0].data_source_id = cls.form_data_source._id
spec.secondary_tables[
1].data_source_id = cls.parent_case_data_source._id
return import_aggregation_models_from_spec(spec)
def test_import_from_basic_definition(self):
spec = self.get_monthly_config_spec()
data_source = DataSourceConfiguration(
domain=spec.domain,
referenced_doc_type='CommCareCase',
table_id='some_table',
)
data_source.save()
self.addCleanup(data_source.delete)
# these just have to be valid data source objects
spec.primary_table.data_source_id = data_source._id
spec.secondary_tables[0].data_source_id = data_source._id
aggregate_table_definition = import_aggregation_models_from_spec(spec)
self.assertEqual(1, AggregateTableDefinition.objects.count())
table_def = AggregateTableDefinition.objects.get(pk=aggregate_table_definition.pk)
self.assertEqual(data_source._id, table_def.primary_data_source_id.hex)
self.assertEqual(4, table_def.primary_columns.count())
aggregation = table_def.time_aggregation
self.assertEqual('month', aggregation.aggregation_unit)
self.assertEqual('opened_date', aggregation.start_column)
self.assertEqual('closed_date', aggregation.end_column)
self.assertEqual(1, table_def.secondary_tables.count())
secondary_table = table_def.secondary_tables.get()
self.assertEqual(data_source._id, secondary_table.data_source_id.hex)
self.assertEqual('doc_id', secondary_table.join_column_primary)
self.assertEqual('form.case.@case_id', secondary_table.join_column_secondary)
self.assertEqual('received_on', secondary_table.time_window_column)
self.assertEqual(2, secondary_table.columns.count())
self.assertEqual(
set(['fu_forms_in_month', 'any_fu_forms_in_month']),
set(secondary_table.columns.values_list('column_id', flat=True))
)
def test_import_from_basic_definition(self):
spec = self.get_monthly_config_spec()
data_source = DataSourceConfiguration(
domain=spec.domain,
referenced_doc_type='CommCareCase',
table_id='some_table',
)
data_source.save()
self.addCleanup(data_source.delete)
# these just have to be valid data source objects
spec.primary_table.data_source_id = data_source._id
spec.secondary_tables[0].data_source_id = data_source._id
aggregate_table_definition = import_aggregation_models_from_spec(spec)
self.assertEqual(1, AggregateTableDefinition.objects.count())
table_def = AggregateTableDefinition.objects.get(
pk=aggregate_table_definition.pk)
self.assertEqual(data_source._id, table_def.primary_data_source_id.hex)
self.assertEqual(4, table_def.primary_columns.count())
aggregation = table_def.time_aggregation
self.assertEqual('month', aggregation.aggregation_unit)
self.assertEqual('opened_date', aggregation.start_column)
self.assertEqual('closed_date', aggregation.end_column)
self.assertEqual(1, table_def.secondary_tables.count())
secondary_table = table_def.secondary_tables.get()
self.assertEqual(data_source._id, secondary_table.data_source_id.hex)
self.assertEqual('doc_id', secondary_table.join_column_primary)
self.assertEqual('form.case.@case_id',
secondary_table.join_column_secondary)
self.assertEqual('received_on', secondary_table.time_window_column)
self.assertEqual(1, secondary_table.columns.count())
secondary_column = secondary_table.columns.get()
self.assertEqual('fu_forms_in_month', secondary_column.column_id)
def _get_weekly_aggregate_table_definition(cls):
spec = cls.get_monthly_config_spec()
spec.table_id = 'weekly_aggregate_pregnancies'
spec.primary_table.data_source_id = cls.case_data_source._id
spec.secondary_tables[0].data_source_id = cls.form_data_source._id
spec.time_aggregation.unit = AGGREGATION_UNIT_CHOICE_WEEK
return import_aggregation_models_from_spec(spec)
def _get_basic_aggregate_table_definition(cls):
spec = cls.get_basic_config_spec()
spec.primary_table.data_source_id = cls.case_data_source._id
spec.secondary_tables[0].data_source_id = cls.form_data_source._id
spec.secondary_tables[1].data_source_id = cls.parent_case_data_source._id
return import_aggregation_models_from_spec(spec)
