|
1 | 1 | # Copyright The IETF Trust 2015-2024, All Rights Reserved
|
2 | 2 | # -*- coding: utf-8 -*-
|
3 | 3 | import base64
|
| 4 | +import copy |
4 | 5 | import datetime
|
5 | 6 | import json
|
6 | 7 | import html
|
|
31 | 32 | from ietf.doc.factories import IndividualDraftFactory, WgDraftFactory, WgRfcFactory
|
32 | 33 | from ietf.group.factories import RoleFactory
|
33 | 34 | from ietf.meeting.factories import MeetingFactory, SessionFactory
|
34 |
| -from ietf.meeting.models import Session |
| 35 | +from ietf.meeting.models import Session, Registration |
35 | 36 | from ietf.nomcom.models import Volunteer
|
36 | 37 | from ietf.nomcom.factories import NomComFactory, nomcom_kwargs_for_year
|
37 | 38 | from ietf.person.factories import PersonFactory, random_faker, EmailFactory, PersonalApiKeyFactory
|
@@ -828,6 +829,196 @@ def test_api_new_meeting_registration_nomcom_volunteer(self):
|
828 | 829 | self.assertEqual(volunteer.nomcom, nomcom)
|
829 | 830 | self.assertEqual(volunteer.origin, 'registration')
|
830 | 831 |
|
| 832 | + @override_settings(APP_API_TOKENS={"ietf.api.views.api_new_meeting_registration_v2": ["valid-token"]}) |
| 833 | + def test_api_new_meeting_registration_v2(self): |
| 834 | + meeting = MeetingFactory(type_id='ietf') |
| 835 | + person = PersonFactory() |
| 836 | + regs = [ |
| 837 | + { |
| 838 | + 'affiliation': "Alguma Corporação", |
| 839 | + 'country_code': 'PT', |
| 840 | + 'email': person.email().address, |
| 841 | + 'first_name': person.first_name(), |
| 842 | + 'last_name': person.last_name(), |
| 843 | + 'meeting': str(meeting.number), |
| 844 | + 'reg_type': 'onsite', |
| 845 | + 'ticket_type': 'week_pass', |
| 846 | + 'checkedin': False, |
| 847 | + 'is_nomcom_volunteer': False, |
| 848 | + 'cancelled': False, |
| 849 | + } |
| 850 | + ] |
| 851 | + |
| 852 | + url = urlreverse('ietf.api.views.api_new_meeting_registration_v2') |
| 853 | + # |
| 854 | + # Test invalid key |
| 855 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "invalid-token"}) |
| 856 | + self.assertEqual(r.status_code, 403) |
| 857 | + # |
| 858 | + # Test invalid data |
| 859 | + bad_regs = copy.deepcopy(regs) |
| 860 | + del(bad_regs[0]['email']) |
| 861 | + r = self.client.post(url, data=json.dumps(bad_regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 862 | + self.assertEqual(r.status_code, 400) |
| 863 | + # |
| 864 | + # Test valid POST |
| 865 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 866 | + self.assertContains(r, "Success", status_code=202) |
| 867 | + # |
| 868 | + # Check record |
| 869 | + reg = regs[0] |
| 870 | + objects = Registration.objects.filter(email=reg['email'], meeting__number=reg['meeting']) |
| 871 | + self.assertEqual(objects.count(), 1) |
| 872 | + obj = objects[0] |
| 873 | + for key in ['affiliation', 'country_code', 'first_name', 'last_name', 'checkedin']: |
| 874 | + self.assertEqual(getattr(obj, key), False if key=='checkedin' else reg.get(key) , "Bad data for field '%s'" % key) |
| 875 | + self.assertEqual(obj.tickets.count(), 1) |
| 876 | + ticket = obj.tickets.first() |
| 877 | + self.assertEqual(ticket.ticket_type.slug, regs[0]['ticket_type']) |
| 878 | + self.assertEqual(ticket.attendance_type.slug, regs[0]['reg_type']) |
| 879 | + self.assertEqual(obj.person, person) |
| 880 | + # |
| 881 | + # Test update (switch to remote) |
| 882 | + regs = [ |
| 883 | + { |
| 884 | + 'affiliation': "Alguma Corporação", |
| 885 | + 'country_code': 'PT', |
| 886 | + 'email': person.email().address, |
| 887 | + 'first_name': person.first_name(), |
| 888 | + 'last_name': person.last_name(), |
| 889 | + 'meeting': str(meeting.number), |
| 890 | + 'reg_type': 'remote', |
| 891 | + 'ticket_type': 'week_pass', |
| 892 | + 'checkedin': False, |
| 893 | + 'is_nomcom_volunteer': False, |
| 894 | + 'cancelled': False, |
| 895 | + } |
| 896 | + ] |
| 897 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 898 | + self.assertContains(r, "Success", status_code=202) |
| 899 | + objects = Registration.objects.filter(email=reg['email'], meeting__number=reg['meeting']) |
| 900 | + self.assertEqual(objects.count(), 1) |
| 901 | + obj = objects[0] |
| 902 | + self.assertEqual(obj.tickets.count(), 1) |
| 903 | + ticket = obj.tickets.first() |
| 904 | + self.assertEqual(ticket.ticket_type.slug, regs[0]['ticket_type']) |
| 905 | + self.assertEqual(ticket.attendance_type.slug, regs[0]['reg_type']) |
| 906 | + # |
| 907 | + # Test multiple |
| 908 | + regs = [ |
| 909 | + { |
| 910 | + 'affiliation': "Alguma Corporação", |
| 911 | + 'country_code': 'PT', |
| 912 | + 'email': person.email().address, |
| 913 | + 'first_name': person.first_name(), |
| 914 | + 'last_name': person.last_name(), |
| 915 | + 'meeting': str(meeting.number), |
| 916 | + 'reg_type': 'onsite', |
| 917 | + 'ticket_type': 'one_day', |
| 918 | + 'checkedin': False, |
| 919 | + 'is_nomcom_volunteer': False, |
| 920 | + 'cancelled': False, |
| 921 | + }, |
| 922 | + |
| 923 | + { |
| 924 | + 'affiliation': "Alguma Corporação", |
| 925 | + 'country_code': 'PT', |
| 926 | + 'email': person.email().address, |
| 927 | + 'first_name': person.first_name(), |
| 928 | + 'last_name': person.last_name(), |
| 929 | + 'meeting': str(meeting.number), |
| 930 | + 'reg_type': 'remote', |
| 931 | + 'ticket_type': 'week_pass', |
| 932 | + 'checkedin': False, |
| 933 | + 'is_nomcom_volunteer': False, |
| 934 | + 'cancelled': False, |
| 935 | + } |
| 936 | + ] |
| 937 | + |
| 938 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 939 | + self.assertContains(r, "Success", status_code=202) |
| 940 | + objects = Registration.objects.filter(email=reg['email'], meeting__number=reg['meeting']) |
| 941 | + self.assertEqual(objects.count(), 1) |
| 942 | + obj = objects[0] |
| 943 | + self.assertEqual(obj.tickets.count(), 2) |
| 944 | + self.assertEqual(obj.tickets.filter(attendance_type__slug='onsite').count(), 1) |
| 945 | + self.assertEqual(obj.tickets.filter(attendance_type__slug='remote').count(), 1) |
| 946 | + |
| 947 | + @override_settings(APP_API_TOKENS={"ietf.api.views.api_new_meeting_registration_v2": ["valid-token"]}) |
| 948 | + def test_api_new_meeting_registration_v2_cancelled(self): |
| 949 | + meeting = MeetingFactory(type_id='ietf') |
| 950 | + person = PersonFactory() |
| 951 | + regs = [ |
| 952 | + { |
| 953 | + 'affiliation': "Acme", |
| 954 | + 'country_code': 'US', |
| 955 | + 'email': person.email().address, |
| 956 | + 'first_name': person.first_name(), |
| 957 | + 'last_name': person.last_name(), |
| 958 | + 'meeting': str(meeting.number), |
| 959 | + 'reg_type': 'onsite', |
| 960 | + 'ticket_type': 'week_pass', |
| 961 | + 'checkedin': False, |
| 962 | + 'is_nomcom_volunteer': False, |
| 963 | + 'cancelled': False, |
| 964 | + } |
| 965 | + ] |
| 966 | + url = urlreverse('ietf.api.views.api_new_meeting_registration_v2') |
| 967 | + self.assertEqual(Registration.objects.count(), 0) |
| 968 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 969 | + self.assertContains(r, "Success", status_code=202) |
| 970 | + self.assertEqual(Registration.objects.count(), 1) |
| 971 | + regs[0]['cancelled'] = True |
| 972 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 973 | + self.assertContains(r, "Success", status_code=202) |
| 974 | + self.assertEqual(Registration.objects.count(), 0) |
| 975 | + |
| 976 | + @override_settings(APP_API_TOKENS={"ietf.api.views.api_new_meeting_registration_v2": ["valid-token"]}) |
| 977 | + def test_api_new_meeting_registration_v2_nomcom(self): |
| 978 | + meeting = MeetingFactory(type_id='ietf') |
| 979 | + person = PersonFactory() |
| 980 | + regs = [ |
| 981 | + { |
| 982 | + 'affiliation': "Acme", |
| 983 | + 'country_code': 'US', |
| 984 | + 'email': person.email().address, |
| 985 | + 'first_name': person.first_name(), |
| 986 | + 'last_name': person.last_name(), |
| 987 | + 'meeting': str(meeting.number), |
| 988 | + 'reg_type': 'onsite', |
| 989 | + 'ticket_type': 'week_pass', |
| 990 | + 'checkedin': False, |
| 991 | + 'is_nomcom_volunteer': False, |
| 992 | + 'cancelled': False, |
| 993 | + } |
| 994 | + ] |
| 995 | + |
| 996 | + url = urlreverse('ietf.api.views.api_new_meeting_registration_v2') |
| 997 | + now = datetime.datetime.now() |
| 998 | + if now.month > 10: |
| 999 | + year = now.year + 1 |
| 1000 | + else: |
| 1001 | + year = now.year |
| 1002 | + # create appropriate group and nomcom objects |
| 1003 | + nomcom = NomComFactory.create(is_accepting_volunteers=True, **nomcom_kwargs_for_year(year)) |
| 1004 | + |
| 1005 | + # first test is_nomcom_volunteer False |
| 1006 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 1007 | + self.assertContains(r, "Success", status_code=202) |
| 1008 | + # assert no Volunteers exists |
| 1009 | + self.assertEqual(Volunteer.objects.count(), 0) |
| 1010 | + |
| 1011 | + # test is_nomcom_volunteer True |
| 1012 | + regs[0]['is_nomcom_volunteer'] = True |
| 1013 | + r = self.client.post(url, data=json.dumps(regs), content_type='application/json', headers={"X-Api-Key": "valid-token"}) |
| 1014 | + self.assertContains(r, "Success", status_code=202) |
| 1015 | + # assert Volunteer exists |
| 1016 | + self.assertEqual(Volunteer.objects.count(), 1) |
| 1017 | + volunteer = Volunteer.objects.last() |
| 1018 | + self.assertEqual(volunteer.person, person) |
| 1019 | + self.assertEqual(volunteer.nomcom, nomcom) |
| 1020 | + self.assertEqual(volunteer.origin, 'registration') |
| 1021 | + |
831 | 1022 | def test_api_version(self):
|
832 | 1023 | DumpInfo.objects.create(date=timezone.datetime(2022,8,31,7,10,1,tzinfo=datetime.timezone.utc), host='testapi.example.com',tz='UTC')
|
833 | 1024 | url = urlreverse('ietf.api.views.version')
|
|
0 commit comments