AIUnitTest 0.0.2__tar.gz → 0.0.5__tar.gz

{aiunittest-0.0.2 → aiunittest-0.0.5}/.github/workflows/ci.yml

@@ -8,9 +8,11 @@ jobs:
   setup:
     name: Setup & Lint
     runs-on: ubuntu-latest
+    permissions:
+      contents: read
 
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
 
       - name: Setup Python 3.13 (with pip cache)
         uses: actions/setup-python@v5
@@ -44,7 +46,7 @@ jobs:
       contents: read
       pull-requests: write
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
 
       - uses: actions/setup-python@v5
         with:
@@ -81,7 +83,7 @@ jobs:
       contents: read
       pull-requests: write
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: actions/setup-python@v5
         with:
           python-version: "3.13"
@@ -89,6 +91,9 @@ jobs:
       - name: Install dependencies
         run: pip install -r requirements.txt
 
+      - name: Install project
+        run: pip install -e .
+
       - name: Run tests
         run: pytest
 
@@ -99,3 +104,5 @@ jobs:
       - name: Label dependabot PR
         if: steps.fetch-metadata.outputs.dependency-type == 'direct:production'
         run: gh pr edit ${{ github.event.pull_request.html_url }} --add-label "dependencies"
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}

aiunittest-0.0.5/.github/workflows/release.yml

@@ -0,0 +1,86 @@
+name: Create and Publish Release
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  create_and_publish_release:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write # To create the release and tag
+      id-token: write # For authentication with PyPI
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v5
+        with:
+          fetch-depth: 0 # Required for setuptools_scm
+
+      - name: Get latest version and calculate next
+        id: get_next_version
+        run: |
+          LATEST_TAG=$(git describe --tags --abbrev=0 2>/dev/null || echo "v0.0.0")
+          echo "Latest tag: $LATEST_TAG"
+
+          # Remove 'v' prefix and split into parts
+          VERSION_PARTS=$(echo $LATEST_TAG | sed 's/^v//' | tr '.' '\n')
+          MAJOR=$(echo $VERSION_PARTS | awk '{print $1}')
+          MINOR=$(echo $VERSION_PARTS | awk '{print $2}')
+          PATCH=$(echo $VERSION_PARTS | awk '{print $3}')
+
+          # Increment patch version
+          NEXT_PATCH=$((PATCH + 1))
+          NEXT_VERSION="v${MAJOR}.${MINOR}.${NEXT_PATCH}"
+          echo "Next version: $NEXT_VERSION"
+
+          echo "NEXT_VERSION=$NEXT_VERSION" >> $GITHUB_OUTPUT
+
+      - name: Create new tag
+        run: |
+          git config user.name "github-actions[bot]"
+          git config user.email "github-actions[bot]@users.noreply.github.com"
+          git tag -a ${{ steps.get_next_version.outputs.NEXT_VERSION }} -m "Release ${{ steps.get_next_version.outputs.NEXT_VERSION }}"
+          git push origin ${{ steps.get_next_version.outputs.NEXT_VERSION }}
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.9"
+
+      - name: Install build dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install setuptools_scm build twine
+
+      - name: Build package
+        run: python -m build
+
+      - name: Create GitHub Release
+        uses: ncipollo/release-action@v1
+        with:
+          tag: ${{ steps.get_next_version.outputs.NEXT_VERSION }}
+          name: Release ${{ steps.get_next_version.outputs.NEXT_VERSION }}
+          generateReleaseNotes: true
+          artifacts: "dist/*"
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+
+      - name: Set up Conda
+        uses: conda-incubator/setup-miniconda@v3
+        with:
+          python-version: 3.9
+          auto-update-conda: true
+          auto-activate-base: false
+
+      - name: Build and upload Conda package
+        shell: bash -l {0}
+        run: |
+          conda install -c conda-forge conda-build anaconda-client -y
+          conda install -c conda-forge --file conda/meta.yaml --yes
+          export GIT_TAG=${{ steps.get_next_version.outputs.NEXT_VERSION }}
+          conda build conda --output-folder conda-dist
+          anaconda login --with-token ${{ secrets.CONDA_TOKEN }}
+          anaconda upload conda-dist/**/*.tar.bz2 --label main --skip-existing

