File size: 3,425 Bytes
550665c |
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 |
from unittest import TestCase
from gcsa.attendee import Attendee, ResponseStatus
from gcsa.serializers.attendee_serializer import AttendeeSerializer
class TestAttendee(TestCase):
def test_repr_str(self):
attendee = Attendee(
email='[email protected]',
display_name='Guest',
comment='I do not know him',
optional=True,
additional_guests=2,
_response_status=ResponseStatus.NEEDS_ACTION
)
self.assertEqual(attendee.__repr__(), "<Attendee '[email protected]' - response: 'needsAction'>")
self.assertEqual(attendee.__str__(), "'[email protected]' - response: 'needsAction'")
class TestAttendeeSerializer(TestCase):
def test_to_json(self):
attendee = Attendee(
email='[email protected]',
display_name='Guest',
comment='I do not know him',
optional=True,
additional_guests=2,
_response_status=ResponseStatus.NEEDS_ACTION
)
attendee_json = AttendeeSerializer.to_json(attendee)
self.assertEqual(attendee.email, attendee_json['email'])
self.assertEqual(attendee.display_name, attendee_json['displayName'])
self.assertEqual(attendee.comment, attendee_json['comment'])
self.assertEqual(attendee.optional, attendee_json['optional'])
self.assertNotIn('resource', attendee_json)
self.assertEqual(attendee.additional_guests, attendee_json['additionalGuests'])
self.assertEqual(attendee.response_status, attendee_json['responseStatus'])
def test_to_object(self):
attendee_json = {
'email': '[email protected]',
'displayName': 'Guest2',
'comment': 'I do not know him either',
'optional': True,
'resource': True,
'additionalGuests': 1,
'responseStatus': ResponseStatus.ACCEPTED
}
attendee = AttendeeSerializer.to_object(attendee_json)
self.assertEqual(attendee_json['email'], attendee.email)
self.assertEqual(attendee_json['displayName'], attendee.display_name)
self.assertEqual(attendee_json['comment'], attendee.comment)
self.assertEqual(attendee_json['optional'], attendee.optional)
self.assertEqual(attendee_json['resource'], attendee.is_resource)
self.assertEqual(attendee_json['additionalGuests'], attendee.additional_guests)
self.assertEqual(attendee_json['responseStatus'], attendee.response_status)
attendee_json_str = """{
"email": "[email protected]",
"displayName": "Guest3",
"comment": "Who are these people?",
"optional": true,
"resource": false,
"additionalGuests": 66,
"responseStatus": "tentative"
}"""
serializer = AttendeeSerializer(attendee_json_str)
attendee = serializer.get_object()
self.assertEqual(attendee.email, "[email protected]")
self.assertEqual(attendee.display_name, "Guest3")
self.assertEqual(attendee.comment, "Who are these people?")
self.assertEqual(attendee.optional, True)
self.assertEqual(attendee.is_resource, False)
self.assertEqual(attendee.additional_guests, 66)
self.assertEqual(attendee.response_status, "tentative")
|