test(schema): repro #7803

Automattic · Jul 7, 2019 · 82442da · 82442da
1 parent 9651ba7
commit 82442da
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/test/schema.test.js b/test/schema.test.js
@@ -2062,4 +2062,23 @@ describe('schema', function() {
     assert.strictEqual(testSchema.pathType('subpaths.list.0.options'),
       'adhocOrUndefined');
   });
+
+  it('supports pre(Array, Function) and post(Array, Function) (gh-7803)', function() {
+    const schema = Schema({ name: String });
+    schema.pre(['save', 'remove'], testMiddleware);
+    function testMiddleware() {
+      console.log('foo');
+    }
+
+    assert.equal(schema.s.hooks._pres.get('save').length, 1);
+    assert.equal(schema.s.hooks._pres.get('save')[0].fn, testMiddleware);
+    assert.equal(schema.s.hooks._pres.get('remove').length, 1);
+    assert.equal(schema.s.hooks._pres.get('remove')[0].fn, testMiddleware);
+
+    schema.post(['save', 'remove'], testMiddleware);
+    assert.equal(schema.s.hooks._posts.get('save').length, 1);
+    assert.equal(schema.s.hooks._posts.get('save')[0].fn, testMiddleware);
+    assert.equal(schema.s.hooks._posts.get('remove').length, 1);
+    assert.equal(schema.s.hooks._posts.get('remove')[0].fn, testMiddleware);
+  });
 });