mirror of https://github.com/MISP/PyMISPGalaxies
24 lines
747 B
Python
24 lines
747 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from pymispgalaxies import Galaxies, Clusters, Cluster
|
||
|
|
|
||
|
|
|
||
|
|
class TestPyMISPGalaxiesApi(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.galaxies = Galaxies()
|
||
|
|
self.clusters = Clusters(skip_duplicates=False)
|
||
|
|
self.maxDiff = None
|
||
|
|
|
||
|
|
def test_get_by_external_id(self):
|
||
|
|
cluster = Cluster(cluster='mitre-attack-pattern')
|
||
|
|
self.assertIsNotNone(cluster)
|
||
|
|
cluster_by_external_id = cluster.get_by_external_id('T1525')
|
||
|
|
cluster_by_value = cluster.get('Implant Internal Image - T1525')
|
||
|
|
self.assertEqual(cluster_by_external_id, cluster_by_value)
|
||
|
|
|
||
|
|
with self.assertRaises(KeyError):
|
||
|
|
cluster.get_by_external_id('XXXXXX')
|