def _create_saved_report(self, domain=None, user_id=None):
config = ReportConfig(domain=domain or self.domain,
owner_id=user_id or self.user._id,
name='Test',
description='Test Saved Report',
report_slug='worker_activity',
report_type='project_report')
config.save()
self.addCleanup(config.delete)
return config
def test_domain_has_shared_configs(self):
config = ReportConfig(domain=self.DOMAIN, owner_id=self.OWNER_ID)
config.save()
self.addCleanup(config.delete)
self._create_scheduled_report(
domain=self.DOMAIN,
owner_id=self.OWNER_ID,
config_ids=[config._id],
)
# Clear cached value
ReportConfig.shared_on_domain.clear(ReportConfig, domain=self.DOMAIN)
configs = list(ReportConfig.shared_on_domain(self.DOMAIN))
self.assertEqual(len(configs), 1)
self.assertEqual(configs[0]._id, config._id)
def test_config_used_in_multiple_report_notifications(self):
config = ReportConfig(domain=self.DOMAIN, owner_id=self.OWNER_ID)
config.save()
self.addCleanup(config.delete)
self._create_scheduled_report(
domain=self.DOMAIN,
owner_id=self.OWNER_ID,
config_ids=[config._id],
)
self._create_scheduled_report(
domain=self.DOMAIN,
owner_id=self.OWNER_ID,
config_ids=[config._id],
)
configs = list(ReportConfig.shared_on_domain(self.DOMAIN))
self.assertEqual(len(configs), 1)
self.assertEqual(configs[0]._id, config._id)
def create_report_config(self, domain, owner_id, **kwargs):
rc = ReportConfig(domain=domain, owner_id=owner_id, **kwargs)
rc.save()
self.addCleanup(rc.delete)
return rc
