]> kaliko git repositories - sid.git/commitdiff
rtbl: Ease testing
authorkaliko <kaliko@azylum.org>
Thu, 23 Mar 2023 16:30:33 +0000 (17:30 +0100)
committerkaliko <kaliko@azylum.org>
Thu, 23 Mar 2023 16:30:33 +0000 (17:30 +0100)
Make testing easier with rtbl.BL.init mapping to add testing item to the
block list (locally).

Generate a hash:
>>> import hashlib
>>> test_hash = hashlib.sha256('user@example.org'.encode('utf-8')).hexdigest()

Then use it to pre-populate the block list:

>>> from sid.rtbl import RTBL, BL
>>> BL.init = {test_hash: 'For testing purposes'}

sid/rtbl.py

index 0ea3bbbf36a8be446305595d36eeac78501e5795..b5c58cbef662635c5105572f3812d47d4903547f 100644 (file)
@@ -21,9 +21,11 @@ def jid_to_sha256(jid: JID) -> str:
 class BL:
     """Plain object to keep track of block list items.
     Only used in RTBL plugin."""
+    #: Initial seed to ease testing
+    init = {}
 
     def __init__(self, initial_bl):
-        self.sha256_jids: Dict[str, Optional[str]] = {}
+        self.sha256_jids: Dict[str, Optional[str]] = dict(BL.init)
         for item in initial_bl:
             self.insert_item(item)