test_suite_analyzer.py 文件源码

def test_error_response(self):
        self.assertEqual(self.analyzer.get_param('config.password'), "secret")
        self.assertEqual(self.analyzer.get_param('config.key'), "secret")
        self.assertEqual(self.analyzer.get_param('config.apikey'), "secret")
        self.assertEqual(self.analyzer.get_param('config.api_key'), "secret")

        # Run the error method
        with self.assertRaises(SystemExit):
            self.analyzer.error('Error', True)

        # Get the output
        output = self.analyzer.fpoutput.getvalue().strip()
        json_output = json.loads(output)

        self.assertEqual(json_output['success'], False)
        self.assertEqual(json_output['errorMessage'], 'Error')
        self.assertEqual(json_output['input']['dataType'], 'ip')
        self.assertEqual(json_output['input']['data'], '1.1.1.1')
        self.assertEqual(json_output['input']['config']['password'], 'REMOVED')
        self.assertEqual(json_output['input']['config']['key'], 'REMOVED')
        self.assertEqual(json_output['input']['config']['apikey'], 'REMOVED')
        self.assertEqual(json_output['input']['config']['api_key'], 'REMOVED')