ensure notification message integrity is encoded as base64 (#484)

* ensure notification message integrity is encoded as base64 * use centre-id for testing message queries
wmo-im · Aug 17, 2023 · c991011 · c991011
1 parent 4c3912e
commit c991011
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/tests/integration/test_workflow.py b/tests/integration/test_workflow.py
@@ -193,10 +193,10 @@ def test_message_api():
 
     # test messages per test dataset
     counts = {
-        'mwi': 24,
-        'ita': 33,
-        'dza': 29,
-        'rou': 185
+        'mwi_met_centre': 24,
+        'roma_met_centre': 33,
+        'alger_met_centre': 29,
+        'rnimh': 185
     }
     for key, value in counts.items():
         url = f'{API_URL}/collections/messages/items?sortby=-datetime&q={key}&limit=1'  # noqa

diff --git a/wis2box-management/wis2box/pubsub/message.py b/wis2box-management/wis2box/pubsub/message.py
@@ -19,6 +19,7 @@
 #
 ###############################################################################
 
+import base64
 from datetime import datetime
 from enum import Enum
 import json
@@ -120,7 +121,7 @@ def _generate_checksum(self, bytes, algorithm: SecureHashAlgorithms) -> str:  #
 
         sh = getattr(hashlib, algorithm)()
         sh.update(bytes)
-        return sh.hexdigest()
+        return base64.b64encode(sh.digest()).decode()
 
 
 class WISNotificationMessage(PubSubMessage):