From a39d21332557b64d704ce9578370910b5ac622f5 Mon Sep 17 00:00:00 2001
From: Sergey Petushkov <petushkov.sergey@gmail.com>
Date: Mon, 1 Feb 2021 19:29:04 +0100
Subject: [PATCH] fix(node-runtime-worker-thread): Expose getShellPrompt method
 on the childProcessRuntime

---
 .../node-runtime-worker-thread/src/index.spec.ts    | 13 +++++++++++++
 packages/node-runtime-worker-thread/src/index.ts    |  8 +++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/packages/node-runtime-worker-thread/src/index.spec.ts b/packages/node-runtime-worker-thread/src/index.spec.ts
index bc9a17ef2d..4f8ee1eb9a 100644
--- a/packages/node-runtime-worker-thread/src/index.spec.ts
+++ b/packages/node-runtime-worker-thread/src/index.spec.ts
@@ -71,6 +71,19 @@ describe('WorkerRuntime', () => {
     });
   });
 
+  describe('getShellPrompt', () => {
+    const testServer = startTestServer('shared');
+
+    it('should return prompt when connected to the server', async() => {
+      const runtime = new WorkerRuntime(await testServer.connectionString());
+      const result = await runtime.getShellPrompt();
+
+      expect(result).to.match(/>/);
+
+      await runtime.terminate();
+    });
+  });
+
   describe('setEvaluationListener', () => {
     it('allows to set evaluation listener for runtime', async() => {
       const evalListener = {
diff --git a/packages/node-runtime-worker-thread/src/index.ts b/packages/node-runtime-worker-thread/src/index.ts
index 9dd94593ec..d02c69621d 100644
--- a/packages/node-runtime-worker-thread/src/index.ts
+++ b/packages/node-runtime-worker-thread/src/index.ts
@@ -61,7 +61,13 @@ class WorkerRuntime implements Runtime {
     });
 
     this.childProcessRuntime = createCaller(
-      ['init', 'evaluate', 'getCompletions', 'setEvaluationListener'],
+      [
+        'init',
+        'evaluate',
+        'getCompletions',
+        'setEvaluationListener',
+        'getShellPrompt'
+      ],
       this.childProcess
     );