Merge pull request #34 from code-yeongyu/feature/tests

Author: code-yeongyu

.github/workflows/test.yml

@@ -1,5 +1,8 @@
 name: Test
 on:
+  push:
+    branches:
+      - master
   pull_request:
 
 jobs:
@@ -32,3 +35,8 @@ jobs:
 
       - name: Run test
         run: poetry run invoke test
+
+      - name: Upload Code Coverage
+        run: bash <(curl -s https://codecov.io/bash)
+        env:
+          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}

aishell/utils/make_executable_command.py

@@ -8,5 +8,5 @@ def make_executable_command(command: str) -> str:
     if command.endswith('`'):
         command = command[:-1]
     command = command.strip()
-    command = command.split('User: ')[0]
+    command = command.split('User: ')[-1]
     return command

aishell/utils/test/__init__.py

@@ -0,0 +1,25 @@
+from aishell.utils import make_executable_command
+
+
+def test_make_executable_command_starts_and_ends_with_newline():
+    command = '\nls -l\n'
+    expected_output = 'ls -l'
+    assert make_executable_command(command) == expected_output
+
+
+def test_make_executable_command_starts_and_ends_with_backtick():
+    command = '`python my_script.py`\n'
+    expected_output = 'python my_script.py'
+    assert make_executable_command(command) == expected_output
+
+
+def test_make_executable_command_starts_with_user():
+    command = 'User: cd ..'
+    expected_output = 'cd ..'
+    assert make_executable_command(command) == expected_output
+
+
+def test_make_executable_command_no_leading_or_trailing_whitespace():
+    command = 'git clone https://github.com/my_repo.git'
+    expected_output = 'git clone https://github.com/my_repo.git'
+    assert make_executable_command(command) == expected_output

aishell/utils/test/test_make_executable_command.py

@@ -38,6 +38,10 @@ allow_split_before_dict_value = false
 line-length = 119
 select = ["PLE", "PLR", "PLW", "E", "W", "F", "I", "Q", "C", "B"]
 
+[tool.pytest.ini_options]
+addopts = ["--cov=aishell"]
+
+
 [tool.poetry.dependencies]
 python = "^3.9"
 poetry = "^1.3.1"