36 lines
536 B
TypeScript
36 lines
536 B
TypeScript
|
module PgMock {
|
||
|
|
const pg: any = {}
|
||
|
|
|
||
|
|
const query = jest.fn(() => ({
|
||
|
|
rows: [
|
||
|
|
{
|
||
|
|
a: "string",
|
||
|
|
b: 1,
|
||
|
|
},
|
||
|
|
],
|
||
|
|
}))
|
||
|
|
|
||
|
|
// constructor
|
||
|
|
function Client() {
|
||
|
|
}
|
||
|
|
|
||
|
|
Client.prototype.query = query
|
||
|
|
Client.prototype.connect = jest.fn()
|
||
|
|
Client.prototype.release = jest.fn()
|
||
|
|
|
||
|
|
function Pool() {
|
||
|
|
}
|
||
|
|
|
||
|
|
Pool.prototype.query = query
|
||
|
|
Pool.prototype.connect = jest.fn(() => {
|
||
|
|
// @ts-ignore
|
||
|
|
return new Client()
|
||
|
|
})
|
||
|
|
|
||
|
|
pg.Client = Client
|
||
|
|
pg.Pool = Pool
|
||
|
|
pg.queryMock = query
|
||
|
|
|
||
|
|
module.exports = pg
|
||
|
|
}
|