From 41f3774ac46bd45e094222af567a9251dfaabda1 Mon Sep 17 00:00:00 2001
From: Lee Miller <lee.miller@tutanota.com>
Date: Mon, 2 Sep 2024 03:54:42 +0300
Subject: [PATCH] Specify some new API commands for managing black/white list

---
 src/tests/test_api.py | 48 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/src/tests/test_api.py b/src/tests/test_api.py
index 0df145bc..82b115c3 100644
--- a/src/tests/test_api.py
+++ b/src/tests/test_api.py
@@ -301,6 +301,54 @@ class TestAPI(TestAPIProto):
         self.assertEqual(
             json.loads(self.api.listSubscriptions())['subscriptions'], [])
 
+    def test_blackwhitelist(self):
+        """Test API commands for managing the black/white list"""
+        # Initially it's black
+        self.assertEqual(self.api.getBlackWhitelistKind(), 'black')
+        # Initially they are empty
+        self.assertEqual(
+            json.loads(self.api.listBlacklistEntries()).get('addresses'), [])
+        self.assertEqual(
+            json.loads(self.api.listWhitelistEntries()).get('addresses'), [])
+
+        # For the Blacklist:
+        # Add known address
+        self.api.addBlacklistEntry(
+            sample_deterministic_addr4,
+            base64.encodestring('tiger_4')
+        )
+        # Check list entry
+        entry = json.loads(self.api.listBlacklistEntries()).get('addresses')[0]
+        self.assertEqual(entry['address'], sample_deterministic_addr4)
+        self.assertEqual(base64.decodestring(entry['label']), 'tiger_4')
+        # Remove known address
+        self.api.deleteBlacklistEntry(sample_deterministic_addr4)
+        self.assertEqual(
+            json.loads(self.api.listBlacklistEntries()).get('addresses'), [])
+
+        # Only two kinds - black and white
+        six.assertRegex(
+            self, self.api.setBlackWhitelistKind('yellow'),
+            r'^API Error 0028:')
+        # Change kind
+        self.api.setBlackWhitelistKind('white')
+        self.assertEqual(self.api.getBlackWhitelistKind(), 'white')
+
+        # For the Whitelist:
+        # Add known address
+        self.api.addWhitelistEntry(
+            sample_deterministic_addr4,
+            base64.encodestring('tiger_4')
+        )
+        # Check list entry
+        entry = json.loads(self.api.listWhitelistEntries()).get('addresses')[0]
+        self.assertEqual(entry['address'], sample_deterministic_addr4)
+        self.assertEqual(base64.decodestring(entry['label']), 'tiger_4')
+        # Remove known address
+        self.api.deleteWhitelistEntry(sample_deterministic_addr4)
+        self.assertEqual(
+            json.loads(self.api.listWhitelistEntries()).get('addresses'), [])
+
     def test_send(self):
         """Test message sending"""
         addr = self._add_random_address('random_2')