Coverage for benefits/enrollment_littlepay/views.py: 89%

1import logging 

2 

3from django.http import JsonResponse 

4from django.urls import reverse 

5from django.views.generic import FormView, View 

6import sentry_sdk 

7 

8from benefits.routes import routes 

9from benefits.core import models, session 

10from benefits.core.mixins import EligibleSessionRequiredMixin 

11 

12from benefits.enrollment import analytics, forms 

13from benefits.enrollment.enrollment import Status, handle_enrollment_results 

14from benefits.enrollment_littlepay.enrollment import enroll, get_card_types_for_js, request_card_tokenization_access 

15from benefits.enrollment_littlepay.session import Session 

16 

17logger = logging.getLogger(__name__) 

18 

19 

20class TokenView(EligibleSessionRequiredMixin, View): 

21 """View handler for the card tokenization access token.""" 

22 

23 def get(self, request, *args, **kwargs): 

24 session = Session(request) 

25 

26 if not session.access_token_valid(): 

27 response = request_card_tokenization_access(request) 

28 

29 if response.status is Status.SUCCESS: 

30 session.access_token = response.access_token 

31 session.access_token_expiry = response.expires_at 

32 elif response.status is Status.SYSTEM_ERROR or response.status is Status.EXCEPTION: 32 ↛ 45line 32 didn't jump to line 45 because the condition on line 32 was always true

33 logger.debug("Error occurred while requesting access token", exc_info=response.exception) 

34 sentry_sdk.capture_exception(response.exception) 

35 analytics.failed_pretokenization_request(request, response.status_code) 

36 

37 if response.status is Status.SYSTEM_ERROR: 

38 redirect = reverse(routes.ENROLLMENT_SYSTEM_ERROR) 

39 else: 

40 redirect = reverse(routes.SERVER_ERROR) 

41 

42 data = {"redirect": redirect} 

43 return JsonResponse(data) 

44 

45 data = {"token": session.access_token} 

46 return JsonResponse(data) 

47 

48 

49class IndexView(EligibleSessionRequiredMixin, FormView): 

50 template_name = "enrollment_littlepay/index.html" 

51 form_class = forms.CardTokenizeSuccessForm 

52 

53 def get_context_data(self, **kwargs): 

54 request = self.request 

55 agency = session.agency(request) 

56 flow = session.flow(request) 

57 

58 tokenize_retry_form = forms.CardTokenizeFailForm(routes.ENROLLMENT_RETRY, "form-card-tokenize-fail-retry") 

59 tokenize_server_error_form = forms.CardTokenizeFailForm(routes.SERVER_ERROR, "form-card-tokenize-fail-server-error") 

60 tokenize_system_error_form = forms.CardTokenizeFailForm( 

61 routes.ENROLLMENT_SYSTEM_ERROR, "form-card-tokenize-fail-system-error" 

62 ) 

63 tokenize_success_form = forms.CardTokenizeSuccessForm( 

64 action_url=routes.ENROLLMENT_LITTLEPAY_INDEX, auto_id=True, label_suffix="" 

65 ) 

66 

67 context = { 

68 "forms": [tokenize_retry_form, tokenize_server_error_form, tokenize_system_error_form, tokenize_success_form], 

69 "cta_button": "tokenize_card", 

70 "enrollment_method": models.EnrollmentMethods.DIGITAL, 

71 "token_field": "card_token", 

72 "form_retry": tokenize_retry_form.id, 

73 "form_server_error": tokenize_server_error_form.id, 

74 "form_success": tokenize_success_form.id, 

75 "form_system_error": tokenize_system_error_form.id, 

76 "overlay_language": self._get_overlay_language(request.LANGUAGE_CODE), 

77 "card_types": get_card_types_for_js(), 

78 } 

79 

80 enrollment_index_context_dict = flow.enrollment_index_context 

81 

82 match agency.littlepay_config.environment: 

83 case models.Environment.QA.value: 83 ↛ 86line 83 didn't jump to line 86 because the pattern on line 83 always matched

84 url = "https://verify.qa.littlepay.com/assets/js/littlepay.min.js" 

85 card_tokenize_env = "https://verify.qa.littlepay.com" 

86 case models.Environment.PROD.value: 

87 url = "https://verify.littlepay.com/assets/js/littlepay.min.js" 

88 card_tokenize_env = "https://verify.littlepay.com" 

89 case _: 

90 raise ValueError("Unrecognized environment value") 

91 

92 transit_processor_context = dict( 

93 name="Littlepay", website="https://littlepay.com", card_tokenize_url=url, card_tokenize_env=card_tokenize_env 

94 ) 

95 

96 enrollment_index_context_dict["transit_processor"] = transit_processor_context 

97 context.update(enrollment_index_context_dict) 

98 

99 return context 

100 

101 def _get_overlay_language(self, django_language_code): 

102 """Given a Django language code, return the corresponding language code to use with Littlepay's overlay.""" 

103 # mapping from Django's I18N LANGUAGE_CODE to Littlepay's overlay language code 

104 overlay_language = {"en": "en", "es": "es-419"}.get(django_language_code, "en") 

105 return overlay_language 

106 

107 def form_valid(self, form): 

108 card_token = form.cleaned_data.get("card_token") 

109 status, exception = enroll(self.request, card_token) 

110 

111 return handle_enrollment_results(self.request, status, exception) 

112 

113 def form_invalid(self, form): 

114 raise Exception("Invalid card token form")