update test

ntk/installer.py

@@ -7,13 +7,18 @@
 # This script is used to create a command alias (`ntk`) for running a Python module (`ntk`) more easily.
 
 # To generate an executable using PyInstaller, use the following command:
-# python -m PyInstaller --onefile --console .\ntk\installer.py -n n
+# python -m PyInstaller --onefile --console .\ntk\installer.py -n ntk
 
-if __name__ == "__main__":
+
+def auto_install_and_generate_ntk_command():
     subprocess.run(['python', '-m', 'pip', 'install',  'next-theme-kit'])
 
     # Set up a command alias: 'ntk' will execute 'python -m ntk' with any passed arguments
     os.system("doskey ntk=python -m ntk $*")
 
     # Open a new command prompt window and keep it open
     os.system("cmd /k")
+
+
+if __name__ == "__main__":
+    auto_install_and_generate_ntk_command()

tests/test_installer.py

@@ -0,0 +1,16 @@
+from unittest.case import TestCase
+from unittest.mock import patch
+
+from ntk.installer import auto_install_and_generate_ntk_command
+
+
+class TestInstaller(TestCase):
+    def test_auto_install_and_generate_ntk_command_should_call_subprocess_and_os_correctly(self):
+        with patch("subprocess.run") as mock_subprocess, patch("os.system") as mock_os:
+            auto_install_and_generate_ntk_command()
+
+            mock_subprocess.assert_called_with(['python', '-m', 'pip', 'install', 'next-theme-kit'])
+
+            mock_os.assert_any_call("doskey ntk=python -m ntk $*")
+
+            mock_os.assert_any_call("cmd /k")