go-api/alert_system/factories.py at 903cbfd911293e38922fd0d868802fef109de976 · 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
import factory

from alert_system.models import AlertEmailLog, AlertEmailThread, Connector, LoadItem


class LoadItemFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = LoadItem

    @factory.post_generation
    def related_montandon_events(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            for event in extracted:
                self.related_montandon_events.add(event)

    @factory.post_generation
    def related_go_events(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            for event in extracted:
                self.related_go_events.add(event)


class AlertEmailLogFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = AlertEmailLog


class ConnectorFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Connector


class AlertEmailThreadFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = AlertEmailThread