Fix entrypoint

safe_cli/main.py

@@ -137,7 +137,11 @@ def build_safe_cli():
     return SafeCli(args.safe_address, args.node_url, args.history)
 
 
-if __name__ == "__main__":
+def main(*args, **kwargs):
     safe_cli = build_safe_cli()
     safe_cli.print_startup_info()
     safe_cli.loop()
+
+
+if __name__ == "__main__":
+    main()

safe_cli/safe_addresses.py

@@ -4,18 +4,20 @@
 https://github.com/gnosis/safe-deployments/tree/main/src/assets/v1.4.1
 https://github.com/gnosis/safe-deployments/tree/main/src/assets/v1.3.0
 """
+from typing import Sequence
+
 from eth_typing import ChecksumAddress
 
 from gnosis.eth import EthereumClient
 
 
 def _get_valid_contract(
-    ethereum_client: EthereumClient, addresses: ChecksumAddress
+    ethereum_client: EthereumClient, addresses: Sequence[ChecksumAddress]
 ) -> ChecksumAddress:
     """
     :param ethereum_client:
     :param addresses:
-    :return: First valid contract found in blockchain
+    :return: First valid contract from the list of addresses provided found in blockchain
     """
 
     for address in addresses:

tests/test_entrypoint.py

@@ -45,7 +45,8 @@ def test_build_safe_cli(self, retrieve_all_info_mock: MagicMock):
         )
 
         safe_cli = self.build_test_safe_cli()
-        self.assertIsNone(safe_cli.print_startup_info())
+        with mock.patch.object(SafeOperator, "is_version_updated", return_value=True):
+            self.assertIsNone(safe_cli.print_startup_info())
         self.assertIsInstance(safe_cli.get_prompt_text(), HTML)
         self.assertIsInstance(safe_cli.get_bottom_toolbar(), HTML)
 

tests/test_safe_addresses.py

@@ -0,0 +1,25 @@
+import unittest
+
+from eth_account import Account
+
+from safe_cli.safe_addresses import _get_valid_contract
+
+from .safe_cli_test_case_mixin import SafeCliTestCaseMixin
+
+
+class TestSafeAddresses(SafeCliTestCaseMixin, unittest.TestCase):
+    def test_get_valid_contract(self):
+        addresses = [
+            Account.create().address,
+            Account.create().address,
+            self.safe_contract_V1_4_1.address,
+            Account.create().address,
+        ]
+        expected_address = self.safe_contract_V1_4_1.address
+        self.assertEqual(
+            _get_valid_contract(self.ethereum_client, addresses), expected_address
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()