feat: add discriminator function to schema model (#89)

asyncapi · Jun 16, 2020 · 5055241 · 5055241
1 parent 580df7e
commit 5055241
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/models/schema.js b/lib/models/schema.js
@@ -344,6 +344,13 @@ class Schema extends Base {
     return this._json.deprecated;
   }
 
+  /**
+   * @returns {string}
+   */
+  discriminator() {
+    return this._json.discriminator;
+  }
+
   /**
    * @returns {boolean}
    */

diff --git a/test/models/schema_test.js b/test/models/schema_test.js
@@ -492,6 +492,15 @@ describe('Schema', function() {
     });
   });
 
+  describe('#discriminator()', function() {
+    it('should return a string', function() {
+      const doc = { type: 'string', discriminator: 'someType' };
+      const d = new Schema(doc);
+      expect(typeof d.discriminator()).to.be.equal('string');
+      expect(d.discriminator()).to.be.equal(doc.discriminator);
+    });
+  });
+
   describe('#readOnly()', function() {
     it('should return a boolean', function() {
       const doc = { type: 'string', readOnly: true };