def test_valid_option_validation(self): user_args = { 'target_port': 500, 'target_ip': '192.168.0.1', 'token_refresh_cmd': 'some command', 'token_refresh_interval': 30, 'client_certificate_path': '\\path\\to\\file.pem', 'fuzzing_mode': 'random-walk', 'ignore_decoding_failures': True } try: settings = RestlerSettings(user_args) settings.validate_options() except: assert False settings.TEST_DeleteInstance() user_args = {'target_port': 500, 'target_ip': '192.168.0.1'} try: settings = RestlerSettings(user_args) settings.validate_options() except: assert False
def test_valid_option_validation(self): user_args = {'target_port': 500, 'target_ip': '192.168.0.1', 'token_refresh_cmd': 'some command', 'token_refresh_interval': 30, 'fuzzing_mode': 'random-walk'} try: settings = RestlerSettings(user_args) settings.validate_options() except: assert False settings.TEST_DeleteInstance() user_args = {'target_port': 500, 'target_ip': '192.168.0.1'} try: settings = RestlerSettings(user_args) settings.validate_options() except: assert False
def test_custom_non_bug_codes(self): user_args = {"custom_non_bug_codes": ["200", "4?4", "3*"]} try: settings = RestlerSettings(user_args) settings.validate_options() except: assert False response200 = HttpResponse('HTTP/1.1 200 response') response201 = HttpResponse('HTTP/1.1 201 response') response400 = HttpResponse('HTTP/1.1 400 response') response404 = HttpResponse('HTTP/1.1 404 response') response414 = HttpResponse('HTTP/1.1 414 response') response300 = HttpResponse('HTTP/1.1 300 response') response301 = HttpResponse('HTTP/1.1 301 response') response500 = HttpResponse('HTTP/1.1 500 response') self.assertFalse(response200.has_bug_code()) self.assertTrue(response201.has_bug_code()) self.assertTrue(response400.has_bug_code()) self.assertFalse(response404.has_bug_code()) self.assertFalse(response414.has_bug_code()) self.assertFalse(response300.has_bug_code()) self.assertFalse(response301.has_bug_code()) self.assertTrue(response500.has_bug_code()) settings.TEST_DeleteInstance() user_args["custom_non_bug_codes"].append("500") try: settings = RestlerSettings(user_args) settings.validate_options() except: assert False self.assertFalse(response500.has_bug_code())
def tearDown(self): RestlerSettings.TEST_DeleteInstance()