# HG changeset patch
# User Goffi <goffi@goffi.org>
# Date 1584557647 -3600
# Node ID 0469c53ed5dde83ff28304331f60f3f179cf0c29
# Parent  8418e0c83ed7974d76537bbafbb40f8c14a9e123
core (memory/encryption): namespace can now be specified to test a specific algorithm in isEncryptionRequested

diff -r 8418e0c83ed7 -r 0469c53ed5dd sat/memory/encryption.py
--- a/sat/memory/encryption.py	Fri Mar 13 17:50:08 2020 +0100
+++ b/sat/memory/encryption.py	Wed Mar 18 19:54:07 2020 +0100
@@ -437,13 +437,22 @@
         mess_data[C.MESS_KEY_ENCRYPTED] = True
         return mess_data
 
-    def isEncryptionRequested(self, mess_data):
+    def isEncryptionRequested(self, mess_data, namespace=None):
         """Helper method to check if encryption is requested in an outgoind message
 
         @param mess_data(dict): message data for outgoing message
+        @param namespace(str, None): if set, check if encryption is requested for the
+            algorithm specified
         @return (bool): True if the encryption flag is present
         """
-        return bool(mess_data.get(C.MESS_KEY_ENCRYPTION, False))
+        encryption = mess_data.get(C.MESS_KEY_ENCRYPTION)
+        if encryption is None:
+            return False
+        # we get plugin even if namespace is None to be sure that the key exists
+        plugin = encryption['plugin']
+        if namespace is None:
+            return True
+        return plugin.namespace == namespace
 
     def isEncrypted(self, mess_data):
         """Helper method to check if an incoming message has the e2e encrypted flag