проверка имени во время активации

pytest.py

@@ -3,15 +3,45 @@ import os
 import hlna
 import time
 import unittest
-from unittest.mock import patch
 import zipfile
 from click.testing import CliRunner
 from pathlib import Path
 home_dir = Path.home()
 config_hlna = f"{home_dir}/.config/hlna/"
 
+from unittest.mock import patch, MagicMock
+
+from hlna import config
 
 
+class TestConfig(unittest.TestCase):
+
+    @patch('builtins.input', side_effect=['1'])
+    @patch('hlna.config_ark')
+    def test_config_ark_valid(self, config_ark_mock, input_mock):
+        config()
+        config_ark_mock.assert_called_once()
+
+    @patch('builtins.input', side_effect=['2'])
+    @patch('hlna.config_7daystodie')
+    def test_config_7daystodie_valid(self, config_7daystodie_mock, input_mock):
+        config()
+        config_7daystodie_mock.assert_called_once()
+
+    @patch('builtins.input', side_effect=['3'])
+    @patch('sys.stdout', new_callable=MagicMock)
+    def test_config_invalid_game(self, mocked_stdout, input_mock):
+        config()
+        self.assertEqual(mocked_stdout.getvalue().strip(), "Пока есть только ARK и 7Days xD")
+
+    @patch('builtins.input', side_effect=['a', '1'])
+    @patch('hlna.config_ark')
+    @patch('sys.stdout', new_callable=MagicMock)
+    def test_config_invalid_input(self, mocked_stdout, config_ark_mock, input_mock):
+        config()
+        config_ark_mock.assert_called_once()
+        self.assertEqual(mocked_stdout.getvalue().strip(),
+                         "Выберите игру для конфигурирования\n1. ARK Survival Evolved\n2. 7 Days to Die\n: ")
 class TestFindFile(unittest.TestCase):
     def setUp(self):
         self.test_dir = 'test_dir'
@@ -94,6 +124,13 @@ class TestBackup(unittest.TestCase):
             self.assertEqual(zip_file.testzip(), None)
         self.assertTrue(os.path.exists(os.path.join(target, f"{config_hlna}ARK/test_file.txt")))
 
+class TestGetExternalIP(unittest.TestCase):
+    def test_backup(self):
+        self.assertTrue(bool(hlna.get_external_ip()))
+
+#class TestGetExternalIP(unittest.TestCase):
+#    def test_backup(self):
+#        self.assertTrue(bool(hlna.get_external_ip()))
 if __name__ == '__main__':
     unittest.main()