fix: case when message is oneOf for one element (#554)

asyncapi · May 24, 2022 · 48d8708 · 48d8708
1 parent 429c8e7
commit 48d8708
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/lib/models/operation.js b/lib/models/operation.js
@@ -50,6 +50,7 @@ class Operation extends OperationTraitable {
    */
   message(index) {
     if (!this._json.message) return null;
+    if (this._json.message.oneOf && this._json.message.oneOf.length === 1) return new Message(this._json.message.oneOf[0]);
     if (!this._json.message.oneOf) return new Message(this._json.message);
     if (typeof index !== 'number') return null;
     if (index > this._json.message.oneOf.length - 1) return null;

diff --git a/test/models/operation_test.js b/test/models/operation_test.js
@@ -93,6 +93,12 @@ describe('Operation', function() {
       expect(d.message(0).json()).to.be.deep.equal(doc.message.oneOf[0]);
       expect(d.message(1).json()).to.be.deep.equal(doc.message.oneOf[1]);
     });
+
+    it('should return a Message object if no index is provided and message is oneOf from one element', function() {
+      const doc = { message: { oneOf: [{ test: true }] } };
+      const d = new Operation(doc);
+      expect(d.message().json()).to.be.deep.equal(doc.message.oneOf[0]);
+    });
 
     it('should return null when index is out of bounds', function() {
       const doc = { message: { oneOf: [{ test: true }, { test: false }] } };