test(work_context): unit and e2e tests for streaming results

src/task/work.spec.ts

@@ -46,6 +46,17 @@ describe("Work Context", () => {
       });
     });
 
+    describe("runAndStream()", () => {
+      it("should execute runAndStream command", async () => {
+        const expectedResult = ActivityMock.createResult({ stdout: "Ok" });
+        activity.mockResults([expectedResult]);
+        const streamOfResults = await context.runAndStream("rm -rf");
+        for await (const result of streamOfResults) {
+          expect(result).toBe(expectedResult);
+        }
+      });
+    });
+
     describe("transfer()", () => {
       it("should execute transfer command", async () => {
         const result = ActivityMock.createResult({ stdout: "Ok" });

tests/e2e/tasks.spec.ts

@@ -180,4 +180,23 @@ describe("Task Executor", function () {
     const result = await executor.run(async (ctx) => ctx.getIp());
     expect(["192.168.0.2", "192.168.0.3"]).toContain(result);
   });
+
+  it("should run simple task and get results as stream", async () => {
+    executor = await TaskExecutor.create({
+      package: "golem/alpine:latest",
+      logger,
+    });
+    const streamOfResults = await executor.run(async (ctx) => ctx.runAndStream("echo 'Hello World'"));
+    expect(streamOfResults).toBeDefined();
+    for await (const result of streamOfResults!) {
+      expect(result.stdout).toContain("Hello World");
+      expect(result.result).toContain("Ok");
+    }
+    expect(logger.logs).toContain("Demand published on the market");
+    expect(logger.logs).toContain("New proposal has been received");
+    expect(logger.logs).toContain("Proposal has been responded");
+    expect(logger.logs).toContain("New proposal added to pool");
+    expect(logger.logs).toMatch(/Agreement confirmed by provider/);
+    expect(logger.logs).toMatch(/Activity .* created/);
+  });
 });