{aiunittest-0.0.2 → aiunittest-0.0.5}/.pre-commit-config.yaml

@@ -72,5 +72,9 @@ repos:
     rev: v1.17.1
     hooks:
       - id: mypy
-        args: ["--config-file", "pyproject.toml"]
+        entry: mypy --config-file pyproject.toml
+        language: python
+        require_serial: true
         additional_dependencies: ["typer", "openai", "coverage", "pytest"]
+        env:
+          PYTHONPATH: src

{aiunittest-0.0.2/src/AIUnitTest.egg-info → aiunittest-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AIUnitTest
-Version: 0.0.2
+Version: 0.0.5
 Summary: CLI to generate and update Python unit tests automatically using coverage and AI
 Author: Ofido
 Project-URL: Homepage, https://github.com/ofido/AIUnitTest

{aiunittest-0.0.2 → aiunittest-0.0.5}/pyproject.toml

@@ -24,6 +24,9 @@ dependencies = ["openai", "coverage", "typer", "tomli"]
 "Homepage" = "https://github.com/ofido/AIUnitTest"
 "Bug Tracker" = "https://github.com/ofido/AIUnitTest/issues"
 
+[project.scripts]
+ai-unit-test = "ai_unit_test.cli:app"
+
 [project.optional-dependencies]
 dev = [
   "black",

{aiunittest-0.0.2 → aiunittest-0.0.5/src/AIUnitTest.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AIUnitTest
-Version: 0.0.2
+Version: 0.0.5
 Summary: CLI to generate and update Python unit tests automatically using coverage and AI
 Author: Ofido
 Project-URL: Homepage, https://github.com/ofido/AIUnitTest

{aiunittest-0.0.2 → aiunittest-0.0.5}/src/AIUnitTest.egg-info/SOURCES.txt

@@ -16,13 +16,13 @@ reset_fake_project.py
 .github/ISSUE_TEMPLATE/documentation.yml
 .github/ISSUE_TEMPLATE/feature_request.yml
 .github/workflows/ci.yml
-.github/workflows/create-release.yml
 .github/workflows/release.yml
 conda/meta.yaml
 docs/initial.md
 src/AIUnitTest.egg-info/PKG-INFO
 src/AIUnitTest.egg-info/SOURCES.txt
 src/AIUnitTest.egg-info/dependency_links.txt
+src/AIUnitTest.egg-info/entry_points.txt
 src/AIUnitTest.egg-info/requires.txt
 src/AIUnitTest.egg-info/top_level.txt
 src/ai_unit_test/__init__.py

aiunittest-0.0.5/src/AIUnitTest.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ai-unit-test = ai_unit_test.cli:app

{aiunittest-0.0.2 → aiunittest-0.0.5}/src/ai_unit_test/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.2'
-__version_tuple__ = version_tuple = (0, 0, 2)
+__version__ = version = '0.0.5'
+__version_tuple__ = version_tuple = (0, 0, 5)

{aiunittest-0.0.2 → aiunittest-0.0.5}/src/ai_unit_test/cli.py

@@ -14,6 +14,7 @@ from ai_unit_test.file_helper import (
     find_relevant_tests,
     find_test_file,
     get_source_code_chunks,
+    insert_new_test,
     read_file_content,
     write_file_content,
 )
@@ -154,15 +155,17 @@ async def _process_missing_info(missing_info: dict[Path, list[int]], tests_folde
                 f"(lines {chunk.start_line}-{chunk.end_line}) with uncovered lines: {chunk_uncovered_lines}"
             )
             try:
+                existing_content = read_file_content(test_file)
                 updated_test: str = await update_test_with_llm(
                     chunk.source_code,  # Pass chunk source code
-                    read_file_content(test_file) or "",  # Still pass the whole test file for context
+                    existing_content or "",  # Still pass the whole test file for context
                     str(source_file_path),
                     chunk_uncovered_lines,  # Pass chunk-specific uncovered lines
                     other_tests_content,
                     test_style,  # Pass the detected test style
                 )
-                write_file_content(test_file, updated_test, mode="a")  # Append the new test content
+                new_content = insert_new_test(existing_content, updated_test)
+                write_file_content(test_file, new_content)  # Overwrite the file with the new content
                 logger.info(f"✅ Test file updated successfully: {test_file}")
             except Exception as exc:  # pragma: no cover
                 logger.error(f"Error updating {test_file}: {exc}")
@@ -244,8 +247,6 @@ def func(
         logger.warning(f"Test file not found for {file_path}, skipping.")
         return
 
-    test_code: str = read_file_content(test_file) or ""
-
     # Detect test style
     test_style = _detect_test_style(test_file)
     logger.debug(f"Detected test style for {test_file}: {test_style}")
@@ -255,10 +256,12 @@ def func(
 
     logger.info(f"Updating {test_file} for function '{function_name}'.")
     try:
+        existing_content = read_file_content(test_file)
         updated_test: str = asyncio.run(
-            update_test_with_llm(source_code, test_code, str(file_path), [], other_tests_content, test_style)
+            update_test_with_llm(source_code, existing_content, str(file_path), [], other_tests_content, test_style)
         )
-        write_file_content(test_file, updated_test)
+        new_content = insert_new_test(existing_content, updated_test)
+        write_file_content(test_file, new_content)
         logger.info(f"✅ Test file updated successfully: {test_file}")
     except Exception as exc:  # pragma: no cover
         logger.error(f"Error updating {test_file}: {exc}")

{aiunittest-0.0.2 → aiunittest-0.0.5}/src/ai_unit_test/file_helper.py

@@ -54,6 +54,19 @@ def write_file_content(file_path: Path, content: str, mode: str = "w") -> None:
         f.write(content)
 
 
+def insert_new_test(existing_content: str, new_test: str) -> str:
+    """
+    Inserts a new test into the existing content, before the `if __name__ == "__main__":` block if it exists.
+    """
+    main_guard = 'if __name__ == "__main__":'
+    if main_guard in existing_content:
+        parts = existing_content.split(main_guard)
+        # Ensure there's a newline before the new test and before the main guard
+        new_test = "\n\n" + new_test.strip()
+        return parts[0].rstrip() + new_test + "\n\n" + main_guard + parts[1]
+    return existing_content + "\n" + new_test
+
+
 def extract_function_source(file_path: str, function_name: str) -> str | None:
     """Extracts the source code of a specific function from a file."""
     try:

{aiunittest-0.0.2 → aiunittest-0.0.5}/tests/unit/test_cli.py

@@ -90,7 +90,8 @@ def test_main_auto_discovery(
     mock_find_test_file.assert_called_once_with(str(Path("src/main.py")), "tests")
     mock_read_file_content.assert_any_call(Path("tests/test_main.py"))
     mock_update_test_with_llm.assert_called_once()
-    mock_write_file_content.assert_called_once_with(Path("tests/test_main.py"), "updated_test_code", mode="a")
+    assert mock_write_file_content.call_args[0][0] == Path("tests/test_main.py")
+    assert "updated_test_code" in mock_write_file_content.call_args[0][1]
 
 
 @patch("ai_unit_test.cli.write_file_content")
@@ -140,7 +141,8 @@ def test_main_explicit_args(
     mock_find_test_file.assert_called_once_with(str(Path("src/main.py")), "tests")
     mock_read_file_content.assert_any_call(Path("tests/test_main.py"))
     mock_update_test_with_llm.assert_called_once()
-    mock_write_file_content.assert_called_once_with(Path("tests/test_main.py"), "updated_test_code", mode="a")
+    assert mock_write_file_content.call_args[0][0] == Path("tests/test_main.py")
+    assert "updated_test_code" in mock_write_file_content.call_args[0][1]
 
 
 @patch("ai_unit_test.cli.logger.error")

aiunittest-0.0.2/.github/workflows/create-release.yml

@@ -1,53 +0,0 @@
-name: Create Release
-
-on:
-  push:
-    branches:
-      - main
-
-jobs:
-  create_release:
-    runs-on: ubuntu-latest
-    permissions:
-      contents: write # Para criar a tag e o release
-
-    steps:
-      - name: Checkout code
-        uses: actions/checkout@v4
-        with:
-          fetch-depth: 0 # Necessário para obter todas as tags
-
-      - name: Get latest version and calculate next
-        id: get_next_version
-        run: |
-          LATEST_TAG=$(git describe --tags --abbrev=0 2>/dev/null || echo "v0.0.0")
-          echo "Latest tag: $LATEST_TAG"
-
-          # Remove 'v' prefix and split into parts
-          VERSION_PARTS=$(echo $LATEST_TAG | sed 's/^v//' | tr '.' '\n')
-          MAJOR=$(echo $VERSION_PARTS | awk '{print $1}')
-          MINOR=$(echo $VERSION_PARTS | awk '{print $2}')
-          PATCH=$(echo $VERSION_PARTS | awk '{print $3}')
-
-          # Increment patch version
-          NEXT_PATCH=$((PATCH + 1))
-          NEXT_VERSION="v${MAJOR}.${MINOR}.${NEXT_PATCH}"
-          echo "Next version: $NEXT_VERSION"
-
-          echo "NEXT_VERSION=$NEXT_VERSION" >> $GITHUB_OUTPUT
-
-      - name: Create new tag
-        run: |
-          git config user.name "github-actions[bot]"
-          git config user.email "github-actions[bot]@users.noreply.github.com"
-          git tag -a ${{ steps.get_next_version.outputs.NEXT_VERSION }} -m "Release ${{ steps.get_next_version.outputs.NEXT_VERSION }}"
-          git push origin ${{ steps.get_next_version.outputs.NEXT_VERSION }}
-
-      - name: Create GitHub Release
-        uses: softprops/action-gh-release@v1
-        with:
-          tag_name: ${{ steps.get_next_version.outputs.NEXT_VERSION }}
-          name: Release ${{ steps.get_next_version.outputs.NEXT_VERSION }}
-          body: Automated release created on push to main.
-          draft: false
-          prerelease: false

aiunittest-0.0.2/.github/workflows/release.yml

@@ -1,60 +0,0 @@
-name: Release
-
-on:
-  release:
-    types: [published]
-
-jobs:
-  release:
-    runs-on: ubuntu-latest
-    permissions:
-      contents: write # Para criar o release e a tag
-      id-token: write # Para autenticação no PyPI
-
-    steps:
-      - name: Checkout code
-        uses: actions/checkout@v4
-        with:
-          fetch-depth: 0 # Necessário para setuptools_scm
-
-      - name: Set up Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: "3.x"
-
-      - name: Install build dependencies
-        run: |
-          python -m pip install --upgrade pip
-          pip install setuptools_scm build twine
-
-      - name: Build package
-        run: python -m build
-
-      - name: Create GitHub Release
-        uses: softprops/action-gh-release@v1
-        with:
-          tag_name: ${{ github.event.release.tag_name }}
-          name: Release ${{ github.event.release.tag_name }}
-          body: Automated release based on GitHub release event.
-          draft: false
-          prerelease: false
-          files: dist/*
-
-      - name: Publish to PyPI
-        uses: pypa/gh-action-pypi-publish@release/v1
-
-      - name: Set up Conda
-        uses: conda-incubator/setup-miniconda@v3
-        with:
-          python-version: 3.x
-          auto-update-conda: true
-          auto-activate-base: false
-
-      - name: Build and upload Conda package
-        shell: bash -l {0}
-        run: |
-          conda install -c conda-forge conda-build anaconda-client -y
-          export GIT_TAG=${{ github.event.release.tag_name }}
-          conda build conda --output-folder conda-dist
-          anaconda login --with-token ${{ secrets.CONDA_TOKEN }}
-          anaconda upload conda-dist/**/*.tar.bz2 --label main --skip-existing