Wait mock server termination

This commit is contained in:
Chocobozzz
2021-09-06 08:13:11 +02:00
parent 6d210220be
commit 70430c2796
6 changed files with 27 additions and 7 deletions

View File

@@ -4,6 +4,7 @@ import { Server } from 'http'
import { pipeline } from 'stream'
import { randomInt } from '@shared/core-utils'
import { ObjectStorageCommand } from '../server'
import { terminateServer } from './utils'
export class MockObjectStorage {
private server: Server
@@ -37,6 +38,6 @@ export class MockObjectStorage {
}
terminate () {
if (this.server) this.server.close()
return terminateServer(this.server)
}
}

View File

@@ -1,6 +1,7 @@
import express, { Request, Response } from 'express'
import { Server } from 'http'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'
type BlocklistResponse = {
data: {
@@ -32,6 +33,6 @@ export class MockBlocklist {
}
terminate () {
if (this.server) this.server.close()
return terminateServer(this.server)
}
}

View File

@@ -2,6 +2,7 @@
import { createServer, Server } from 'http'
import proxy from 'proxy'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'
class MockProxy {
private server: Server
@@ -16,7 +17,7 @@ class MockProxy {
}
terminate () {
if (this.server) this.server.close()
return terminateServer(this.server)
}
}

View File

@@ -0,0 +1,17 @@
import { Server } from 'http'
function terminateServer (server: Server) {
if (!server) return Promise.resolve()
return new Promise<void>((res, rej) => {
server.close(err => {
if (err) return rej(err)
return res()
})
})
}
export {
terminateServer
}