rsvp-lib/test_rsvp.py at master · pythonpune/rsvp-lib · 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
# test cases for rsvp
import os
import unittest

from datetime import date
from uuid import UUID

from rsvp.constants import MEMBER_KEYS
import rsvp


class TestRSVP(unittest.TestCase):
    """
    RSVP Test Cases
    """

    def setUp(self):
        os.environ["RSVP_TEST_CONFIG"] = 'true'
        self.rsvp_object = rsvp

        event_source = 'PythonPune MeetUp'
        self.new_event_rsvp = self.rsvp_object.create_rsvp_store(
            event_source, **{
                'event_slug': 'python-pune-meetup',
                'event_start_date': date.today()
            }
        )

    def tearDown(self):
        del os.environ["RSVP_TEST_CONFIG"]
        self.rsvp_object.delete_rsvp_store(self.new_event_rsvp)
        del self.rsvp_object

    def test_create_delete_rsvp_store(self):
        """
        test create_rsvp store
        """
        self.assertIsInstance(UUID(self.new_event_rsvp), UUID)

    def test_add_members(self):
        """
        test add_members
        """
        rsvp_members = ['Rahul', 'Shyam', 'Anu', 'Isha']

        add_result = self.rsvp_object.add_members(self.new_event_rsvp, rsvp_members)
        self.assertTrue(add_result)

    def test_adjust_rsvp_state(self):
        """
        test adjust_rsvp_state
        """
        rsvp_members = ['Rahul', 'Shyam', 'Anu', 'Isha']
        add_result = self.rsvp_object.add_members(self.new_event_rsvp, rsvp_members)

        if add_result:
            rsvp_members_to_update = ['Rahul', 'Shyam']
            state1 = {MEMBER_KEYS[0]: True}

            update_result_1 = self.rsvp_object.adjust_rsvp_state(
                self.new_event_rsvp, rsvp_members_to_update, **state1
            )
            self.assertTrue(update_result_1)

            state2 = {MEMBER_KEYS[1]: True}
            update_result_2 = self.rsvp_object.adjust_rsvp_state(
                self.new_event_rsvp, rsvp_members_to_update, **state2
            )
            self.assertTrue(update_result_2)

    def test_get_rsvp_state(self):
        """
        test get_rsvp_state
        """
        rsvp_members = ['Rahul', 'Shyam', 'Anu', 'Isha']
        add_result = self.rsvp_object.add_members(self.new_event_rsvp, rsvp_members)

        if add_result:
            rsvp_members_to_update = ['Rahul', 'Shyam']
            state1 = {MEMBER_KEYS[0]: True, MEMBER_KEYS[1]: True, MEMBER_KEYS[2]: True}

            update_result_1 = self.rsvp_object.adjust_rsvp_state(
                self.new_event_rsvp, rsvp_members_to_update, **state1
            )
            self.assertTrue(update_result_1)

            result = self.rsvp_object.get_rsvp_state(
                self.new_event_rsvp, [rsvp_members_to_update[0]]
            )
            self.assertIsInstance(result, list)
            self.assertDictEqual(
                result[0], {'Rahul': {'confirm_for_seat': True,
                                      'attended_event': True, 'in_queue_waiting': True}}
            )


def rsvp_test_suit():
    """
    Test Suit
    """
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(TestRSVP))
    return suite