Python Analyzer.get_dataの例

コード例 #1

class AnalyzerExtractorOutputTest(unittest.TestCase):
    def setUp(self):
        sys.stdin = StringIO(json.dumps({"data": "8.8.8.8", "dataType": "ip"}))
        sys.stdout = StringIO()
        self.analyzer = Analyzer()

    def test_output(self):
        # Run the report method
        self.analyzer.report({'result': '1.2.3.4'})

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

        # Checks
        self.assertNotIn(self.analyzer.get_data(), output)
        self.assertEqual(json_output['artifacts'][0]['value'], '1.2.3.4')
        self.assertEqual(json_output['artifacts'][0]['type'], 'ip')

コード例 #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")

コード例 #4

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

class AnalyzerExtractorOutputTest(unittest.TestCase):
    def setUp(self):
        sys.stdin = StringIO(json.dumps({
            "data": "8.8.8.8",
            "dataType": "ip"
        }))
        sys.stdout = StringIO()
        self.analyzer = Analyzer()

    def test_output(self):
        # Run the report method
        self.analyzer.report({'result': '1.2.3.4'})

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

        # Checks
        self.assertNotIn(self.analyzer.get_data(), output)
        self.assertEqual(json_output['artifacts'][0]['value'], '1.2.3.4')
        self.assertEqual(json_output['artifacts'][0]['type'], 'ip')