Exemplo n.º 1
0
    def testCreateReport(self):
        provider = Provider.objects.get(id=2)
        clinic = provider.clinic

        report = ReportMalaria()
        report.provider = provider
        recipients = report.get_alert_recipients()
        assert len(recipients) == 3, [r.id for r in recipients]

        one = Provider.objects.get(id=1)
        one.alerts = False
        one.save()

        recipients = report.get_alert_recipients()
        assert len(recipients) == 2

        one = Provider.objects.get(id=1)
        one.alerts = True
        one.save()

        recipients = report.get_alert_recipients()
        assert len(recipients) == 3

        four = Provider.objects.get(id=4)
        four.following_users.remove(provider)
        four.save()

        recipients = report.get_alert_recipients()
        assert len(recipients) == 2

        one.following_clinics.remove(clinic)
        one.save()

        recipients = report.get_alert_recipients()
        assert len(recipients) == 1, [r.id for r in recipients]