1 files changed, 104 insertions, 0 deletions

diff --git a/packages/tests/src/shared/peertube-runner-process.ts b/packages/tests/src/shared/peertube-runner-process.ts
new file mode 100644
index 000000000..3d1f299f2
--- /dev/null
+++ b/packages/tests/src/shared/peertube-runner-process.ts
@@ -0,0 +1,104 @@
+import { ChildProcess, fork, ForkOptions } from 'child_process'
+import execa from 'execa'
+import { join } from 'path'
+import { root } from '@peertube/peertube-node-utils'
+import { PeerTubeServer } from '@peertube/peertube-server-commands'
+
+export class PeerTubeRunnerProcess {
+  private app?: ChildProcess
+
+  constructor (private readonly server: PeerTubeServer) {
+
+  }
+
+  runServer (options: {
+    hideLogs?: boolean // default true
+  } = {}) {
+    const { hideLogs = true } = options
+
+    return new Promise<void>((res, rej) => {
+      const args = [ 'server', '--verbose', ...this.buildIdArg() ]
+
+      const forkOptions: ForkOptions = {
+        detached: false,
+        silent: true,
+        execArgv: [] // Don't inject parent node options
+      }
+
+      this.app = fork(this.getRunnerPath(), args, forkOptions)
+
+      this.app.stdout.on('data', data => {
+        const str = data.toString() as string
+
+        if (!hideLogs) {
+          console.log(str)
+        }
+      })
+
+      res()
+    })
+  }
+
+  registerPeerTubeInstance (options: {
+    registrationToken: string
+    runnerName: string
+    runnerDescription?: string
+  }) {
+    const { registrationToken, runnerName, runnerDescription } = options
+
+    const args = [
+      'register',
+      '--url', this.server.url,
+      '--registration-token', registrationToken,
+      '--runner-name', runnerName,
+      ...this.buildIdArg()
+    ]
+
+    if (runnerDescription) {
+      args.push('--runner-description')
+      args.push(runnerDescription)
+    }
+
+    return this.runCommand(this.getRunnerPath(), args)
+  }
+
+  unregisterPeerTubeInstance (options: {
+    runnerName: string
+  }) {
+    const { runnerName } = options
+
+    const args = [ 'unregister', '--url', this.server.url, '--runner-name', runnerName, ...this.buildIdArg() ]
+    return this.runCommand(this.getRunnerPath(), args)
+  }
+
+  async listRegisteredPeerTubeInstances () {
+    const args = [ 'list-registered', ...this.buildIdArg() ]
+    const { stdout } = await this.runCommand(this.getRunnerPath(), args)
+
+    return stdout
+  }
+
+  kill () {
+    if (!this.app) return
+
+    process.kill(this.app.pid)
+
+    this.app = null
+  }
+
+  getId () {
+    return 'test-' + this.server.internalServerNumber
+  }
+
+  private getRunnerPath () {
+    return join(root(), 'apps', 'peertube-runner', 'dist', 'peertube-runner.js')
+  }
+
+  private buildIdArg () {
+    return [ '--id', this.getId() ]
+  }
+
+  private runCommand (path: string, args: string[]) {
+    return execa.node(path, args, { env: { ...process.env, NODE_OPTIONS: '' } })
+  }
+}