improve test setup

Author: smmoosavi

typescript/src/application.test.ts

@@ -5,47 +5,60 @@ import { TaskList } from '../src/task_list';
 class TestContext {
   input = new PassThrough();
   output = new PassThrough();
-  expectations: (() => boolean)[] = [];
-  tl = new TaskList(this.input, this.output);
-
-  async run() {
-    this.tl.run();
+  taskList = new TaskList(this.input, this.output);
+  run() {
+    this.taskList.run();
+  }
 
-    for (const expectation of this.expectations) {
-      await new Promise<void>((resolve) =>
-        this.output.once('readable', () => {
-          if (expectation()) resolve();
-        }),
-      );
+  emptyOutput() {
+    while (this.output.read()) {
+      // Clear the output stream
     }
+  }
 
-    this.input.end();
-    this.output.end();
+  sendCommand(command: string) {
+    this.emptyOutput();
+    this.input.write(`${command}\n`);
   }
 
-  expectOutput(lines: string[]) {
-    let text = lines.join('\n') + '\n';
-    this.expectations.push(() => {
-      const data = this.output.read(text.length)?.toString();
-      expect(data).toBe(text);
-      return !!data;
-    });
+  getOutput() {
+    let output = '';
+    let chunk = this.output.read();
+    if (chunk === null) {
+      return null;
+    }
+    while (chunk !== null) {
+      output += chunk.toString();
+      chunk = this.output.read();
+    }
+    return output;
+  }
+  getOutputWithoutPrompt() {
+    const output = this.getOutput();
+    if (output === null) {
+      return null;
+    }
+    // expect ends with '> ' and remove it
+    expect(output.endsWith('\n> ')).toBe(true);
+    return output.slice(0, -3); // Remove the trailing '> '
   }
 
-  sendCommand(command: string) {
-    this.expectations.push(() => {
-      const prompt = this.output.read(2)?.toString();
-      expect(prompt).toBe('> ');
-      this.input.write(`${command}\n`);
-      return !!prompt;
-    });
+  expectOutput(lines: string[]) {
+    const output = this.getOutputWithoutPrompt();
+    if (output === null) {
+      throw new Error('Output is null');
+    }
+    const expectedOutput = lines.join('\n');
+    expect(output).toBe(expectedOutput);
   }
 }
 
 describe('TaskList Application', () => {
   test('full interaction test', async () => {
     const ctx = new TestContext();
 
+    ctx.run();
+
     ctx.sendCommand('show');
 
     ctx.sendCommand('add project secrets');
@@ -84,7 +97,5 @@ describe('TaskList Application', () => {
       '',
     ]);
     ctx.sendCommand('quit');
-
-    await ctx.run();
   });
 });