Update test

playground/app.js

@@ -22,7 +22,6 @@ openapm.instrument('express');
 openapm.instrument('mysql');
 
 const app = express();
-// app.use(openapm.REDMiddleware);
 
 const pool = mysql2.createPool(
   `mysql://express-app:[email protected]/express` //  If this throws an error, Change the db url to the one you're running on your machine locally or the testing instance you might have hosted.

src/OpenAPM.ts

@@ -63,16 +63,12 @@ export interface OpenAPMOptions {
   excludeDefaultLabels?: Array<DefaultLabels>;
 }
 
-export enum SupportedModules {
-  Express = 'express',
-  MySQL = 'mysql',
-  NestJS = 'nestjs'
-}
+export type SupportedModules = 'express' | 'mysql' | 'nestjs';
 
 const moduleNames = {
-  [SupportedModules.Express]: 'express',
-  [SupportedModules.MySQL]: 'mysql2',
-  [SupportedModules.NestJS]: '@nestjs/core'
+  express: 'express',
+  mysql: 'mysql2',
+  nestjs: '@nestjs/core'
 };
 
 const packageJson = getPackageJson();
@@ -301,20 +297,20 @@ export class OpenAPM extends EventEmitter {
 
   public instrument(moduleName: SupportedModules) {
     try {
-      if (moduleName === SupportedModules.Express) {
+      if (moduleName === 'express') {
         const express = require('express');
         instrumentExpress(express, this._REDMiddleware);
       }
-      if (moduleName === SupportedModules.MySQL) {
+      if (moduleName === 'mysql') {
         const mysql2 = require('mysql2');
         instrumentMySQL(mysql2);
       }
-      if (moduleName === SupportedModules.NestJS) {
+      if (moduleName === 'nestjs') {
         const { NestFactory } = require('@nestjs/core');
         instrumentNestFactory(NestFactory, this._REDMiddleware);
       }
     } catch (error) {
-      if (Object.values(SupportedModules).includes(moduleName)) {
+      if (Object.keys(moduleNames).includes(moduleName)) {
         console.log(error);
         throw new Error(
           `OpenAPM couldn't import the ${moduleNames[moduleName]} package, please install it.`

tests/metrics.test.ts

@@ -14,9 +14,10 @@ describe('REDMiddleware', () => {
   let parsedData: Array<Record<string, any>> = [];
 
   beforeAll(async () => {
-    app = express();
     openapm = new OpenAPM();
-    app.use(openapm.REDMiddleware);
+    openapm.instrument('express');
+
+    app = express();
 
     addRoutes(app);
     app.listen(3002);