Jest: SpyOn-Funktionalität

Ich versuche herauszufinden, wie ich den folgenden Anwendungsfall im Scherz am besten angehen kann.

Ich möchte keine Funktion verspotten, sondern einfach ausspionieren und die ursprüngliche Implementierung die Arbeit machen lassen. Nehmen wir an, ich möchte verfolgen, wie oft die Abfrage für den Postgres-Treiber aufgerufen wurde. Folgende Vorgehensweise funktioniert:

const querySpy = pg.Client.prototype.query = jest.fn(pg.Client.prototype.query);
expect(querySpy.mock.calls.length).toBe(2)

Das Problem ist, dass ich am Ende des Tests auch die ursprüngliche Implementierung zuweisen muss, damit andere in derselben Datei nicht beeinträchtigt werden. Es ist also einfach, die Testisolation zu durchbrechen, wenn eine Person eine solche Aufräumung vergisst.

Mir ist auch aufgefallen, dass Jasmin SpyOn hat, das auch im Scherz zu funktionieren scheint (aber nicht dokumentiert), was ziemlich elegant ist, so dass dasselbe erreicht werden kann durch:

const querySpy = spyOn(pg.Client.prototype, 'query').and.callThrough();
expect(querySpy.calls.count()).toEqual(3);

Und Jasmin behauptet, den Spion nach dem Test automatisch zu entfernen, was großartig ist.

Aber nicht sicher, ob das etwas ist, das offiziell Teil des Scherzes wird? Das Problem ist, dass Jasminspione eine andere Benutzeroberfläche haben als Spott im Scherz. Ich denke also, am besten wäre es, einfach die Scheinfunktionalität zu erweitern, damit sie leicht zum Ausspionieren mit spyOn oder ähnlichem verwendet werden kann.

Jedes Feedback ist willkommen.