class TestErrorResponse(unittest.TestCase): def setUp(self): load_test_fixture('fixtures/test-error-response.json') self.analyzer = Analyzer() 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')
class TestMinimalConfig(unittest.TestCase): def setUp(self): load_test_fixture('fixtures/test-minimal-config.json') self.analyzer = Analyzer() def test_default_config(self): self.assertEqual(self.analyzer.data_type, 'ip') self.assertEqual(self.analyzer.tlp, 2) self.assertEqual(self.analyzer.enable_check_tlp, False) self.assertEqual(self.analyzer.max_tlp, 2) self.assertEqual(self.analyzer.http_proxy, None) self.assertEqual(self.analyzer.https_proxy, None) def test_artifact_data(self): self.assertEqual(self.analyzer.getData(), "1.1.1.1") self.assertEqual(self.analyzer.get_data(), "1.1.1.1") def test_params_data(self): self.assertEqual(self.analyzer.getParam('data'), "1.1.1.1") self.assertEqual(self.analyzer.get_param('data'), "1.1.1.1")