flask-base-api/tests/test_event_descriptor_model.py at master · sammyrails/flask-base-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
# project/tests/test_event_descriptor_model.py

from sqlalchemy.exc import IntegrityError
from project import db
from project.models.event import Event
from project.models.event_descriptor import EventDescriptor
from tests.base import BaseTestCase


class TestEventDescriptorModel(BaseTestCase):

    def test_add_event_descriptor(self):
        event_descriptor = EventDescriptor(id = 1, name="event_name", description="event_description")
        db.session.add(event_descriptor)
        db.session.commit()

        self.assertTrue(event_descriptor.id)
        self.assertEqual(event_descriptor.name, 'event_name')
        self.assertEqual(event_descriptor.description, 'event_description')

    def test_event_descriptor_relations(self):
        event_descriptor = EventDescriptor(id = 1, name="event_name", description="event_description")
        db.session.add(event_descriptor)
        db.session.commit()

        event = Event(event_descriptor_id=33)
        db.session.add(event)
        self.assertRaises(IntegrityError, db.session.commit)
        db.session.rollback()

        event.event_descriptor_id = event_descriptor.id
        db.session.add(event)
        db.session.commit()

        self.assertTrue(event.id)
        self.assertTrue(event.created_at)
        self.assertTrue(event.updated_at)

        self.assertEqual(event.event_descriptor.id, event_descriptor.id)

        event = Event(event_descriptor_id=event_descriptor.id)
        db.session.add(event)
        db.session.commit()

        self.assertTrue(event.id)
        self.assertTrue(event.created_at)
        self.assertTrue(event.updated_at)

        self.assertEqual(len(event_descriptor.events), 2)