@@ -159,4 +159,40 @@ def test_bypass_monitor_header_on_valid_request(self):
159159 context = PxContext (request , request_handler .config )
160160 context .score = 0
161161 response = request_handler .handle_verification (context , request )
162- self .assertEqual (response , True )
162+ self .assertEqual (response , True )
163+
164+ def test_specific_enforced_routes_with_enforced_route (self ):
165+ config = PxConfig ({'app_id' : 'PXfake_app_id' ,
166+ 'auth_token' : '' ,
167+ 'module_mode' : px_constants .MODULE_MODE_BLOCKING ,
168+ 'enforced_specific_routes' : ['/profile' ],
169+ });
170+ headers = {'X-FORWARDED-FOR' : '127.0.0.1' ,
171+ 'remote-addr' : '127.0.0.1' ,
172+ 'content_length' : '100' }
173+ request_handler = PxRequestVerifier (config )
174+ builder = EnvironBuilder (headers = headers , path = '/profile' )
175+ env = builder .get_environ ()
176+ request = Request (env )
177+ context = PxContext (request , request_handler .config )
178+ context .score = 100
179+ response = request_handler .handle_verification (context , request )
180+ self .assertEqual (response .status , '403 Forbidden' )
181+
182+ def test_specific_enforced_routes_with_unenforced_route (self ):
183+ config = PxConfig ({'app_id' : 'PXfake_app_id' ,
184+ 'auth_token' : '' ,
185+ 'module_mode' : px_constants .MODULE_MODE_BLOCKING ,
186+ 'enforced_specific_routes' : ['/profile' ],
187+ });
188+ headers = {'X-FORWARDED-FOR' : '127.0.0.1' ,
189+ 'remote-addr' : '127.0.0.1' ,
190+ 'content_length' : '100' }
191+ request_handler = PxRequestVerifier (config )
192+ builder = EnvironBuilder (headers = headers , path = '/' )
193+ env = builder .get_environ ()
194+ request = Request (env )
195+ context = PxContext (request , request_handler .config )
196+ context .score = 100
197+ response = request_handler .verify_request (context , request )
198+ self .assertEqual (response , True )
0 commit comments