go-api/api/test_models.py at dc9ccc06502727223dfb971fb3bab84eac907204 · IFRCGo/go-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
from unittest.mock import patch

from django.contrib.auth.models import User
from django.test import TestCase
from rest_framework.test import APITestCase

import api.models as models
from api.factories import country as countryFactory
from api.factories import event as eventFactory
from api.factories import field_report as fieldReportFactory
from main.mock import erp_request_side_effect_mock


class DisasterTypeTest(TestCase):

    fixtures = ["DisasterTypes"]

    def test_disaster_type_data(self):
        objs = models.DisasterType.objects.all()
        self.assertEqual(len(objs), 24)


class EventTest(TestCase):

    fixtures = ["DisasterTypes"]

    def setUp(self):
        dtype = models.DisasterType.objects.get(pk=1)
        models.Event.objects.create(title="disaster1", summary="test disaster", dtype=dtype)
        event = models.Event.objects.create(title="disaster2", summary="another test disaster", dtype=dtype)
        models.KeyFigure.objects.create(event=event, number=7, deck="things", source="website")
        models.Snippet.objects.create(event=event, snippet="this is a snippet")

    def test_disaster_create(self):
        obj1 = models.Event.objects.get(title="disaster1")
        obj2 = models.Event.objects.get(title="disaster2")
        self.assertEqual(obj1.summary, "test disaster")
        self.assertEqual(obj2.summary, "another test disaster")
        keyfig = obj2.key_figures.all()
        self.assertEqual(keyfig[0].deck, "things")
        self.assertEqual(keyfig[0].number, "7")
        snippet = obj2.snippets.all()
        self.assertEqual(snippet[0].snippet, "this is a snippet")


class CountryTest(TestCase):

    fixtures = ["Regions", "Countries"]

    def test_country_data(self):
        regions = models.Region.objects.all()
        self.assertEqual(regions.count(), 5)
        countries = models.Country.objects.all()
        self.assertEqual(countries.count(), 274)


class ProfileTest(TestCase):
    def setUp(self):
        user = User.objects.create(username="username", first_name="pat", last_name="smith", password="password")
        user.profile.org = "org"
        user.profile.save()

    def test_profile_create(self):
        profile = models.Profile.objects.get(user__username="username")
        self.assertEqual(profile.org, "org")


class AppealTest(APITestCase):
    def setUp(self):
        # An appeal with needs_confirmation=True should not return the event in the API response.
        event = models.Event.objects.create(title="associated event", summary="foo")
        country = models.Country.objects.create(name="country")
        models.Appeal.objects.create(
            aid="test1", name="appeal", atype=1, code="abc", needs_confirmation=True, event=event, country=country
        )

    def test_unconfirmed_event(self):
        response = self.client.get("/api/v2/appeal/?code=abc")
        response = dict(dict(response.data)["results"][0])
        self.assertIsNone(response["event"])
        self.assertIsNotNone(response["country"])


class FieldReportTest(TestCase):

    fixtures = ["DisasterTypes"]

    def setUp(self):
        dtype = models.DisasterType.objects.get(pk=1)
        event = models.Event.objects.create(title="disaster1", summary="test disaster", dtype=dtype)
        country = models.Country.objects.create(name="country")
        report = models.FieldReport.objects.create(rid="test1", event=event, dtype=dtype)
        report.countries.add(country)

    def test_field_report_create(self):
        event = models.Event.objects.get(title="disaster1")
        country = models.Country.objects.get(name="country")
        self.assertEqual(event.field_reports.all()[0].countries.all()[0], country)
        obj = models.FieldReport.objects.get(rid="test1")
        self.assertEqual(obj.rid, "test1")
        self.assertEqual(obj.countries.all()[0], country)
        self.assertEqual(obj.event, event)
        self.assertIsNotNone(obj.report_date)

    @patch("requests.post", side_effect=erp_request_side_effect_mock)
    def test_ERP_related_field_report(self, mocked_requests_post):
        dtype = models.DisasterType.objects.get(pk=1)
        country = countryFactory.CountryFactory()
        event = eventFactory.EventFactory(name="disaster2", summary="test disaster 2", dtype=dtype)
        event.countries.set([country])
        report = fieldReportFactory.FieldReportFactory.create(
            rid="test2",
            event=event,
            dtype=dtype,
            ns_request_assistance=True,
        )
        ERP = models.ERPGUID.objects.get(api_guid="FindThisGUID")
        self.assertEqual(ERP.field_report_id, report.id)
        self.assertEqual(mocked_requests_post.called, True)


class ProfileTestDepartment(TestCase):
    def setUp(self):
        user = User.objects.create(username="test1", password="12345678!")
        user.profile.department = "testdepartment"
        user.save()

    def test_profile_create(self):
        obj = models.Profile.objects.get(user__username="test1")
        self.assertEqual(obj.department, "testdepartment")