def test_b64_pattern(self): head = '<!DOCTYPE html>' for code in ['400', '401', '402', '403', '404', '500']: pattern = Zappa.selection_pattern(code) document = base64.b64encode(head + code + random_string(50)) self.assertRegexpMatches(document, pattern) for bad_code in ['200', '301', '302']: document = base64.b64encode(head + bad_code + random_string(50)) self.assertNotRegexpMatches(document, pattern)
def test_redirect_pattern(self): test_urls = [ # a regular endpoint url 'https://asdf1234.execute-api.us-east-1.amazonaws.com/env/path/to/thing', # an external url (outside AWS) 'https://github.com/Miserlou/zappa/issues?q=is%3Aissue+is%3Aclosed', # a local url '/env/path/to/thing' ] for code in ['301', '302']: pattern = Zappa.selection_pattern(code) for url in test_urls: self.assertRegexpMatches(url, pattern)
def test_200_pattern(self): pattern = Zappa.selection_pattern('200') self.assertEqual(pattern, '')