test: automate the main unit test

Author: lpm0073

app/agent.py

@@ -3,6 +3,8 @@
 User registration and management for Stackademy.
 """
 
+from typing import Optional, Tuple
+
 from .logging_config import get_logger, setup_logging
 from .prompt import completion
 
@@ -11,15 +13,18 @@
 logger = get_logger(__name__)
 
 
-def main():
+def main(prompts: Optional[Tuple[str, ...]] = None) -> None:
     """Main function to demonstrate user registration."""
+    print("=" * 50)
     print("Stackademy User Registration Demo")
     print("=" * 50)
 
-    user_prompt = input("Welcome to Stackademy! How can I assist you today? ")
+    i = 0
+    user_prompt = prompts[i] if prompts else input("Welcome to Stackademy! How can I assist you today? ")
 
     response, functions_called = completion(prompt=user_prompt)
     while response.choices[0].message.content != "Goodbye!":
+        i += 1
         message = response.choices[0].message
         response_message = message.content or ""
         logger.info("ChatGPT: %s", response_message.strip())
@@ -34,9 +39,17 @@ def main():
             followup_question = None
 
         if "get_courses" in functions_called:
-            user_prompt = input(followup_question or "Would you like to register for a course? ")
+            user_prompt = (
+                prompts[i]
+                if prompts and len(prompts) >= i
+                else input(followup_question or "Would you like to register for a course? ")
+            )
         elif "register_course" in functions_called:
-            user_prompt = input(followup_question or "Can I help you with anything else? ")
+            user_prompt = (
+                prompts[i]
+                if prompts and len(prompts) >= i
+                else input(followup_question or "Can I help you with anything else? ")
+            )
 
         response, functions_called = completion(prompt=user_prompt)
 

app/tests/test_application.py

@@ -26,7 +26,14 @@ def test_application_does_not_crash(self):
         """Test that the application returns a value."""
 
         # pylint: disable=broad-exception-caught
+        prompts = (
+            "Show me a list of available courses.",
+            "I would like to register for the CS107 Algorithms course. My name is John Doe and my email is john.doe@example.com",
+            "Thank you, that's all for now. Goodbye!",
+            "Goodbye!",
+            "Go away!",
+        )
         try:
-            main()
+            main(prompts=prompts)
         except Exception as e:
             self.fail(f"main raised an exception: {e}")