Coverage for benefits/metro_mobility_wallet/session.py: 100%

1import logging 

2 

3from django.http import HttpRequest 

4 

5from benefits.core.models.enrollment import EnrollmentFlow, SystemName 

6 

7logger = logging.getLogger(__name__) 

8 

9 

10class Session: 

11 def __init__(self, request: HttpRequest): 

12 """Initialize a new Metro Mobility Wallet session wrapper for this request""" 

13 

14 self.request = request 

15 self.session = request.session 

16 

17 @property 

18 def flow(self) -> EnrollmentFlow: 

19 return EnrollmentFlow.objects.filter(system_name=SystemName.METRO_MOBILITY_WALLET).first()