universal-tester 1.1.0

AI-powered automatic test case generation for Java and Kotlin projects with Web UI and CLI

Universal Tester - AI-Powered Test Generation

Automatically generate high-quality JUnit tests for Java and Kotlin projects using AI.

Universal Tester leverages Large Language Models (LLMs) to understand your code and create comprehensive test suites with proper mocking, assertions, and edge case coverage. It comes with both a Command-Line Interface (CLI) and a Web UI for maximum flexibility.

---

🚀 Features

• ✅ Multi-LLM Support: Azure OpenAI, Google Gemini, Ollama
• ✅ Smart Import Detection: Automatic dependency analysis and import management
• ✅ Two Interfaces: CLI for automation, Web UI for interactive use
• ✅ Java & Kotlin: Full support for both languages
• ✅ Multiple Frameworks: JUnit 5, Mockito, MockK, Kotest
• ✅ Spring Boot Ready: Special handling for Spring annotations and dependency injection
• ✅ Incremental Testing: Generate tests for specific files or entire projects
• ✅ Context-Aware: Understands your code structure and dependencies

---

📦 Installation

pip install universal-tester

That's it! Both CLI and Web UI are included.

---

🎯 Quick Start

1. Configure Your LLM Provider

Create a .env file:

# Azure OpenAI
AZURE_OPENAI_API_KEY=your-api-key
AZURE_OPENAI_ENDPOINT=https://your-endpoint.openai.azure.com/
AZURE_OPENAI_DEPLOYMENT_NAME=gpt-4
LLM_PROVIDER=azure

# OR Google Gemini
GOOGLE_API_KEY=your-google-api-key
LLM_PROVIDER=google

# OR Ollama (local - FREE!)
OLLAMA_BASE_URL=http://localhost:11434
OLLAMA_MODEL=llama2
LLM_PROVIDER=ollama

2. Generate Tests (CLI)

# Generate tests from a ZIP file
universal-tester --input /path/to/your-project.zip

# With custom output directory
universal-tester -i project.zip -o ./generated-tests

3. Or Use the Web UI

# Launch the web interface
universal-tester-ui

Then upload your ZIP file and click "Generate Tests"!

---

💡 Usage Examples

CLI Examples

Basic test generation:

universal-tester --input myproject.zip

Kotlin with MockK:

universal-tester --input kotlin-app.zip --language kotlin --framework mockk

Use specific LLM provider:

universal-tester --input project.zip --llm-provider google

Verbose mode:

universal-tester --input project.zip --verbose

Web UI

1. Start the UI: universal-tester-ui
2. Open http://localhost:8000 in your browser
3. Upload your Java/Kotlin project (ZIP format)
4. Click "Generate Tests"
5. Download the generated test files

---

🎨 What Tests Look Like

Input: UserService.java

@Service
public class UserService {
    @Autowired
    private UserRepository userRepository;
    
    public User findById(Long id) {
        return userRepository.findById(id)
            .orElseThrow(() -> new UserNotFoundException(id));
    }
}

Generated: UserServiceTest.java

@ExtendWith(MockitoExtension.class)
class UserServiceTest {
    @Mock
    private UserRepository userRepository;
    
    @InjectMocks
    private UserService userService;
    
    @Test
    void findById_WhenUserExists_ReturnsUser() {
        // Arrange
        Long userId = 1L;
        User expectedUser = new User(userId, "John Doe");
        when(userRepository.findById(userId)).thenReturn(Optional.of(expectedUser));
        
        // Act
        User result = userService.findById(userId);
        
        // Assert
        assertNotNull(result);
        assertEquals(expectedUser.getId(), result.getId());
        assertEquals(expectedUser.getName(), result.getName());
        verify(userRepository).findById(userId);
    }
    
    @Test
    void findById_WhenUserNotFound_ThrowsException() {
        // Arrange
        Long userId = 999L;
        when(userRepository.findById(userId)).thenReturn(Optional.empty());
        
        // Act & Assert
        assertThrows(UserNotFoundException.class, () -> userService.findById(userId));
        verify(userRepository).findById(userId);
    }
}

---

🔧 Configuration

Environment Variables

Variable	Description	Required
LLM_PROVIDER	LLM provider: azure, google, or ollama	✅ Yes
AZURE_OPENAI_API_KEY	Azure OpenAI API key	If using Azure
AZURE_OPENAI_ENDPOINT	Azure endpoint URL	If using Azure
AZURE_OPENAI_DEPLOYMENT_NAME	Azure deployment name	If using Azure
GOOGLE_API_KEY	Google Gemini API key	If using Google
OLLAMA_BASE_URL	Ollama server URL	If using Ollama
OLLAMA_MODEL	Ollama model name	If using Ollama

