chore: cleanup

Author: gagik

packages/mongodb-redact/src/index.ts

@@ -46,6 +46,7 @@ export function redact<T>(
   return message;
 }
 
+export { shouldRedactCommand } from './should-redact-command';
 export { redactUriCredentials } from './redact-uri-credentials';
 
 export default redact;

packages/mongodb-redact/src/should-redact-command.spec.ts

@@ -2,51 +2,27 @@ import { expect } from 'chai';
 import { shouldRedactCommand } from '.';
 
 describe('shouldRedactCommand', function () {
-  describe('shouldRedactCommand', function () {
-    it('returns true for createUser commands', function () {
-      expect(shouldRedactCommand('db.createUser({ user: "test" })')).to.be.true;
-    });
-
-    it('returns true for auth commands', function () {
-      expect(shouldRedactCommand('db.auth("user", "pass")')).to.be.true;
-    });
-
-    it('returns true for updateUser commands', function () {
-      expect(shouldRedactCommand('db.updateUser("user", { roles: [] })')).to.be
-        .true;
-    });
-
-    it('returns true for changeUserPassword commands', function () {
-      expect(shouldRedactCommand('db.changeUserPassword("user", "newpass")')).to
-        .be.true;
-    });
-
-    it('returns true for connect commands', function () {
-      expect(shouldRedactCommand('db = connect("mongodb://localhost")')).to.be
-        .true;
-    });
-
-    it('returns true for Mongo constructor', function () {
-      expect(shouldRedactCommand('new Mongo("mongodb://localhost")')).to.be
-        .true;
-    });
-
-    it('returns false for non-sensitive commands', function () {
-      expect(shouldRedactCommand('db.collection.find()')).to.be.false;
-    });
-
-    it('returns false for partial words like "authentication"', function () {
-      // The \b (word boundary) should prevent matching "auth" within "authentication"
-      expect(shouldRedactCommand('db.collection.find({authentication: true})'))
-        .to.be.false;
-    });
-
-    it('returns false for getUsers command', function () {
-      expect(shouldRedactCommand('db.getUsers()')).to.be.false;
-    });
-
-    it('returns false for show commands', function () {
-      expect(shouldRedactCommand('show dbs')).to.be.false;
-    });
-  });
+  for (const command of [
+    'db.createUser({ user: "test" })',
+    'db.auth("user", "pass")',
+    'db.updateUser("user", { roles: [] })',
+    'db.changeUserPassword("user", "newpass")',
+    'db = connect("mongodb://localhost")',
+    'new Mongo("mongodb://localhost")',
+  ]) {
+    it(`returns true for ${command}`, function () {
+      expect(shouldRedactCommand(command)).to.be.true;
+    });
+  }
+
+  for (const command of [
+    'db.collection.find()',
+    'db.collection.find({authentication: true})',
+    'db.getUsers()',
+    'show dbs',
+  ]) {
+    it(`returns false for ${command}`, function () {
+      expect(shouldRedactCommand(command)).to.be.false;
+    });
+  }
 });