Added test and rewrote the whole thing basically

Author: Arpan-206

internal/stage4.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"strings"
 
+	"github.com/codecrafters-io/shell-tester/internal/condition_reader"
 	"github.com/codecrafters-io/shell-tester/internal/logged_shell_asserter"
 	"github.com/codecrafters-io/shell-tester/internal/shell_executable"
 	"github.com/codecrafters-io/shell-tester/internal/test_cases"
@@ -43,9 +44,14 @@ func testExit(stageHarness *test_case_harness.TestCaseHarness) error {
 	readErr := shell.ReadUntilConditionOrTimeout(utils.AsBool(assertFn), logged_shell_asserter.SUBSEQUENT_READ_TIMEOUT)
 	output := virtual_terminal.BuildCleanedRow(shell.GetScreenState()[asserter.GetLastLoggedRowIndex()+1])
 
+	asserter.LogRemainingOutput()
+
 	// We're expecting EOF since the program should've terminated
 	if !errors.Is(readErr, shell_executable.ErrProgramExited) {
 		if readErr == nil {
+			return fmt.Errorf("Expected program to exit with 0 exit code, program is still running.")
+		} else if errors.Is(readErr, condition_reader.ErrConditionNotMet) {
+
 			return fmt.Errorf("Expected program to exit with 0 exit code, program is still running.")
 		} else {
 			// TODO: Other than ErrProgramExited, what other errors could we get? Are they user errors or internal errors?

internal/stages_test.go

@@ -33,6 +33,13 @@ func TestStages(t *testing.T) {
 			StdoutFixturePath:   "./test_helpers/fixtures/escape_codes",
 			NormalizeOutputFunc: normalizeTesterOutput,
 		},
+		"exit_error_fail": {
+			UntilStageSlug:      "pn5",
+			CodePath:            "./test_helpers/scenarios/exit_error",
+			ExpectedExitCode:    1,
+			StdoutFixturePath:   "./test_helpers/fixtures/exit_error",
+			NormalizeOutputFunc: normalizeTesterOutput,
+		},
 		"base_pass_bash": {
 			UntilStageSlug:      "ip1",
 			CodePath:            "./test_helpers/bash",

internal/test_helpers/fixtures/exit_error

@@ -0,0 +1,12 @@
+Debug = true
+
+[stage-4] Running tests for Stage #4: pn5
+[stage-4] Running ./your_shell.sh
+[your-program] $ invalid_apple_command
+[your-program] invalid_apple_command: command not found
+[stage-4] ✓ Received command not found message
+[your-program] $ exit 0
+[your-program] exit: command not found
+[your-program] $ 
+[stage-4] Expected program to exit with 0 exit code, program is still running.
+[stage-4] Test failed

internal/test_helpers/scenarios/exit_error/codecrafters.yml

@@ -0,0 +1,5 @@
+# Set this to true if you want debug logs.
+#
+# These can be VERY verbose, so we suggest turning them off
+# unless you really need them.
+debug: true 

internal/test_helpers/scenarios/exit_error/main.py

@@ -0,0 +1,22 @@
+import sys
+
+
+def main():
+    while True:
+        sys.stdout.write("$ ")
+        sys.stdout.flush()
+
+        command = input().strip()
+        parts = command.split(" ")
+        cmd = parts[0]
+        args = parts[1:]
+
+        if cmd == "exitt":
+            sys.exit(0)
+        else:
+            sys.stdout.write(f"{cmd}: command not found\n")
+            sys.stdout.flush()
+
+
+if __name__ == "__main__":
+    main() 

internal/test_helpers/scenarios/exit_error/your_shell.sh

@@ -0,0 +1,8 @@
+#!/bin/sh
+#
+# DON'T EDIT THIS!
+#
+# CodeCrafters uses this file to test your code. Don't make any changes here!
+#
+# DON'T EDIT THIS!
+exec python3 $(dirname "$0")/main.py "$@"