ultralytics-actions 0.0.72__tar.gz → 0.0.73__tar.gz

{ultralytics_actions-0.0.72/ultralytics_actions.egg-info → ultralytics_actions-0.0.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.0.72
+Version: 0.0.73
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -33,6 +33,7 @@ Requires-Dist: ruff>=0.9.1
 Requires-Dist: docformatter>=1.7.5
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
 Dynamic: license-file
 
 <a href="https://www.ultralytics.com/"><img src="https://raw.githubusercontent.com/ultralytics/assets/main/logo/Ultralytics_Logotype_Original.svg" width="320" alt="Ultralytics logo"></a>
@@ -42,12 +43,16 @@ Dynamic: license-file
 Welcome to the [Ultralytics Actions](https://github.com/ultralytics/actions) repository, your go-to solution for maintaining consistent code quality across Ultralytics Python and Swift projects. This GitHub Action is designed to automate the formatting of Python, Markdown, and Swift files, ensuring adherence to our coding standards and enhancing project maintainability.
 
 [![GitHub Actions Marketplace](https://img.shields.io/badge/Marketplace-Ultralytics_Actions-blue?style=flat&logo=github)](https://github.com/marketplace/actions/ultralytics-actions)
+
+[![Actions CI](https://github.com/ultralytics/actions/actions/workflows/ci.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/ci.yml)
 [![Ultralytics Actions](https://github.com/ultralytics/actions/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/format.yml)
+[![codecov](https://codecov.io/github/ultralytics/actions/graph/badge.svg?token=DoizJ1WS6j)](https://codecov.io/github/ultralytics/actions)
+[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
+[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
+
 [![Ultralytics Discord](https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue)](https://discord.com/invite/ultralytics)
 [![Ultralytics Forums](https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue)](https://community.ultralytics.com/)
 [![Ultralytics Reddit](https://img.shields.io/reddit/subreddit-subscribers/ultralytics?style=flat&logo=reddit&logoColor=white&label=Reddit&color=blue)](https://reddit.com/r/ultralytics)
-[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
-[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
 
 ## 📄 Actions Description
 
@@ -84,6 +89,11 @@ To integrate this action into your Ultralytics repository:
 2.  **Add the Action:** Configure the Ultralytics Actions in your workflow file as shown below:
 
     ```yaml
+    # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+    # Ultralytics Actions https://github.com/ultralytics/actions
+    # This workflow formats code and documentation in PRs to Ultralytics standards
+
     name: Ultralytics Actions
 
     on:
@@ -91,24 +101,31 @@ To integrate this action into your Ultralytics repository:
         types: [opened]
       pull_request:
         branches: [main]
-        types: [opened, closed]
+        types: [opened, closed, synchronize, review_requested]
+
+    permissions:
+      contents: write # Modify code in PRs
+      pull-requests: write # Add comments and labels to PRs
+      issues: write # Add comments and labels to issues
 
     jobs:
-      format:
-        runs-on: ubuntu-latest # Use 'macos-latest' if 'swift: true'
+      actions:
+        runs-on: ubuntu-latest
         steps:
-          - name: Run Ultralytics Formatting
+          - name: Run Ultralytics Actions
             uses: ultralytics/actions@main
             with:
-              token: ${{ secrets.GITHUB_TOKEN }} # Automatically generated, do not modify
-              labels: true # Autolabel issues and PRs using GPT-4.1 (requires 'openai_api_key')
-              python: true # Format Python code and docstrings with Ruff and docformatter
-              prettier: true # Format YAML, JSON, Markdown, and CSS with Prettier
-              swift: false # Format Swift code with swift-format (requires 'runs-on: macos-latest')
+              token: ${{ secrets.GITHUB_TOKEN }} # Auto-generated token
+              labels: true # Auto-label issues/PRs using AI
+              python: true # Format Python with Ruff and docformatter
+              prettier: true # Format YAML, JSON, Markdown, CSS
+              swift: false # Format Swift (requires macos-latest)
+              dart: false # Format Dart/Flutter
               spelling: true # Check spelling with codespell
-              links: true # Check for broken links with Lychee
-              summary: true # Generate PR summary with GPT-4.1 (requires 'openai_api_key')
-              openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Add your OpenAI API key as a repository secret
+              links: true # Check broken links with Lychee
+              summary: true # Generate AI-powered PR summaries
+              openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Powers PR summaries, labels and comments
+              brave_api_key: ${{ secrets.BRAVE_API_KEY }} # Used for broken link resolution
     ```
 
 3.  **Customize:** Adjust the `runs-on` runner and the boolean flags (`labels`, `python`, `prettier`, `swift`, `spelling`, `links`, `summary`) based on your project's needs. Remember to add your `OPENAI_API_KEY` as a secret in your repository settings if you enable `labels` or `summary`.

{ultralytics_actions-0.0.72 → ultralytics_actions-0.0.73}/README.md

@@ -5,12 +5,16 @@
 Welcome to the [Ultralytics Actions](https://github.com/ultralytics/actions) repository, your go-to solution for maintaining consistent code quality across Ultralytics Python and Swift projects. This GitHub Action is designed to automate the formatting of Python, Markdown, and Swift files, ensuring adherence to our coding standards and enhancing project maintainability.
 
 [![GitHub Actions Marketplace](https://img.shields.io/badge/Marketplace-Ultralytics_Actions-blue?style=flat&logo=github)](https://github.com/marketplace/actions/ultralytics-actions)
+
+[![Actions CI](https://github.com/ultralytics/actions/actions/workflows/ci.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/ci.yml)
 [![Ultralytics Actions](https://github.com/ultralytics/actions/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/actions/actions/workflows/format.yml)
+[![codecov](https://codecov.io/github/ultralytics/actions/graph/badge.svg?token=DoizJ1WS6j)](https://codecov.io/github/ultralytics/actions)
+[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
+[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
+
 [![Ultralytics Discord](https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue)](https://discord.com/invite/ultralytics)
 [![Ultralytics Forums](https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue)](https://community.ultralytics.com/)
 [![Ultralytics Reddit](https://img.shields.io/reddit/subreddit-subscribers/ultralytics?style=flat&logo=reddit&logoColor=white&label=Reddit&color=blue)](https://reddit.com/r/ultralytics)
-[![PyPI version](https://badge.fury.io/py/ultralytics-actions.svg)](https://badge.fury.io/py/ultralytics-actions)
-[![Downloads](https://static.pepy.tech/badge/ultralytics-actions)](https://www.pepy.tech/projects/ultralytics-actions)
 
 ## 📄 Actions Description
 
@@ -47,6 +51,11 @@ To integrate this action into your Ultralytics repository:
 2.  **Add the Action:** Configure the Ultralytics Actions in your workflow file as shown below:
 
     ```yaml
+    # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+    # Ultralytics Actions https://github.com/ultralytics/actions
+    # This workflow formats code and documentation in PRs to Ultralytics standards
+
     name: Ultralytics Actions
 
     on:
@@ -54,24 +63,31 @@ To integrate this action into your Ultralytics repository:
         types: [opened]
       pull_request:
         branches: [main]
-        types: [opened, closed]
+        types: [opened, closed, synchronize, review_requested]
+
+    permissions:
+      contents: write # Modify code in PRs
+      pull-requests: write # Add comments and labels to PRs
+      issues: write # Add comments and labels to issues
 
     jobs:
-      format:
-        runs-on: ubuntu-latest # Use 'macos-latest' if 'swift: true'
+      actions:
+        runs-on: ubuntu-latest
         steps:
-          - name: Run Ultralytics Formatting
+          - name: Run Ultralytics Actions
             uses: ultralytics/actions@main
             with:
-              token: ${{ secrets.GITHUB_TOKEN }} # Automatically generated, do not modify
-              labels: true # Autolabel issues and PRs using GPT-4.1 (requires 'openai_api_key')
-              python: true # Format Python code and docstrings with Ruff and docformatter
-              prettier: true # Format YAML, JSON, Markdown, and CSS with Prettier
-              swift: false # Format Swift code with swift-format (requires 'runs-on: macos-latest')
+              token: ${{ secrets.GITHUB_TOKEN }} # Auto-generated token
+              labels: true # Auto-label issues/PRs using AI
+              python: true # Format Python with Ruff and docformatter
+              prettier: true # Format YAML, JSON, Markdown, CSS
+              swift: false # Format Swift (requires macos-latest)
+              dart: false # Format Dart/Flutter
               spelling: true # Check spelling with codespell
-              links: true # Check for broken links with Lychee
-              summary: true # Generate PR summary with GPT-4.1 (requires 'openai_api_key')
-              openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Add your OpenAI API key as a repository secret
+              links: true # Check broken links with Lychee
+              summary: true # Generate AI-powered PR summaries
+              openai_api_key: ${{ secrets.OPENAI_API_KEY }} # Powers PR summaries, labels and comments
+              brave_api_key: ${{ secrets.BRAVE_API_KEY }} # Used for broken link resolution
     ```
 
 3.  **Customize:** Adjust the `runs-on` runner and the boolean flags (`labels`, `python`, `prettier`, `swift`, `spelling`, `links`, `summary`) based on your project's needs. Remember to add your `OPENAI_API_KEY` as a secret in your repository settings if you enable `labels` or `summary`.

{ultralytics_actions-0.0.72 → ultralytics_actions-0.0.73}/actions/__init__.py

@@ -22,4 +22,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.0.72"
+__version__ = "0.0.73"

ultralytics_actions-0.0.73/actions/update_file_headers.py

@@ -0,0 +1,191 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+import os
+from pathlib import Path
+
+from actions.utils import Action
+
+# Base header text
+HEADER = os.getenv("HEADER")
+
+# Map file extensions to comment styles
+COMMENT_MAP = {
+    # Python style
+    ".py": ("# ", None, None),
+    ".yml": ("# ", None, None),
+    ".yaml": ("# ", None, None),
+    ".toml": ("# ", None, None),
+    ".sh": ("# ", None, None),  # Bash scripts
+    ".bash": ("# ", None, None),  # Bash scripts
+    # C/C++/Java/JS style
+    ".c": ("// ", "/* ", " */"),  # C files
+    ".cpp": ("// ", "/* ", " */"),  # C++ files
+    ".h": ("// ", "/* ", " */"),  # C/C++ header files
+    ".hpp": ("// ", "/* ", " */"),  # C++ header files
+    ".swift": ("// ", "/* ", " */"),
+    ".js": ("// ", "/* ", " */"),
+    ".ts": ("// ", "/* ", " */"),  # TypeScript files
+    ".dart": ("// ", "/* ", " */"),  # Dart/Flutter files
+    ".rs": ("// ", "/* ", " */"),  # Rust files
+    ".java": ("// ", "/* ", " */"),  # Android Java
+    ".kt": ("// ", "/* ", " */"),  # Android Kotlin
+    # CSS style
+    ".css": (None, "/* ", " */"),
+    # HTML/XML style
+    ".html": (None, "<!-- ", " -->"),
+    ".xml": (None, "<!-- ", " -->"),  # Android XML
+    # MATLAB style
+    ".m": ("% ", None, None),
+}
+
+# Ignore these Paths (do not update their headers)
+IGNORE_PATHS = [
+    ".idea",
+    ".venv",
+    "env",
+    "node_modules",
+    ".git",
+    "__pycache__",
+    "mkdocs_github_authors.yaml",
+    # Build and distribution directories
+    "dist",
+    "build",
+    ".eggs",
+    "site",  # mkdocs build directory
+    # Generated code
+    "generated",
+    "auto_gen",
+    # Lock files
+    "lock",
+    # Minified files
+    ".min.js",
+    ".min.css",
+]
+
+
+def update_file(file_path, prefix, block_start, block_end, base_header):
+    """Update file with the correct header and proper spacing."""
+    try:
+        with open(file_path, encoding="utf-8") as f:
+            lines = f.readlines()
+    except Exception as e:
+        print(f"Error reading {file_path}: {e}")
+        return False
+
+    if not lines:
+        return False
+
+    # Format the header based on comment style
+    if prefix:
+        formatted_header = f"{prefix}{base_header}\n"
+    elif block_start and block_end:
+        formatted_header = f"{block_start}{base_header}{block_end}\n"
+    else:
+        formatted_header = f"# {base_header}\n"
+
+    # Keep shebang line if it exists
+    start_idx = 0
+    if lines and lines[0].startswith("#!"):
+        start_idx = 1
+
+    modified = False
+    new_lines = lines[:start_idx]
+    remaining_lines = lines[start_idx:]
+
+    # If first line is already the exact header we want
+    if remaining_lines and remaining_lines[0] == formatted_header:
+        # Check if spacing is correct
+        new_lines.append(remaining_lines[0])
+        if len(remaining_lines) > 1:
+            second_line = remaining_lines[1].strip()
+            if second_line == "" or second_line in ["#", "//", "/*", "*", "<!--", "%"]:
+                # Spacing is correct, append the rest
+                new_lines.extend(remaining_lines[1:])
+            else:
+                # Add blank line
+                new_lines.append("\n")
+                new_lines.extend(remaining_lines[1:])
+                modified = True
+        else:
+            # Only header exists, no need for blank line
+            pass
+    # Check if first line has AGPL but is not the exact header
+    elif remaining_lines and "AGPL" in remaining_lines[0] and remaining_lines[0] != formatted_header:
+        # Replace with proper header
+        new_lines.append(formatted_header)
+        modified = True
+
+        # Check if second line is blank or commented
+        if len(remaining_lines) > 1:
+            second_line = remaining_lines[1].strip()
+            if second_line == "" or second_line in ["#", "//", "/*", "*", "<!--", "%"]:
+                # Keep existing blank/comment line
+                new_lines.append(remaining_lines[1])
+                new_lines.extend(remaining_lines[2:])
+            else:
+                # Add blank line
+                new_lines.append("\n")
+                new_lines.extend(remaining_lines[1:])
+        else:
+            # Only header line, no need for blank line after
+            pass
+    # No header found, add it
+    else:
+        # Add header at the beginning
+        new_lines.append(formatted_header)
+        # Add blank line if content follows
+        if remaining_lines and remaining_lines[0].strip():
+            new_lines.append("\n")
+        new_lines.extend(remaining_lines)
+        modified = True
+
+    if modified:
+        try:
+            with open(file_path, "w", encoding="utf-8") as f:
+                f.writelines(new_lines)
+            return True
+        except Exception as e:
+            print(f"Error writing {file_path}: {e}")
+            return False
+
+    return False
+
+
+def main(*args, **kwargs):
+    """Automates file header updates for all files in the specified directory."""
+    event = Action(*args, **kwargs)
+
+    if "ultralytics" in event.repository.lower():
+        if event.is_repo_private() and event.repository.startswith("ultralytics/"):
+            from datetime import datetime
+
+            notice = f"Copyright © 2014-{datetime.now().year}"
+            header = f"Ultralytics Inc. 🚀 {notice} - CONFIDENTIAL - https://ultralytics.com - All Rights Reserved"
+        else:
+            header = "Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license"
+    elif HEADER and HEADER.lower() not in {"true", "false", "none"}:
+        header = HEADER
+    else:
+        return
+
+    directory = Path.cwd()
+    total = changed = unchanged = 0
+    for ext, comment_style in COMMENT_MAP.items():
+        prefix, block_start, block_end = comment_style
+
+        for file_path in directory.rglob(f"*{ext}"):
+            if any(part in str(file_path) for part in IGNORE_PATHS):
+                continue
+
+            total += 1
+            if update_file(file_path, prefix, block_start, block_end, header):
+                print(f"Updated: {file_path.relative_to(directory)}")
+                changed += 1
+            else:
+                unchanged += 1
+
+    print(f"Headers: {total}, Updated: {changed}, Unchanged: {unchanged}")
+
+
+if __name__ == "__main__":
+    main()

{ultralytics_actions-0.0.72 → ultralytics_actions-0.0.73}/actions/utils/github_utils.py

@@ -91,6 +91,10 @@ class Action:
             return json.loads(Path(event_path).read_text())
         return {}
 
+    def is_repo_private(self) -> bool:
+        """Checks if the repository is public using event data or GitHub API if needed."""
+        return self.event_data.get("repository", {}).get("private")
+
     def get_username(self) -> str | None:
         """Gets username associated with the GitHub token."""
         response = self.post(GITHUB_GRAPHQL_URL, json={"query": "query { viewer { login } }"})
@@ -152,6 +156,7 @@ class Action:
             "github.event_name": self.event_name,
             "github.event.action": self.event_data.get("action"),
             "github.repository": self.repository,
+            "github.repository.private": self.is_repo_private(),
             "github.event.pull_request.number": self.pr.get("number"),
             "github.event.pull_request.head.repo.full_name": self.pr.get("head", {}).get("repo", {}).get("full_name"),
             "github.actor": os.environ.get("GITHUB_ACTOR"),

{ultralytics_actions-0.0.72 → ultralytics_actions-0.0.73}/pyproject.toml

@@ -73,6 +73,7 @@ dependencies = [
 [project.optional-dependencies]
 dev = [
     "pytest",
+    "pytest-cov",
 ]
 
 [project.urls]
@@ -87,6 +88,7 @@ ultralytics-actions-first-interaction = "actions.first_interaction:main"
 ultralytics-actions-summarize-pr = "actions.summarize_pr:main"
 ultralytics-actions-summarize-release = "actions.summarize_release:main"
 ultralytics-actions-update-markdown-code-blocks = "actions.update_markdown_code_blocks:main"
+ultralytics-actions-header = "actions.update_file_headers:main"
 ultralytics-actions-info = "actions.utils:ultralytics_actions_info"
 
 [tool.setuptools]

ultralytics_actions-0.0.73/tests/test_cli_commands.py

@@ -0,0 +1,19 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+
+# Import CLI command modules
+from actions import (
+    first_interaction,
+    summarize_pr,
+    summarize_release,
+    update_markdown_code_blocks,
+)
+
+
+def test_importable_modules():
+    """Test that all modules can be imported without errors."""
+    # This is a simple test to ensure modules can be imported successfully
+    assert hasattr(first_interaction, "main")
+    assert hasattr(summarize_pr, "main")
+    assert hasattr(summarize_release, "main")
+    assert hasattr(update_markdown_code_blocks, "main")

ultralytics_actions-0.0.73/tests/test_common_utils.py

@@ -0,0 +1,63 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+from unittest.mock import MagicMock, patch
+
+from actions.utils.common_utils import (
+    allow_redirect,
+    brave_search,
+    clean_url,
+    remove_html_comments,
+)
+
+
+def test_remove_html_comments():
+    """Test removing HTML comments from strings."""
+    test_str = "Before <!-- Comment --> After"
+    assert remove_html_comments(test_str) == "Before  After"
+
+    # Multiline comment
+    test_str = "Before\n<!-- Comment\nline 2\nline 3 -->\nAfter"
+    assert remove_html_comments(test_str) == "Before\n\nAfter"
+
+    # No comments
+    test_str = "No comments here"
+    assert remove_html_comments(test_str) == "No comments here"
+
+
+def test_clean_url():
+    """Test cleaning URL strings."""
+    # Test removing quotes and trailing characters
+    assert clean_url('"https://example.com"') == "https://example.com"
+    assert clean_url("'https://example.com'") == "https://example.com"
+    assert clean_url("https://example.com.") == "https://example.com"
+    assert clean_url("https://example.com,") == "https://example.com"
+
+    # Test git URLs
+    assert clean_url("git+https://github.com/user/repo.git@main") == "https://github.com/user/repo"
+
+
+def test_allow_redirect():
+    """Test allowing URL redirects based on rules."""
+    # Should not allow - start ignores
+    assert not allow_redirect("https://youtu.be/xyz", "https://youtube.com")
+
+    # Should not allow - end ignores
+    assert not allow_redirect("https://example.com", "https://example.com/404")
+
+    # Empty end URL
+    assert not allow_redirect("https://example.com", "")
+
+
+@patch("requests.get")
+def test_brave_search(mock_get):
+    """Test Brave search API integration."""
+    mock_response = MagicMock()
+    mock_response.status_code = 200
+    mock_response.json.return_value = {
+        "web": {"results": [{"url": "https://example.com"}, {"url": "https://example.org"}]}
+    }
+    mock_get.return_value = mock_response
+
+    results = brave_search("test query", "test-api-key", count=2)
+    assert results == ["https://example.com", "https://example.org"]
+    mock_get.assert_called_once()

ultralytics_actions-0.0.73/tests/test_dispatch_actions.py

@@ -0,0 +1,146 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+from datetime import datetime
+from unittest.mock import MagicMock, patch
+
+from actions.dispatch_actions import (
+    RUN_CI_KEYWORD,
+    get_pr_branch,
+    main,
+    trigger_and_get_workflow_info,
+    update_comment,
+)
+
+
+def test_get_pr_branch():
+    """Test getting PR branch name."""
+    mock_event = MagicMock()
+    mock_event.event_data = {"issue": {"number": 123}}
+    mock_event.get_repo_data.return_value = {"head": {"ref": "feature-branch"}}
+
+    branch = get_pr_branch(mock_event)
+
+    assert branch == "feature-branch"
+    mock_event.get_repo_data.assert_called_once_with("pulls/123")
+
+
+def test_trigger_and_get_workflow_info():
+    """Test triggering workflows and getting info."""
+    mock_event = MagicMock()
+    mock_event.repository = "test/repo"
+
+    # Mock the workflow and runs responses separately
+    workflow_response = MagicMock()
+    workflow_response.status_code = 200
+    workflow_response.json.return_value = {"name": "CI Workflow"}
+
+    runs_response = MagicMock()
+    runs_response.status_code = 200
+    runs_response.json.return_value = {
+        "workflow_runs": [{"html_url": "https://github.com/test/repo/actions/runs/123", "run_number": 42}]
+    }
+
+    # Set up get method to return different responses for different URLs
+    def get_side_effect(url):
+        if "workflows/ci.yml" in url and "runs" not in url:
+            return workflow_response
+        elif "workflows/ci.yml/runs" in url:
+            return runs_response
+        # Return default response for unexpected URLs
+        default = MagicMock()
+        default.status_code = 404
+        return default
+
+    mock_event.get.side_effect = get_side_effect
+
+    # Use patch to skip time.sleep and limit to one workflow
+    with patch("time.sleep"):
+        with patch("actions.dispatch_actions.WORKFLOW_FILES", ["ci.yml"]):
+            results = trigger_and_get_workflow_info(mock_event, "feature-branch")
+
+    # Check results
+    assert len(results) == 1
+    assert results[0]["name"] == "CI Workflow"
+    assert results[0]["run_number"] == 42
+
+
+def test_update_comment_function():
+    """Test updating comment with workflow info."""
+    mock_event = MagicMock()
+    mock_event.repository = "test/repo"
+    mock_event.event_data = {"comment": {"id": 456}}
+
+    comment_body = f"Run tests please {RUN_CI_KEYWORD}"
+    triggered_actions = [
+        {
+            "name": "CI Workflow",
+            "file": "ci.yml",
+            "url": "https://github.com/test/repo/actions/workflows/ci.yml",
+            "run_number": 42,
+        }
+    ]
+
+    # Mock datetime to have a consistent timestamp
+    with patch("actions.dispatch_actions.datetime") as mock_datetime:
+        mock_datetime.now.return_value = datetime(2023, 1, 1, 12, 0, 0)
+        # Call without capturing return value
+        update_comment(mock_event, comment_body, triggered_actions, "feature-branch")
+
+    # Check that patch was called with expected content
+    mock_event.patch.assert_called_once()
+
+    # Verify key content in the comment body
+    args, kwargs = mock_event.patch.call_args
+    assert "https://api.github.com/repos/test/repo/issues/comments/456" in args[0]
+    assert "Actions Trigger" in kwargs["json"]["body"]
+    assert "CI Workflow" in kwargs["json"]["body"]
+    assert "2023-01-01 12:00:00 UTC" in kwargs["json"]["body"]
+
+
+def test_main_triggers_workflows():
+    """Test main function when comment contains trigger keyword."""
+    with patch("actions.dispatch_actions.Action") as MockAction:
+        # Configure mock
+        mock_event = MockAction.return_value
+        mock_event.event_name = "issue_comment"
+        mock_event.repository = "test/repo"
+        mock_event.event_data = {
+            "action": "created",
+            "issue": {"pull_request": {}},
+            "comment": {"body": f"Please run CI {RUN_CI_KEYWORD}", "user": {"login": "testuser"}, "id": 789},
+        }
+        mock_event.is_org_member.return_value = True
+
+        # Create minimal patches for the functions called by main
+        with patch("actions.dispatch_actions.get_pr_branch") as mock_get_branch:
+            with patch("actions.dispatch_actions.trigger_and_get_workflow_info") as mock_trigger:
+                with patch("actions.dispatch_actions.update_comment"):
+                    # Set return values
+                    mock_get_branch.return_value = "feature-branch"
+                    mock_trigger.return_value = [{"name": "CI", "file": "ci.yml", "url": "url", "run_number": 1}]
+
+                    # Call the function
+                    main()
+
+        # Verify main component calls were made
+        mock_event.is_org_member.assert_called_once_with("testuser")
+        mock_get_branch.assert_called_once()
+        mock_trigger.assert_called_once()
+
+
+def test_main_skips_non_pr_comments():
+    """Test main function skips non-PR comments."""
+    with patch("actions.dispatch_actions.Action") as MockAction:
+        # Configure mock
+        mock_event = MockAction.return_value
+        mock_event.event_name = "issue_comment"
+        mock_event.event_data = {
+            "action": "created",
+            "issue": {},  # No pull_request key
+            "comment": {"body": f"Please run CI {RUN_CI_KEYWORD}", "user": {"login": "testuser"}},
+        }
+
+        main()
+
+        # Verify toggle_eyes_reaction was not called
+        mock_event.toggle_eyes_reaction.assert_not_called()