fix: Make sure the tests work with the new unique ID

MISP · Jun 20, 2024 · 751ed3d · 751ed3d
1 parent 8df24c1
commit 751ed3d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/tests.py b/tests/tests.py
@@ -43,10 +43,10 @@ def test_dump_galaxies(self):
         for galaxy_file in glob(os.path.join(self.galaxies.root_dir_galaxies, '*.json')):
             with open(galaxy_file, 'r') as f:
                 galaxy = json.load(f)
-            galaxies_from_files[galaxy['name']] = galaxy
-        for name, g in self.galaxies.items():
+            galaxies_from_files[galaxy['type']] = galaxy
+        for _, g in self.galaxies.items():
             out = g.to_dict()
-            self.assertDictEqual(out, galaxies_from_files[g.name])
+            self.assertDictEqual(out, galaxies_from_files[g.type])
 
     def test_dump_clusters(self):
         clusters_from_files = {}