def _create_report(self, domain=None, owner_id=None): domain = domain or self.domain owner_id = owner_id or self.user._id report = ReportNotification(domain=domain, owner_id=owner_id) report.save() self.addCleanup(report.delete) return report
def _create_scheduled_report(self, domain=None, owner_id=None, config_ids=[]): report = ReportNotification(domain=domain, owner_id=owner_id, config_ids=config_ids) report.save() self.addCleanup(report.delete) return report
def create_report_notification(self, configs, owner_id): domain = configs[0].domain config_ids = [c._id for c in configs] rn = ReportNotification( domain=domain, config_ids=config_ids, owner_id=owner_id, interval='daily', ) rn.save() self.addCleanup(rn.delete) return rn
def test_get_scheduled_report_response(self): domain = self.domain report_config = ReportConfig.wrap({ "date_range": "last30", "days": 30, "domain": domain, "report_slug": "worker_activity", "report_type": "project_report", "owner_id": self.user._id, }) report_config.save() report = ReportNotification( hour=12, minute=None, day=30, interval='monthly', config_ids=[report_config._id] ) report.save() response = get_scheduled_report_response( couch_user=self.user, domain=domain, scheduled_report_id=report._id )[0] self.assertTrue(self.user.username in response.decode('utf-8'))
def test_get_scheduled_report_response(self): domain = self.domain report_config = ReportConfig.wrap({ "date_range": "last30", "days": 30, "domain": domain, "report_slug": "worker_activity", "report_type": "project_report", "owner_id": self.user._id, }) report_config.save() report = ReportNotification(hour=12, minute=None, day=30, interval='monthly', config_ids=[report_config._id]) report.save() response = get_scheduled_report_response( couch_user=self.user, domain=domain, scheduled_report_id=report._id)[0] self.assertTrue(self.user.username in response.decode('utf-8'))