mirror of
https://github.com/yu-i-i/overleaf-cep.git
synced 2026-06-06 15:49:01 +02:00
Jakob Ackermann
ad79c85cea
* [metrics] mongo: fail when command monitoring is not available * [metrics] mongo: add optional client label to pool metrics * [web] collect mongo stats on native client * [metrics] mongo: record namespace of find commands * [metrics] mongo: add counter for all the commands with collection label * [web] add missing mock GitOrigin-RevId: 9f378d8aa8d7167f56cf512681d63ef115c6dd98
90 lines
1.9 KiB
JavaScript
90 lines
1.9 KiB
JavaScript
import { afterEach, beforeEach, chai, vi } from 'vitest'
|
|
import 'sinon-mongoose'
|
|
import sinon from 'sinon'
|
|
import logger from '@overleaf/logger'
|
|
import Metrics from '@overleaf/metrics'
|
|
import sinonChai from 'sinon-chai'
|
|
import chaiAsPromised from 'chai-as-promised'
|
|
import mongoose from 'mongoose'
|
|
import mongodb from 'mongodb-legacy'
|
|
|
|
mongodb.ObjectId.cacheHexString = true
|
|
|
|
/*
|
|
* Chai configuration
|
|
*/
|
|
|
|
// add chai.should()
|
|
chai.should()
|
|
|
|
// Load sinon-chai assertions so expect(stubFn).to.have.been.calledWith('abc')
|
|
// has a nicer failure messages
|
|
chai.use(sinonChai)
|
|
|
|
// Load promise support for chai
|
|
chai.use(chaiAsPromised)
|
|
|
|
// Do not truncate assertion errors
|
|
chai.config.truncateThreshold = 0
|
|
vi.mock('@overleaf/logger', async () => {
|
|
return {
|
|
default: {
|
|
debug: vi.fn(),
|
|
info: vi.fn(),
|
|
log: vi.fn(),
|
|
warn: vi.fn(),
|
|
err: vi.fn(),
|
|
error: vi.fn(),
|
|
fatal: vi.fn(),
|
|
},
|
|
}
|
|
})
|
|
|
|
// Mock metrics in unit tests, can be overridden
|
|
vi.mock('@overleaf/metrics', () => {
|
|
return {
|
|
default: {
|
|
inc: sinon.stub(),
|
|
count: sinon.stub(),
|
|
timing: sinon.stub(),
|
|
summary: sinon.stub(),
|
|
gauge: sinon.stub(),
|
|
Timer: class {
|
|
constructor() {
|
|
this.labels = {}
|
|
}
|
|
|
|
done() {
|
|
return 1
|
|
}
|
|
},
|
|
prom: { Counter: sinon.stub(), Histogram: sinon.stub() },
|
|
mongodb: { monitor: sinon.stub() },
|
|
},
|
|
}
|
|
})
|
|
|
|
beforeEach(ctx => {
|
|
// This function is a utility to duplicate the behaviour of passing `done` in place of `next` in an express route handler.
|
|
ctx.rejectOnError = reject => {
|
|
return err => {
|
|
if (err) {
|
|
reject(err)
|
|
}
|
|
}
|
|
}
|
|
ctx.logger = logger
|
|
ctx.Metrics = Metrics
|
|
})
|
|
|
|
afterEach(() => {
|
|
vi.resetAllMocks()
|
|
vi.resetModules()
|
|
sinon.restore()
|
|
sinon.resetHistory()
|
|
const modelNames = mongoose.modelNames()
|
|
modelNames.forEach(name => {
|
|
delete mongoose.connection.models[name]
|
|
})
|
|
})
|