Update app.py

app.py

@@ -6,9 +6,11 @@ from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
 
-# Define test functions
+# Test code snippets
+test_code = {
+    "Unit Test": """
 def run_unit_test():
-    """Unit Test: Test individual functions for correctness."""
+    \"\"\"Unit Test: Test individual functions for correctness.\"\"\"
     def add_numbers(a, b):
         return a + b
 
@@ -22,9 +24,10 @@ def run_unit_test():
     suite = unittest.TestLoader().loadTestsFromTestCase(TestAddNumbers)
     result = unittest.TextTestRunner(stream=sys.stdout, verbosity=2).run(suite)
     return result
-
+""",
+    "Smoke Test": """
 def run_smoke_test():
-    """Smoke Test: Quick test of core functionality."""
+    \"\"\"Smoke Test: Quick test of core functionality.\"\"\"
     import requests
     try:
         response = requests.get("https://jsonplaceholder.typicode.com/posts/1")
@@ -32,9 +35,10 @@ def run_smoke_test():
         return "Smoke Test Passed! API responded successfully."
     except AssertionError:
         return "Smoke Test Failed!"
-
+""",
+    "Integration Test": """
 def run_integration_test():
-    """Integration Test: Test the interaction between modules."""
+    \"\"\"Integration Test: Test the interaction between modules.\"\"\"
     import sqlite3
     try:
         conn = sqlite3.connect(":memory:")
@@ -48,9 +52,10 @@ def run_integration_test():
         return "Integration Test Passed! Database interaction works."
     except AssertionError:
         return "Integration Test Failed!"
-
+""",
+    "Regression Test": """
 def run_regression_test():
-    """Regression Test: Ensure changes do not break existing functionality."""
+    \"\"\"Regression Test: Ensure changes do not break existing functionality.\"\"\"
     def login(username, password):
         if username == "admin" and password == "password":
             return "Login successful"
@@ -63,62 +68,65 @@ def run_regression_test():
         return "Regression Test Passed!"
     except AssertionError:
         return "Regression Test Failed!"
-
+""",
+    "End-to-End Test (Selenium)": """
 def run_e2e_test():
-    """End-to-End Test: Simulate full user workflows using Selenium."""
+    \"\"\"End-to-End Test: Simulate full user workflows using Selenium.\"\"\"
     try:
-        # Simulating a basic Google search workflow
         driver = webdriver.Chrome()
         driver.get("https://www.google.com")
-
-        # Search for a term
         search_box = driver.find_element(By.NAME, "q")
         search_box.send_keys("Streamlit Selenium Test")
         search_box.send_keys(Keys.RETURN)
-
-        # Validate the results page
         assert "Streamlit Selenium Test" in driver.page_source
-
         driver.quit()
         return "E2E Test Passed! Google search workflow is functional."
     except Exception as e:
         return f"E2E Test Failed! Error: {e}"
-
+""",
+    "Acceptance Test": """
 def run_acceptance_test():
-    """Acceptance Test: Validate the application meets requirements."""
+    \"\"\"Acceptance Test: Validate the application meets requirements.\"\"\"
     try:
         meets_requirements = True  # Simulated check
         assert meets_requirements
         return "Acceptance Test Passed! Application meets requirements."
     except AssertionError:
         return "Acceptance Test Failed!"
+""",
+}
+
+# Test functions
+test_functions = {
+    "Unit Test": lambda: exec(test_code["Unit Test"], globals()),
+    "Smoke Test": lambda: exec(test_code["Smoke Test"], globals()),
+    "Integration Test": lambda: exec(test_code["Integration Test"], globals()),
+    "Regression Test": lambda: exec(test_code["Regression Test"], globals()),
+    "End-to-End Test (Selenium)": lambda: exec(test_code["End-to-End Test (Selenium)"], globals()),
+    "Acceptance Test": lambda: exec(test_code["Acceptance Test"], globals()),
+}
 
 # Streamlit Interface
-st.title("Interactive Testing Suite with Selenium")
+st.title("Interactive Testing Suite with Code Display")
 
 # Dropdown for test selection
-test_options = {
-    "Unit Test": run_unit_test,
-    "Smoke Test": run_smoke_test,
-    "Integration Test": run_integration_test,
-    "Regression Test": run_regression_test,
-    "End-to-End Test (Selenium)": run_e2e_test,
-    "Acceptance Test": run_acceptance_test,
-}
+selected_test = st.selectbox("Select a test to run:", list(test_functions.keys()), index=0)
 
-selected_test = st.selectbox("Select a test to run:", list(test_options.keys()))
-st.write(f"**Description:** {test_options[selected_test].__doc__}")
+# Show the code for the selected test
+st.subheader(f"Code for {selected_test}")
+st.code(test_code[selected_test], language="python")
 
 # Run the selected test
 if st.button("Run Test"):
-    if selected_test == "Unit Test":
-        # Capture stdout for unittest
-        with io.StringIO() as buf:
-            sys.stdout = buf
-            test_options[selected_test]()
+    # Redirect stdout to capture output
+    with io.StringIO() as buf:
+        sys.stdout = buf
+        try:
+            test_functions[selected_test]()
+        except Exception as e:
+            st.error(f"Error running {selected_test}: {e}")
+        finally:
             sys.stdout = sys.__stdout__
-            st.text(buf.getvalue())
-    else:
-        # Run other test functions
-        test_result = test_options[selected_test]()
-        st.write(f"**Result:** {test_result}")
+        output = buf.getvalue()
+    st.subheader(f"Results of {selected_test}")
+    st.text(output)