CLI Options

universal-tester [OPTIONS]

Options:
  -i, --input PATH       Input ZIP file path (required)
  -o, --output PATH      Output directory (default: ./generated_tests)
  --llm-provider TEXT    LLM provider override (azure/google/ollama)
  --language TEXT        Source language (java/kotlin)
  --framework TEXT       Test framework (junit/mockito/mockk/kotest)
  -v, --verbose          Enable verbose logging
  --version              Show version
  --help                 Show help message

---

📚 Documentation

• Complete User Guide - Detailed documentation
• LLM Provider Setup - Configure Azure, Google, or Ollama
• Troubleshooting - Common issues and solutions
• FAQ - Frequently asked questions

---

🎯 Use Cases

For Developers

• Quickly create test scaffolding for new code
• Improve test coverage on legacy code
• Learn testing best practices from generated examples
• Speed up TDD workflow

For Teams

• Standardize test patterns across the codebase
• Onboard new team members with consistent test examples
• Reduce time spent writing boilerplate tests
• Focus on complex business logic testing

For CI/CD

• Integrate into build pipelines
• Generate tests automatically on code commits
• Maintain test coverage metrics
• Automate test creation for new features

---

🌟 Why Universal Tester?

Feature	Universal Tester	Manual Testing	Other Tools
Speed	⚡ Seconds	🐌 Hours	⚡ Fast
Quality	✅ AI-powered	👤 Variable	⚠️ Template-based
Customization	✅ Flexible	✅ Full control	❌ Limited
LLM Choice	✅ 3 providers	N/A	❌ Usually locked
UI + CLI	✅ Both	N/A	⚠️ Usually one
Cost	💰 Free + LLM API	💰💰 Developer time	💰💰💰 Expensive

---

🔒 Privacy & Security

• Your code stays yours: Code is only sent to your chosen LLM provider
• Use Ollama for local processing: 100% offline operation available
• No data storage: We don't store or log your code
• Open source: Audit the code yourself (Apache 2.0 License)

---

🛠️ Requirements

• Python 3.8 or higher
• Access to an LLM provider:
  • Azure OpenAI (paid)
  • Google Gemini (paid, but has free tier)
  • Ollama (free, runs locally)

---

📊 Project Structure

universal-tester/
├── universal_tester/
│   ├── core.py                    # Main test generation logic
│   ├── cli.py                     # Command-line interface
│   ├── llm/
│   │   ├── factory.py             # Multi-LLM support
│   │   ├── health_check.py        # LLM health monitoring
│   │   └── java_validator.py     # Code validation
│   ├── detectors/
│   │   ├── enhanced_import_detector.py  # Java import detection
│   │   └── kotlin_import_detector.py    # Kotlin import detection
│   ├── prompts/
│   │   ├── prompt_builder.py      # LLM prompt construction
│   │   ├── system_prompts.py      # System-level prompts
│   │   └── ui_messages.py         # UI messages
│   └── ui/
│       ├── chainlit_ui.py         # Web UI implementation
│       └── main.py                # UI entry point
├── USER_GUIDE.md                  # Complete documentation
└── README.md                      # This file

---

🤝 Contributing

Contributions are welcome! Here's how:

1. Fork the repository
2. Create a feature branch: git checkout -b feature/amazing-feature
3. Commit your changes: git commit -m 'Add amazing feature'
4. Push to the branch: git push origin feature/amazing-feature
5. Open a Pull Request

---

📝 License

This project is licensed under the Apache License 2.0 - see the LICENSE file for details.

Note on Commercial Use: While the software is open source under Apache 2.0, commercial use requires a separate license. Contact [e-mail] for commercial licensing.

Patent Notice: The author reserves all patent rights related to the concepts and methods embodied in this software.

---

🙏 Acknowledgments

• Built with LangChain
• UI powered by Chainlit
• Inspired by the need for better automated testing tools

---

📧 Contact & Support

• Author: Senthil Kumar Thanapal
• Email: [e-mail]
• PyPI Package: https://pypi.org/project/universal-tester/

---

🚀 Quick Links

• PyPI Package
• User Guide
• Documentation included in package

---

Made with ❤️ for the testing community

Star ⭐ this repo if you find it useful!