From cce1e3d448e7a2692a88350ec46cabdec4bb7b0c Mon Sep 17 00:00:00 2001
From: leba-atr <>
Date: Thu, 6 Mar 2025 12:13:35 +0000
Subject: [PATCH] prefer Atomic Test UUID over calculated hash for Ability ID

---
 app/atomic_svc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/app/atomic_svc.py b/app/atomic_svc.py
index 38e3690b..d7f43db6 100644
--- a/app/atomic_svc.py
+++ b/app/atomic_svc.py
@@ -301,7 +301,7 @@ async def _save_ability(self, entries, test):
         """
         Return True if an ability was saved.
         """
-        ability_id = hashlib.md5(json.dumps(test).encode()).hexdigest()
+        ability_id = test.get('auto_generated_guid') or hashlib.md5(json.dumps(test).encode()).hexdigest()
 
         tactics_li = self.technique_to_tactics.get(entries['attack_technique'], ['redcanary-unknown'])
         tactic = 'multiple' if len(tactics_li) > 1 else tactics_li[0]