Python Analyzer.get_paramの例

コード例 #1

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')

コード例 #2

ファイル: test_suite_analyzer.py プロジェクト: saadkadhi/Cortex-Analyzers

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")

コード例 #3

ファイル: test_suite_analyzer.py プロジェクト: Cyberprotect/Cortex-Analyzers

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")