🚀 TestGen AI

The Autonomous QA Agent from Your CLI

TestGen AI is a CLI tool that automatically generates, runs, and reports on test suites for your code using LLMs. Point it at a directory, set an API key, and let it handle the rest.

📦 Installation

Recommended: Use a virtual environment:

python -m venv testgen-env
# Activate it:
# Windows:
testgen-env\Scripts\activate
# macOS/Linux:
source testgen-env/bin/activate

# Install
pip install testgen-ai

Or install globally:

pip install testgen-ai

Verify:

testgen --version

That's it! Everything is configured automatically. On Windows, the first run will set up your system PATH automatically.

⚡ Quick Start

1. Set your API key (one-time)

# Google Gemini (default — has a free tier)
testgen config set GEMINI_API_KEY AIza...

# OpenAI
testgen config set OPENAI_API_KEY sk-...

# Anthropic / Claude
testgen config set ANTHROPIC_API_KEY sk-ant-...

# Ollama (local, no key needed)
testgen config set LLM_PROVIDER ollama

Keys are saved globally to ~/.testgen/.env and apply to every project automatically.

2. Generate tests

testgen generate ./src

3. Run tests

testgen test

4. Generate a report

testgen report

5. Or do everything in one shot

testgen auto ./src

✨ Features

• 🤖 AI-Powered Generation — Uses GPT-4, Claude, Gemini, or local Ollama to write real test cases, not boilerplate
• 🌍 14 Languages — Python, JavaScript, TypeScript, Go, Rust, Java, C#, Ruby, PHP, Swift, Kotlin, C++, HTML, CSS
• 👀 Watch Mode — Detects file saves and regenerates tests in real time (--watch)
• 📊 Terminal Dashboard — Color-coded test matrix with pass/fail/skip/duration per test
• 📈 HTML Reports — Professional reports with metrics, coverage insights, and execution distributions
• ⚡ Smart Context — AST-based extraction sends only relevant code to the LLM, keeping token costs low
• 🔄 One-command Workflow — testgen auto runs the full generate → execute → report pipeline

🎯 The AGER Architecture

TestGen AI operates on a 4-step loop:

┌──────────┐     ┌──────────┐     ┌──────────┐     ┌──────────┐
│ Analyze  │────▶│ Generate │────▶│ Execute  │────▶│ Report   │
│ (Scanner)│     │ (Brain)  │     │ (Runner) │     │ (Visuals)│
└──────────┘     └──────────┘     └──────────┘     └──────────┘

🎨 CLI Commands

📊 Terminal Output

╭─ ✅ Workflow Complete ────────────────────────╮
│ All Phases Complete!                          │
│                                               │
│ Summary:                                      │
│   • Files Scanned:    5                       │
│   • Tests Generated:  5                       │
│   • Tests Passed:     67 / 67                 │
│   • Tests Failed:     0                       │
│   • Total Duration:   85.81s                  │
│   • Report:           Generated               │
╰───────────────────────────────────────────────╯

🌍 Supported Languages

The target language's runtime (Node, Go, JDK, etc.) must be installed on your machine. pip install testgen-ai only installs the TestGen AI tool itself.

⚙️ Configuration

Config can be set globally (persists across all projects):

testgen config set GEMINI_API_KEY AIza...
testgen config set LLM_MODEL gemini-2.0-flash
testgen config set LLM_PROVIDER gemini

Or per-project via a .env file in the project root (overrides global):

OPENAI_API_KEY=sk-project-key
LLM_PROVIDER=openai
LLM_MODEL=gpt-4o
TEST_OUTPUT_DIR=./tests
MAX_CONTEXT_TOKENS=8000

Supported Providers

🔧 Optional: Browser / UI Testing

For Playwright-based UI test generation:

pip install testgen-ai[browser]
playwright install

The playwright install step downloads browser binaries and is a one-time setup per machine.

🛠️ Technology Stack

🤝 Contributing

1. Fork the repository
2. Create a feature branch: git checkout -b feature/my-feature
3. Make your changes
4. Run tests: pytest
5. Commit: git commit -m "Add my feature"
6. Push: git push origin feature/my-feature
7. Open a Pull Request

📚 Documentation

Full documentation: JayPatil165.github.io/TestGen-AI

📧 Contact

• Author: Jay Ajitkumar Patil
• Email: patiljay32145@gmail.com
• GitHub: @JayPatil165
• LinkedIn: jay-patil-4ab857326
• Issues: GitHub Issues

⭐ Star this repo if TestGen AI saves you time! ⭐
Made with ❤️ by developers, for developers