[#778][Test] Don't send an existing public key

I added a new test to check that an existing public key is not being
sent to leap on every login

1 files changed, 17 insertions, 2 deletions

diff --git a/service/test/unit/bitmask_libraries/test_keymanager.py b/service/test/unit/bitmask_libraries/test_keymanager.py
index cdd6f3de..0c46460a 100644
--- a/service/test/unit/bitmask_libraries/test_keymanager.py
+++ b/service/test/unit/bitmask_libraries/test_keymanager.py
@@ -19,7 +19,6 @@ from mockito import when
 from test_abstract_leap import AbstractLeapTest
 from leap.keymanager import openpgp, KeyNotFound
 from pixelated.bitmask_libraries.keymanager import Keymanager
-from pixelated.bitmask_libraries.certs import LeapCertificate
 from pixelated.config import leap_config
 
 
@@ -64,4 +63,20 @@ class KeymanagerTest(AbstractLeapTest):
         keymanager.generate_openpgp_key()
 
         keyman.get_key.assert_called_with('test_user@some-server.test', private=True, fetch_remote=False)
-        keyman.gen_key.assert_called()
+        keyman.gen_key.assert_called_once()
+        keyman.send_key.assert_called_once()
+
+    @patch('pixelated.bitmask_libraries.keymanager.KeyManager')
+    def test_existing_key_not_sent_to_leap(self, keymanager_mock):
+        keyman = keymanager_mock.return_value
+        keyman.get_key.side_effect = KeyNotFound
+        keymanager = Keymanager(self.provider,
+                                self.soledad,
+                                'test_user@some-server.test',
+                                self.auth.token,
+                                self.auth.uuid)
+
+        when(keymanager)._key_exists('test_user@some-server.test').thenReturn(True)
+        keymanager.generate_openpgp_key()
+        keyman.gen_key.assert_not_called()
+        keyman.send_key.assert_not_called()