Merge pull request #321 from Cloud-Code-AI/303-update-cli-to-include-…

…unittest-generation

feat: added unit test to cli

cli/kaizen_cli/cli.py

@@ -2,6 +2,7 @@
 import os
 import json
 from kaizen.generator.ui import UITestGenerator
+from kaizen.generator.unit_test import UnitTestGenerator
 
 CONFIG_FILE = os.path.expanduser("~/.myapp_config.json")
 
@@ -84,6 +85,14 @@ def ui_tests(url):
     UITestGenerator().generate_ui_tests(url)
 
 
+@cli.command()
+@click.argument("file_path", required=True)
+@click.argument("output_folder", default=None)
+def unit_test(file_path, output_folder):
+    """Run unit test generation"""
+    UnitTestGenerator().generate_tests(file_path, output_path=output_folder)
+
+
 @cli.group()
 def reviewer():
     """reviewer command group"""

cli/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kaizen-cli"
-version = "0.1.3"
+version = "0.1.4"
 description = ""
 authors = ["Saurav Panda <sgp65@cornell.edu>"]
 readme = "README.md"