Run test_steps within futurecoder in test_frontend

Author: alexmojaki

core/translation.py

@@ -32,6 +32,11 @@
 def set_language(language):
     global current_language
     global translation
+
+    if language == "None":
+        # Hack to explicitly set the language to None with an env var, which must be a string
+        return
+
     current_language = language
     translation = gettext.translation(
         "futurecoder",

tests/test_frontend.py

@@ -13,7 +13,8 @@
 
 del sys.modules["urllib3"]  # so that stub_module doesn't complain
 
-assets_dir = Path(__file__).parent / "test_frontend_assets"
+this_dir = Path(__file__).parent
+assets_dir = this_dir / "test_frontend_assets"
 assets_dir.mkdir(exist_ok=True)
 
 
@@ -104,6 +105,21 @@ def _tests(driver):
 >>> """,
     )
 
+    editor = driver.find_element_by_css_selector("#editor textarea")
+    run_button = driver.find_element_by_css_selector(".editor-buttons .btn-primary")
+    snoop_button = driver.find_element_by_css_selector(".editor-buttons .btn-success")
+
+    # Run test_steps within futurecoder!
+    run_code(editor, run_button, get_test_steps_code())
+    driver.implicitly_wait(20)
+    await_result(driver, ">>>", ">>> ")
+    driver.implicitly_wait(5)
+
+    # The above directly modifies the runner callbacks,
+    # so we need a separate run to reset the callback and print stuff
+    run_code(editor, run_button, 'print(open("golden_files/None/test_transcript.json").read())')
+    await_result(driver, "Introducing", (this_dir / "golden_files/en/test_transcript.json").read_text() + "\n>>> ")
+
     # Reverse until at first step
     for _ in range(10):
         reverse_button.click()
@@ -130,9 +146,6 @@ def _tests(driver):
     )
 
     # Run code in editor
-    editor = driver.find_element_by_css_selector("#editor textarea")
-    run_button = driver.find_element_by_css_selector(".editor-buttons .btn-primary")
-    snoop_button = driver.find_element_by_css_selector(".editor-buttons .btn-success")
     run_code(editor, run_button, code)
 
     # Check result in terminal
@@ -490,3 +503,13 @@ def check_choice_status(driver, choice_index, status):
         choice.get_attribute("class")
         for choice in driver.find_elements_by_class_name("prediction-choice")
     ]
+
+
+def get_test_steps_code():
+    code = (this_dir / "test_steps.py").read_text()
+    code += "os.environ['FUTURECODER_LANGUAGE'] = 'None'\n"
+    code += "os.environ['FIX_TESTS'] = '1'\n"
+    code += "test_steps()\n"
+    # Put all code in one line to avoid ace indentation issues
+    code = f"exec({code!r}, globals())"
+    return code

tests/test_steps.py

@@ -18,7 +18,8 @@
 
 
 def test_steps():
-    t.set_language(os.environ.get("FUTURECODER_LANGUAGE", "en"))
+    lang = os.environ.get("FUTURECODER_LANGUAGE", "en")
+    t.set_language(lang)
     list(load_chapters())
     runner.reset()
     transcript = []
@@ -62,7 +63,9 @@ def output_callback(data):
 
         assert response["passed"] == (not is_message)
 
-    path = Path(__file__).parent / "golden_files" / t.current_language / "test_transcript.json"
+    dirpath = Path(__file__).parent / "golden_files" / lang
+    dirpath.mkdir(parents=True, exist_ok=True)
+    path = dirpath / "test_transcript.json"
     if os.environ.get("FIX_TESTS", 0):
         dump = json.dumps(transcript, indent=4, sort_keys=True)
         path.write_text(dump)